Updated and tested retry logic, still more to be done
This commit is contained in:
parent
e1556a9795
commit
c7c757217c
2 changed files with 17 additions and 2 deletions
15
level_3/database/database_crud.py
Normal file
15
level_3/database/database_crud.py
Normal file
|
|
@ -0,0 +1,15 @@
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@contextmanager
|
||||||
|
def session_scope(session):
|
||||||
|
"""Provide a transactional scope around a series of operations."""
|
||||||
|
try:
|
||||||
|
yield session
|
||||||
|
session.commit()
|
||||||
|
except Exception as e:
|
||||||
|
session.rollback()
|
||||||
|
logger.error(f"Session rollback due to: {str(e)}")
|
||||||
|
raise
|
||||||
|
finally:
|
||||||
|
session.close()
|
||||||
|
|
@ -227,6 +227,6 @@ class BaseMemory:
|
||||||
n_of_observations=n_of_observations
|
n_of_observations=n_of_observations
|
||||||
)
|
)
|
||||||
|
|
||||||
async def delete_memories(self, params: Optional[str] = None):
|
async def delete_memories(self, namespace:str, params: Optional[str] = None):
|
||||||
return await self.vector_db.delete_memories(params)
|
return await self.vector_db.delete_memories(namespace,params)
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue