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

added tests

Browse source
This commit is contained in:
vasilije 2025-07-13 17:15:14 +02:00
parent 7055bd8a8f
commit 324a196552
1 changed files with 19 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
cognee/tests/test_falkordb.py
View file

@ -113,19 +113,23 @@ async def main():
entity_result = vector_engine.query(entity_query) entity_result = vector_engine.query(entity_query)
if entity_result.result_set: if entity_result.result_set:
# Convert to ScoredResult format # Convert to ScoredResult format
from cognee.infrastructure.databases.vector.models.ScoredResult import ScoredResult from cognee.infrastructure.databases.vector.models.ScoredResult import (
ScoredResult,
)
from cognee.shared.utils import parse_id from cognee.shared.utils import parse_id
node = entity_result.result_set[0][0] node = entity_result.result_set[0][0]
payload = dict(node.properties) if hasattr(node, 'properties') else {} payload = dict(node.properties) if hasattr(node, "properties") else {}
if 'text' not in payload and 'name' in payload: if "text" not in payload and "name" in payload:
payload['text'] = payload['name'] payload["text"] = payload["name"]
search_results = [ScoredResult( search_results = [
id=parse_id(payload.get('id', str(hash(str(node))))), ScoredResult(
score=1.0, id=parse_id(payload.get("id", str(hash(str(node))))),
payload=payload score=1.0,
)] payload=payload,
)
]
except Exception as e: except Exception as e:
print(f"❌ Error in fallback search: {e}") print(f"❌ Error in fallback search: {e}")