From 35195f7f6c762930cdb58e142ee025e803c93933 Mon Sep 17 00:00:00 2001 From: Vasilije <8619304+Vasilije1990@users.noreply.github.com> Date: Sun, 19 Nov 2023 11:52:08 +0100 Subject: [PATCH] Added a few updates to poetry and tested packages --- .../database/graph_database/graph.py | 6 +++--- level_4/main.py | 10 ++++++---- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/level_4/cognitive_architecture/database/graph_database/graph.py b/level_4/cognitive_architecture/database/graph_database/graph.py index 99226d486..6c399f17f 100644 --- a/level_4/cognitive_architecture/database/graph_database/graph.py +++ b/level_4/cognitive_architecture/database/graph_database/graph.py @@ -199,9 +199,9 @@ class Neo4jGraphDB(AbstractGraphDB): # Create the user and memory components if they don't exist user_memory_cypher = f""" MERGE (user:User {{userId: '{user_id}'}}) - MERGE (semantic:SemanticMemory {{userId: '{user_id}', name: 'SemanticMemory' }}) - MERGE (episodic:EpisodicMemory {{userId: '{user_id}', name: 'EpisodicMemory'}}) - MERGE (buffer:Buffer {{userId: '{user_id}', name: 'Buffer' }}) + MERGE (semantic:SemanticMemory {{name: 'SemanticMemory', userId: '{user_id}' }}) + MERGE (episodic:EpisodicMemory {{name: 'EpisodicMemory' , userId: '{user_id}'}}) + MERGE (buffer:Buffer {{name: 'Buffer' , userId: '{user_id}' }}) MERGE (user)-[:HAS_SEMANTIC_MEMORY]->(semantic) MERGE (user)-[:HAS_EPISODIC_MEMORY]->(episodic) MERGE (user)-[:HAS_BUFFER]->(buffer) diff --git a/level_4/main.py b/level_4/main.py index 3a9c2bc94..e138b62d5 100644 --- a/level_4/main.py +++ b/level_4/main.py @@ -165,9 +165,11 @@ async def load_documents_to_vectorstore(session: AsyncSession, user_id: str, job namespace_class = namespace_id + "_class" logging.info("Namespace created with id %s", namespace_id) - - new_user = User(id=user_id) - await add_entity(session, new_user) + try: + new_user = User(id=user_id) + await add_entity(session, new_user) + except: + pass if job_id is None: job_id = str(uuid.uuid4()) @@ -469,7 +471,7 @@ async def main(): "path": [".data"], "strategy": "SUMMARY", } - # await load_documents_to_vectorstore(session, user_id, loader_settings=loader_settings) + await load_documents_to_vectorstore(session, user_id, loader_settings=loader_settings) await user_query_to_graph_db(session, user_id, "I walked in the forest yesterday and added to my list I need to buy some milk in the store and get a summary from a classical book i read yesterday") await add_documents_to_graph_db(session, user_id, loader_settings=loader_settings)