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

Update docs

Browse source
This commit is contained in:
Vasilije 2024-03-17 18:31:16 +01:00
parent ef94708a98
commit 70fdf80add
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

BIN
.data/short_stories/soldiers-home.pdf Normal file
View file

Binary file not shown.

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

@ -35,12 +35,12 @@ class SearchParameters(BaseModel):
return value return value
async def perform_search(graph, search_type: str, params: Dict[str, Any]) -> List: async def search(graph, search_type: str, params: Dict[str, Any]) -> List:
search_params = SearchParameters(search_type=search_type, params=params) search_params = SearchParameters(search_type=search_type, params=params)
return await search(graph, [search_params]) return await specific_search(graph, [search_params])
async def search(graph, query_params: List[SearchParameters]) -> List: async def specific_search(graph, query_params: List[SearchParameters]) -> List:
search_functions: Dict[SearchType, Callable] = { search_functions: Dict[SearchType, Callable] = {
SearchType.ADJACENT: search_adjacent, SearchType.ADJACENT: search_adjacent,
SearchType.SIMILARITY: search_similarity, SearchType.SIMILARITY: search_similarity,
@ -83,7 +83,7 @@ if __name__ == "__main__":
search_type = 'ADJACENT' search_type = 'ADJACENT'
params = {'query': 'example query', 'other_param': {"node_id": "LLM_LAYER_SUMMARY:DOCUMENT:881ecb36-2819-54c3-8147-ed80293084d6"}} params = {'query': 'example query', 'other_param': {"node_id": "LLM_LAYER_SUMMARY:DOCUMENT:881ecb36-2819-54c3-8147-ed80293084d6"}}
results = await perform_search(graph, search_type, params) results = await search(graph, search_type, params)
print(results) print(results)
# Run the async main function # Run the async main function