From c7c757217c9c5e6476685e74416619624dd612dd Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Sun, 8 Oct 2023 22:29:35 +0200 Subject: [PATCH] Updated and tested retry logic, still more to be done --- level_3/database/database_crud.py | 15 +++++++++++++++ level_3/vectordb/basevectordb.py | 4 ++-- 2 files changed, 17 insertions(+), 2 deletions(-) create mode 100644 level_3/database/database_crud.py 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)