### What problem does this PR solve? #9240 ### Type of change - [x] Bug Fix (non-breaking change which fixes an issue)tags/v0.20.1
| 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) |
| 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 | ||||