gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fixes to graph memory manager

Browse source
This commit is contained in:
Vasilije 2023-12-12 10:11:22 +01:00
parent 1ed03efa70
commit 47bc6796fa
2 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
level_4/Dockerfile
View file

@ -38,9 +38,6 @@ RUN apt-get update -q && \
#RUN playwright install
#RUN playwright install-deps
WORKDIR /app
COPY cognitive_architecture/ /app/cognitive_architecture
COPY main.py /app

4
level_4/cognitive_architecture/database/graph_database/graph.py
View file

@ -272,7 +272,7 @@ class Neo4jGraphDB(AbstractGraphDB):
# Update the function to generate Cypher CREATE statements for edges with unique variable names
def generate_create_statements_for_edges_with_uuid(self, user_id, edges, unique_mapping, base_node_mapping):
create_statements = []
with_statement = f"WITH {', '.join(unique_mapping.values())}, {user_id}, semantic, episodic, buffer"
with_statement = f"WITH {', '.join(unique_mapping.values())}, user , semantic, episodic, buffer"
create_statements.append(with_statement)
for edge in edges:
@ -285,7 +285,7 @@ class Neo4jGraphDB(AbstractGraphDB):
def generate_memory_type_relationships_with_uuid_and_time_context(self, user_id, nodes, unique_mapping, base_node_mapping):
create_statements = []
with_statement = f"WITH {', '.join(unique_mapping.values())}, {user_id}, semantic, episodic, buffer"
with_statement = f"WITH {', '.join(unique_mapping.values())}, user, semantic, episodic, buffer"
create_statements.append(with_statement)
# Loop through each node and create relationships based on memory_type