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

Fix: model type error. (#9308)

Browse Source 
### What problem does this PR solve?

#9240

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-08-07 16:14:47 +08:00
committed by GitHub
parent a2e1f5618d
commit 5749aa30b0
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
agent/tools/retrieval.py
View File

@ -86,7 +86,7 @@ class Retrieval(ToolBase, ABC):
kb_ids.append(id) kb_ids.append(id)
continue continue
kb_nm = self._canvas.get_variable_value(id) kb_nm = self._canvas.get_variable_value(id)
e, kb = KnowledgebaseService.get_by_name(kb_nm) e, kb = KnowledgebaseService.get_by_name(kb_nm, self._canvas._tenant_id)
if not e: if not e:
raise Exception(f"Dataset({kb_nm}) does not exist.") raise Exception(f"Dataset({kb_nm}) does not exist.")
kb_ids.append(kb.id) kb_ids.append(kb.id)

3
api/db/services/llm_service.py
View File

@ -225,6 +225,9 @@ class TenantLLMService(CommonService):
if llm_id == llm["llm_name"]: if llm_id == llm["llm_name"]:
return llm["model_type"].split(",")[-1] return llm["model_type"].split(",")[-1]
for llm in TenantLLMService.query(llm_name=llm_id):
return llm.model_type
for llm in LLMService.query(llm_name=llm_id): for llm in LLMService.query(llm_name=llm_id):
return llm.model_type return llm.model_type