24 lines
No EOL
677 B
Python
24 lines
No EOL
677 B
Python
|
|
from typing import Union, Dict
|
|
import re
|
|
|
|
import networkx as nx
|
|
from pydantic import BaseModel
|
|
|
|
from cognee.modules.search.llm.extraction.categorize_relevant_category import categorize_relevant_category
|
|
from cognee.shared.data_models import GraphDBType
|
|
|
|
|
|
async def search_cypher(query:str, graph: Union[nx.Graph, any]):
|
|
"""
|
|
Use a Cypher query to search the graph and return the results.
|
|
"""
|
|
|
|
|
|
from cognee.infrastructure import infrastructure_config
|
|
if infrastructure_config.get_config()["graph_engine"] == GraphDBType.NEO4J:
|
|
result = await graph.run(query)
|
|
return result
|
|
|
|
else:
|
|
raise ValueError("Unsupported graph engine type.") |