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

Fixed api issues, missing deps and got first endpoint to work

Browse source
This commit is contained in:
Vasilije 2023-11-17 21:04:53 +01:00
parent f3a5ee929e
commit ef7d1b3bc7
3 changed files with 16 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
level_4/cognitive_architecture/database/postgres/models/memory.py
View file

@ -10,7 +10,7 @@ class MemoryModel(Base):
id = Column(String, primary_key=True)
user_id = Column(String, ForeignKey('users.id'), index=True)
operation_id = Column(String, ForeignKey('operations.id'), index=True)
operation_id = Column(String, nullable=True)
memory_name = Column(String, nullable=True)
created_at = Column(DateTime, default=datetime.utcnow)
updated_at = Column(DateTime, onupdate=datetime.utcnow)

23
level_4/cognitive_architecture/vectorstore_manager.py
View file

@ -146,6 +146,7 @@ class Memory:
namespace: str = None,
memory_id: str = None,
memory_class = None,
job_id:str = None
) -> None:
self.load_environment_variables()
self.memory_id = memory_id
@ -156,6 +157,7 @@ class Memory:
self.namespace = namespace
self.memory_instances = []
self.memory_class = memory_class
self.job_id=job_id
# self.memory_class = DynamicBaseMemory(
# "Memory", user_id, str(self.memory_id), index_name, db_type, namespace
# )
@ -194,10 +196,10 @@ class Memory:
)
memory_class = DynamicBaseMemory(
memory_label, user_id, str(memory_id), index_name=memory_label, db_type='weaviate', **kwargs
memory_label, user_id, str(memory_id), index_name=memory_label , db_type='weaviate', **kwargs
)
return cls(user_id=user_id, session=session, memory_id=memory_id, memory_class=memory_class, **kwargs)
return cls(user_id=user_id, session=session, memory_id=memory_id, job_id =job_id, memory_class=memory_class, **kwargs)
async def list_memory_classes(self):
"""
@ -267,6 +269,7 @@ class Memory:
"""
try:
memory_id = str(uuid.uuid4())
logging.info("Job id %s", job_id)
memory = MemoryModel(
id=memory_id,
user_id=user_id,
@ -458,15 +461,15 @@ async def main():
await memory.add_dynamic_memory_class("semanticmemory", "SEMANTICMEMORY")
await memory.add_method_to_class(memory.semanticmemory_class, "add_memories")
await memory.add_method_to_class(memory.semanticmemory_class, "fetch_memories")
# sss = await memory.dynamic_method_call(memory.semanticmemory_class, 'add_memories',
# observation='some_observation', params=params, loader_settings=loader_settings)
sss = await memory.dynamic_method_call(memory.semanticmemory_class, 'add_memories',
observation='some_observation', params=params, loader_settings=loader_settings)
susu = await memory.dynamic_method_call(
memory.semanticmemory_class,
"fetch_memories",
observation="document summary",
)
print(susu)
# susu = await memory.dynamic_method_call(
# memory.semanticmemory_class,
# "fetch_memories",
# observation="document summary",
# )
# print(susu)
# Adding a dynamic memory class
# dynamic_memory = memory.add_dynamic_memory_class("DynamicMemory", "ExampleNamespace")

4
level_4/main.py
View file

@ -504,8 +504,8 @@ async def main():
"source": "URL",
"path": ["https://www.ibiblio.org/ebooks/London/Call%20of%20Wild.pdf"],
}
# 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")
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")
# await add_documents_to_graph_db(session, user_id)
# memory_instance = Memory(namespace='SEMANTICMEMORY')
# sss = await memory_instance.dynamic_method_call(memory_instance.semantic_memory_class, 'fetch_memories', observation='some_observation')