Public/kotaemon
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: nano-graphrag import when not installed

Browse Source
This commit is contained in:
Tadashi 2024-10-31 09:22:55 +07:00
parent cce3639074
commit fb241a467c
No known key found for this signature in database
GPG Key ID: 399380A00CC9028D
1 changed files with 5 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
libs/ktem/ktem/index/file/graph/nano_pipelines.py
View File

@ -29,13 +29,6 @@ try:
_find_most_related_text_unit_from_entities, _find_most_related_text_unit_from_entities,
) )
from nano_graphrag._utils import EmbeddingFunc from nano_graphrag._utils import EmbeddingFunc
from nano_graphrag.base import (
BaseGraphStorage,
BaseKVStorage,
BaseVectorStorage,
CommunitySchema,
TextChunkSchema,
)
except ImportError: except ImportError:
print( print(
@ -124,12 +117,12 @@ def clean_quote(input: str) -> str:
async def nano_graph_rag_build_local_query_context( async def nano_graph_rag_build_local_query_context(
graph_func, graph_func,
query, query,
query_param: QueryParam, query_param,
): ):
knowledge_graph_inst: BaseGraphStorage = graph_func.chunk_entity_relation_graph knowledge_graph_inst = graph_func.chunk_entity_relation_graph
entities_vdb: BaseVectorStorage = graph_func.entities_vdb entities_vdb = graph_func.entities_vdb
community_reports: BaseKVStorage[CommunitySchema] = graph_func.community_reports community_reports = graph_func.community_reports
text_chunks_db: BaseKVStorage[TextChunkSchema] = graph_func.text_chunks text_chunks_db = graph_func.text_chunks
results = await entities_vdb.query(query, top_k=query_param.top_k) results = await entities_vdb.query(query, top_k=query_param.top_k)
if not len(results): if not len(results):