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

fix: uuid parsing of search results

Browse source
This commit is contained in:
Boris Arzentar 2024-08-06 16:35:39 +02:00
parent a34acbc0ce
commit 1c89119487
2 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
cognee/api/client.py
View file

@ -328,10 +328,10 @@ def start_api_server(host: str = "0.0.0.0", port: int = 8000):
try:
logger.info("Starting server at %s:%s", host, port)
# import asyncio
# from cognee.modules.data.deletion import prune_system, prune_data
# asyncio.run(prune_data())
# asyncio.run(prune_system(metadata = True))
import asyncio
from cognee.modules.data.deletion import prune_system, prune_data
asyncio.run(prune_data())
asyncio.run(prune_system(metadata = True))
uvicorn.run(app, host = host, port = port)
except Exception as e:

5
cognee/api/v1/search/search.py
View file

@ -55,9 +55,10 @@ async def search(search_type: str, params: Dict[str, Any], user: User = None) ->
filtered_search_results = []
for search_result in search_results:
result_document_id = UUID(search_result["document_id"]) if "document_id" in search_result else None
document_id = search_result["document_id"] if "document_id" in search_result else None
document_id = UUID(document_id) if type(document_id) == str else document_id
if result_document_id is None or result_document_id in own_document_ids:
if document_id is None or document_id in own_document_ids:
filtered_search_results.append(search_result)
return filtered_search_results