From 5749aa30b099553e6b98c78128deac35a9866342 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Thu, 7 Aug 2025 16:14:47 +0800 Subject: [PATCH] Fix: model type error. (#9308) ### What problem does this PR solve? #9240 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue) --- agent/tools/retrieval.py | 2 +- api/db/services/llm_service.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/agent/tools/retrieval.py b/agent/tools/retrieval.py index cffbc75ea..046967956 100644 --- a/agent/tools/retrieval.py +++ b/agent/tools/retrieval.py @@ -86,7 +86,7 @@ class Retrieval(ToolBase, ABC): kb_ids.append(id) continue 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: raise Exception(f"Dataset({kb_nm}) does not exist.") kb_ids.append(kb.id) diff --git a/api/db/services/llm_service.py b/api/db/services/llm_service.py index 68d7c5e84..f18db1b01 100644 --- a/api/db/services/llm_service.py +++ b/api/db/services/llm_service.py @@ -225,6 +225,9 @@ class TenantLLMService(CommonService): if llm_id == llm["llm_name"]: 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): return llm.model_type