AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 12:45:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Reformat error message. (#4829)

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

#4828

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-02-10 16:47:53 +08:00 committed by GitHub
parent fa5c7edab4
commit 78982d88e0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
api/apps/llm_app.py
View File

@ -209,21 +209,22 @@ def add_llm():
} }
msg = "" msg = ""
mdl_nm = llm["llm_name"].split("___")[0]
if llm["model_type"] == LLMType.EMBEDDING.value: if llm["model_type"] == LLMType.EMBEDDING.value:
mdl = EmbeddingModel[factory]( mdl = EmbeddingModel[factory](
key=llm['api_key'], key=llm['api_key'],
model_name=llm["llm_name"], model_name=mdl_nm,
base_url=llm["api_base"]) base_url=llm["api_base"])
try: try:
arr, tc = mdl.encode(["Test if the api key is available"]) arr, tc = mdl.encode(["Test if the api key is available"])
if len(arr[0]) == 0: if len(arr[0]) == 0:
raise Exception("Fail") raise Exception("Fail")
except Exception as e: except Exception as e:
msg += f"\nFail to access embedding model({llm['llm_name']})." + str(e) msg += f"\nFail to access embedding model({mdl_nm})." + str(e)
elif llm["model_type"] == LLMType.CHAT.value: elif llm["model_type"] == LLMType.CHAT.value:
mdl = ChatModel[factory]( mdl = ChatModel[factory](
key=llm['api_key'], key=llm['api_key'],
model_name=llm["llm_name"], model_name=mdl_nm,
base_url=llm["api_base"] base_url=llm["api_base"]
) )
try: try:
@ -232,12 +233,12 @@ def add_llm():
if not tc: if not tc:
raise Exception(m) raise Exception(m)
except Exception as e: except Exception as e:
msg += f"\nFail to access model({llm['llm_name']})." + str( msg += f"\nFail to access model({mdl_nm})." + str(
e) e)
elif llm["model_type"] == LLMType.RERANK: elif llm["model_type"] == LLMType.RERANK:
mdl = RerankModel[factory]( mdl = RerankModel[factory](
key=llm["api_key"], key=llm["api_key"],
model_name=llm["llm_name"], model_name=mdl_nm,
base_url=llm["api_base"] base_url=llm["api_base"]
) )
try: try:
@ -245,12 +246,12 @@ def add_llm():
if len(arr) == 0: if len(arr) == 0:
raise Exception("Not known.") raise Exception("Not known.")
except Exception as e: except Exception as e:
msg += f"\nFail to access model({llm['llm_name']})." + str( msg += f"\nFail to access model({mdl_nm})." + str(
e) e)
elif llm["model_type"] == LLMType.IMAGE2TEXT.value: elif llm["model_type"] == LLMType.IMAGE2TEXT.value:
mdl = CvModel[factory]( mdl = CvModel[factory](
key=llm["api_key"], key=llm["api_key"],
model_name=llm["llm_name"], model_name=mdl_nm,
base_url=llm["api_base"] base_url=llm["api_base"]
) )
try: try:
@ -259,16 +260,16 @@ def add_llm():
if not tc: if not tc:
raise Exception(m) raise Exception(m)
except Exception as e: except Exception as e:
msg += f"\nFail to access model({llm['llm_name']})." + str(e) msg += f"\nFail to access model({mdl_nm})." + str(e)
elif llm["model_type"] == LLMType.TTS: elif llm["model_type"] == LLMType.TTS:
mdl = TTSModel[factory]( mdl = TTSModel[factory](
key=llm["api_key"], model_name=llm["llm_name"], base_url=llm["api_base"] key=llm["api_key"], model_name=mdl_nm, base_url=llm["api_base"]
) )
try: try:
for resp in mdl.tts("Hello~ Ragflower!"): for resp in mdl.tts("Hello~ Ragflower!"):
pass pass
except RuntimeError as e: except RuntimeError as e:
msg += f"\nFail to access model({llm['llm_name']})." + str(e) msg += f"\nFail to access model({mdl_nm})." + str(e)
else: else:
# TODO: check other type of models # TODO: check other type of models
pass pass