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

chore: linting fix

Browse source
This commit is contained in:
hajdul88 2025-10-17 14:24:25 +02:00
parent 2182b7be5f
commit 7357c81079
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/tests/conversation_history.py → cognee/tests/test_conversation_history.py
View file

@ -75,7 +75,7 @@ async def main():
history1 = await cache_engine.get_latest_qa(str(user.id), session_id_1, last_n=10)
assert len(history1) == 1, f"Expected at least 1 Q&A in history, got {len(history1)}"
our_qa = [h for h in history1 if h["question"] == "What is TechCorp?"]
assert len(our_qa) >= 1, f"Expected to find 'What is TechCorp?' in history"
assert len(our_qa) >= 1, "Expected to find 'What is TechCorp?' in history"
assert "answer" in our_qa[0] and "context" in our_qa[0], (
"Q&A should contain answer and context fields"
)
@ -112,7 +112,7 @@ async def main():
history3 = await cache_engine.get_latest_qa(str(user.id), session_id_2, last_n=10)
our_qa_session2 = [h for h in history3 if h["question"] == "What is DataCo?"]
assert len(our_qa_session2) == 1, f"Session 2 should have 'What is DataCo?' question"
assert len(our_qa_session2) == 1, "Session 2 should have 'What is DataCo?' question"
result4 = await cognee.search(
query_type=SearchType.GRAPH_COMPLETION,