Updated and tested retry logic, still more to be done

This commit is contained in:
Vasilije 2023-10-08 22:29:35 +02:00
parent e1556a9795
commit c7c757217c
2 changed files with 17 additions and 2 deletions

View 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()

View file

@ -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)