pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-26 17:16:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: gen metadata error. (#12212)

Browse Source 
### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-12-25 19:01:22 +08:00
committed by GitHub
parent 1812491679
commit c8eeba5880
2 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
api/apps/document_app.py
View File

@ -611,6 +611,13 @@ async def run():
settings.docStoreConn.delete({"doc_id": id}, search.index_name(tenant_id), doc.kb_id) settings.docStoreConn.delete({"doc_id": id}, search.index_name(tenant_id), doc.kb_id)
if str(req["run"]) == TaskStatus.RUNNING.value: if str(req["run"]) == TaskStatus.RUNNING.value:
if req.get("apply_kb"):
e, kb = KnowledgebaseService.get_by_id(doc.kb_id)
if not e:
raise LookupError("Can't find this dataset!")
doc.parser_config["enable_metadata"] = kb.parser_config.get("enable_metadata", False)
doc.parser_config["metadata"] = kb.parser_config.get("metadata", {})
DocumentService.update_parser_config(doc.id, doc.parser_config)
doc_dict = doc.to_dict() doc_dict = doc.to_dict()
DocumentService.run(tenant_id, doc_dict, kb_table_num_map) DocumentService.run(tenant_id, doc_dict, kb_table_num_map)

2
rag/prompts/generator.py
View File

@ -827,7 +827,7 @@ async def relevant_chunks_with_toc(query: str, toc:list[dict], chat_mdl, topn: i
META_DATA = load_prompt("meta_data") META_DATA = load_prompt("meta_data")
async def gen_metadata(chat_mdl, schema:dict, content:str): async def gen_metadata(chat_mdl, schema:dict, content:str):
template = PROMPT_JINJA_ENV.from_string(META_DATA) template = PROMPT_JINJA_ENV.from_string(META_DATA)
for k, desc in schema.items(): for k, desc in schema["properties"].items():
if "enum" in desc and not desc.get("enum"): if "enum" in desc and not desc.get("enum"):
del desc["enum"] del desc["enum"]
if desc.get("enum"): if desc.get("enum"):