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

Add docs

Browse source
This commit is contained in:
Vasilije 2024-03-13 17:52:42 +01:00
parent 13551dacf1
commit 231e16e656
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/__init__.py
View file

@ -1,4 +1,4 @@
from .api.v1.add.add import add from .api.v1.add.add import add
from .api.v1.cognify.cognify import cognify from .api.v1.cognify.cognify import cognify
from .api.v1.list_datasets.list_datasets import list_datasets from .api.v1.list_datasets.list_datasets import list_datasets
from .api.v1.search import search from .api.v1.search.search import search

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

@ -8,7 +8,7 @@ from cognee.modules.search.vector.search_similarity import search_similarity
from cognee.modules.search.graph.search_categories import search_categories from cognee.modules.search.graph.search_categories import search_categories
from cognee.modules.search.graph.search_neighbour import search_neighbour from cognee.modules.search.graph.search_neighbour import search_neighbour
from cognee.shared.data_models import GraphDBType from cognee.shared.data_models import GraphDBType
import asyncio
class SearchType(Enum): class SearchType(Enum):
ADJACENT = auto() ADJACENT = auto()
@ -48,7 +48,7 @@ async def search(graph, query_params: Dict[SearchType, Dict[str, Any]]) -> List:
return results return results
if __name__ == "__main__": if __name__ == "__main__":
import asyncio