Procházet zdrojové kódy

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)
tags/v0.20.1
Kevin Hu před 2 měsíci
rodič
revize
5749aa30b0
Žádný účet není propojen s e-mailovou adresou tvůrce revize
2 změnil soubory, kde provedl 4 přidání a 1 odebrání
  1. 1
    1
      agent/tools/retrieval.py
  2. 3
    0
      api/db/services/llm_service.py

+ 1
- 1
agent/tools/retrieval.py Zobrazit soubor

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
- 0
api/db/services/llm_service.py Zobrazit soubor

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



Načítá se…
Zrušit
Uložit