diff --git a/level_3/database/database_crud.py b/level_3/database/database_crud.py new file mode 100644 index 000000000..74eee3a34 --- /dev/null +++ b/level_3/database/database_crud.py @@ -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() \ No newline at end of file diff --git a/level_3/vectordb/basevectordb.py b/level_3/vectordb/basevectordb.py index a764c280b..d39c2601b 100644 --- a/level_3/vectordb/basevectordb.py +++ b/level_3/vectordb/basevectordb.py @@ -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)