adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[fix]build knowledge graph

Browse Source
This commit is contained in:
lixiangcheng1
2025-12-27 17:17:29 +08:00
parent cd12844a7c
commit b4596eb47b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/app/tasks.py
View File

@@ -280,10 +280,10 @@ def build_graphrag_for_kb(kb_id: uuid.UUID):
build knowledge graph build knowledge graph
""" """
db = next(get_db()) # Manually call the generator db = next(get_db()) # Manually call the generator
db_document = None db_documents = None
db_knowledge = None db_knowledge = None
try: try:
db_document = db.query(Document).filter(Document.kb_id == kb_id).all() db_documents = db.query(Document).filter(Document.kb_id == kb_id).all()
db_knowledge = db.query(Knowledge).filter(Knowledge.id == kb_id).first() db_knowledge = db.query(Knowledge).filter(Knowledge.id == kb_id).first()
# 1. Prepare to configure chat_mdl、embedding_model、vision_model information # 1. Prepare to configure chat_mdl、embedding_model、vision_model information
chat_model = Base( chat_model = Base(
@@ -306,7 +306,7 @@ def build_graphrag_for_kb(kb_id: uuid.UUID):
# 2. get all document_ids from knowledge base # 2. get all document_ids from knowledge base
vector_service = ElasticSearchVectorFactory().init_vector(knowledge=db_knowledge) vector_service = ElasticSearchVectorFactory().init_vector(knowledge=db_knowledge)
total, items = vector_service.search_by_segment(document_id=None, query=None, pagesize=9999, page=1, asc=True) total, items = vector_service.search_by_segment(document_id=None, query=None, pagesize=9999, page=1, asc=True)
document_ids = [item.id for item in db_document] document_ids = [item.id for item in db_documents]
# 2. using graphrag # 2. using graphrag
if db_knowledge.parser_config.get("graphrag", {}).get("use_graphrag", False): if db_knowledge.parser_config.get("graphrag", {}).get("use_graphrag", False):