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

Reduce size of context, hack for now

Browse source
This commit is contained in:
Vasilije 2024-01-17 12:53:25 +01:00
parent d8f621f056
commit 94ef465444
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
main.py
View file

@ -322,9 +322,8 @@ async def add_documents_to_graph_db(session: AsyncSession, user_id: str= None, d
return e return e
class ResponseString(BaseModel): class ResponseString(BaseModel):
response: str = Field(..., default_factory=list) response: str = Field(default=None) # Defaulting to None or you can use a default string like ""
quotation: str = Field(..., default_factory=list) quotation: str = Field(default=None) # Same here
# #
@ -485,7 +484,8 @@ async def user_context_enrichment(session, user_id:str, query:str, generative_re
return context return context
else: else:
generative_result = generate_graph(context) generative_result = generate_graph(context)
return generative_result.response logging.info("Generative result is %s", generative_result.model_dump_json())
return generative_result.model_dump_json()
async def create_public_memory(user_id: str=None, labels:list=None, topic:str=None) -> Optional[int]: async def create_public_memory(user_id: str=None, labels:list=None, topic:str=None) -> Optional[int]: