Fix: graphrag, raptor can be null for api created kb issue (#8743)
### What problem does this PR solve? When knowledgebase/dataset created by API, graphrag and raptor can be null, and will trigger NoneType error when reach to this code, causing chunking task not able to finish.  Proposed solution will result in None and pass the condition check without error.  ### Type of change - ✅ Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
parent
19419281c3
commit
779932dcb0
1 changed files with 2 additions and 2 deletions
|
|
@ -564,10 +564,10 @@ class DocumentService(CommonService):
|
|||
prg = -1
|
||||
status = TaskStatus.FAIL.value
|
||||
elif finished:
|
||||
if d["parser_config"].get("raptor", {}).get("use_raptor") and not has_raptor:
|
||||
if (d["parser_config"].get("raptor") or {}).get("use_raptor") and not has_raptor:
|
||||
queue_raptor_o_graphrag_tasks(d, "raptor", priority)
|
||||
prg = 0.98 * len(tsks) / (len(tsks) + 1)
|
||||
elif d["parser_config"].get("graphrag", {}).get("use_graphrag") and not has_graphrag:
|
||||
elif (d["parser_config"].get("graphrag") or {}).get("use_graphrag") and not has_graphrag:
|
||||
queue_raptor_o_graphrag_tasks(d, "graphrag", priority)
|
||||
prg = 0.98 * len(tsks) / (len(tsks) + 1)
|
||||
else:
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue