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

Make spark model robuster to model name (#3514)

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


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2024-11-20 12:16:36 +08:00 committed by Yingfeng Zhang
parent cad341e794
commit 81c7b6afc5
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
conf/llm_factories.json
View File

@ -3,7 +3,7 @@
{ {
"name": "OpenAI", "name": "OpenAI",
"logo": "", "logo": "",
"tags": "LLM,TEXT EMBEDDING,SPEECH2TEXT,MODERATION", "tags": "LLM,TEXT EMBEDDING,TTS,TEXT RE-RANK,SPEECH2TEXT,MODERATION",
"status": "1", "status": "1",
"llm": [ "llm": [
{ {
@ -89,7 +89,7 @@
{ {
"name": "Tongyi-Qianwen", "name": "Tongyi-Qianwen",
"logo": "", "logo": "",
"tags": "LLM,TEXT EMBEDDING,TEXT RE-RANK,SPEECH2TEXT,MODERATION", "tags": "LLM,TEXT EMBEDDING,TEXT RE-RANK,TTS,SPEECH2TEXT,MODERATION",
"status": "1", "status": "1",
"llm": [ "llm": [
{ {
@ -352,7 +352,7 @@
{ {
"name": "Xinference", "name": "Xinference",
"logo": "", "logo": "",
"tags": "LLM,TEXT EMBEDDING,SPEECH2TEXT,MODERATION,TEXT RE-RANK", "tags": "LLM,TEXT EMBEDDING,TTS,SPEECH2TEXT,MODERATION,TEXT RE-RANK",
"status": "1", "status": "1",
"llm": [] "llm": []
}, },
@ -2303,7 +2303,7 @@
{ {
"name": "XunFei Spark", "name": "XunFei Spark",
"logo": "", "logo": "",
"tags": "LLM", "tags": "LLM,TTS",
"status": "1", "status": "1",
"llm": [] "llm": []
}, },

6
rag/llm/chat_model.py
View File

@ -1164,7 +1164,11 @@ class SparkChat(Base):
"Spark-Pro-128K": "pro-128k", "Spark-Pro-128K": "pro-128k",
"Spark-4.0-Ultra": "4.0Ultra", "Spark-4.0-Ultra": "4.0Ultra",
} }
model_version = model2version[model_name] version2model = {v: k for k, v in model2version.items()}
assert model_name in model2version or model_name in version2model, f"The given model name is not supported yet. Support: {list(model2version.keys())}"
if model_name in model2version:
model_version = model2version[model_name]
else: model_version = model_name
super().__init__(key, model_version, base_url) super().__init__(key, model_version, base_url)