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
|
||||
)
|
||||
|
||||
async def delete_memories(self, params: Optional[str] = None):
|
||||
return await self.vector_db.delete_memories(params)
|
||||
async def delete_memories(self, namespace:str, params: Optional[str] = None):
|
||||
return await self.vector_db.delete_memories(namespace,params)
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue