AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-06-04 11:24:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add more mistral models. (#3826)

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

#3647

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
Kevin Hu 2024-12-03 15:18:38 +08:00 committed by GitHub
parent 95da6de9e1
commit 934dbc2e2b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 37 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
agent/canvas.py
View File

@ -234,6 +234,7 @@ class Canvas(ABC):
except Exception as e: except Exception as e:
yield {"content": "*Exception*: {}".format(e), "running_status": True} yield {"content": "*Exception*: {}".format(e), "running_status": True}
logging.exception("Canvas.run got exception") logging.exception("Canvas.run got exception")
ran += 1
if ran >= len(self.path[-1]) and waiting: if ran >= len(self.path[-1]) and waiting:
without_dependent_checking = waiting without_dependent_checking = waiting

48
conf/llm_factories.json
View File

@ -618,26 +618,32 @@
}, },
{ {
"llm_name": "open-mistral-7b", "llm_name": "open-mistral-7b",
"tags": "LLM,CHAT,32k", "tags": "LLM,CHAT,128k",
"max_tokens": 32000, "max_tokens": 128000,
"model_type": "chat"
},
{
"llm_name": "ministral-8b-latest",
"tags": "LLM,CHAT,128k",
"max_tokens": 128000,
"model_type": "chat"
},
{
"llm_name": "ministral-3b-latest",
"tags": "LLM,CHAT,128k",
"max_tokens": 128000,
"model_type": "chat" "model_type": "chat"
}, },
{ {
"llm_name": "mistral-large-latest", "llm_name": "mistral-large-latest",
"tags": "LLM,CHAT,32k", "tags": "LLM,CHAT,128k",
"max_tokens": 32000, "max_tokens": 128000,
"model_type": "chat" "model_type": "chat"
}, },
{ {
"llm_name": "mistral-small-latest", "llm_name": "mistral-small-latest",
"tags": "LLM,CHAT,32k", "tags": "LLM,CHAT,128k",
"max_tokens": 32000, "max_tokens": 128000,
"model_type": "chat"
},
{
"llm_name": "mistral-medium-latest",
"tags": "LLM,CHAT,32k",
"max_tokens": 32000,
"model_type": "chat" "model_type": "chat"
}, },
{ {
@ -646,11 +652,29 @@
"max_tokens": 32000, "max_tokens": 32000,
"model_type": "chat" "model_type": "chat"
}, },
{
"llm_name": "mistral-nemo",
"tags": "LLM,CHAT,128k",
"max_tokens": 128000,
"model_type": "chat"
},
{ {
"llm_name": "mistral-embed", "llm_name": "mistral-embed",
"tags": "LLM,CHAT,8k", "tags": "LLM,CHAT,8k",
"max_tokens": 8192, "max_tokens": 8192,
"model_type": "embedding" "model_type": "embedding"
},
{
"llm_name": "pixtral-large-latest",
"tags": "LLM,CHAT,32k",
"max_tokens": 32000,
"model_type": "image2text"
},
{
"llm_name": "pixtral-12b",
"tags": "LLM,CHAT,32k",
"max_tokens": 32000,
"model_type": "image2text"
} }
] ]
}, },