Cyril Dupland commited on
Commit ·
52214c6
1
Parent(s): a564144
Enable Mistral Large model & Disable GPT-5
Browse files- services/llm_service.py +21 -21
services/llm_service.py
CHANGED
|
@@ -137,20 +137,20 @@ class LLMService:
|
|
| 137 |
|
| 138 |
# OpenAI models
|
| 139 |
openai_models = [
|
| 140 |
-
{
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
|
| 144 |
-
|
| 145 |
-
|
| 146 |
-
},
|
| 147 |
-
{
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
},
|
| 154 |
{
|
| 155 |
"name": ModelName.GPT_4.value,
|
| 156 |
"provider": "openai",
|
|
@@ -183,13 +183,13 @@ class LLMService:
|
|
| 183 |
|
| 184 |
# Mistral AI models
|
| 185 |
mistral_models = [
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
|
| 191 |
-
|
| 192 |
-
|
| 193 |
|
| 194 |
# {
|
| 195 |
# "name": ModelName.MISTRAL_MEDIUM.value,
|
|
|
|
| 137 |
|
| 138 |
# OpenAI models
|
| 139 |
openai_models = [
|
| 140 |
+
# {
|
| 141 |
+
# "name": ModelName.GPT_5.value,
|
| 142 |
+
# "provider": "openai",
|
| 143 |
+
# "description": "GPT-5",
|
| 144 |
+
# "supports_streaming": False,
|
| 145 |
+
# # "context_window": 128000
|
| 146 |
+
# },
|
| 147 |
+
# {
|
| 148 |
+
# "name": ModelName.GPT_5_CHAT.value,
|
| 149 |
+
# "provider": "openai",
|
| 150 |
+
# "description": "GPT-5 Chat",
|
| 151 |
+
# "supports_streaming": True,
|
| 152 |
+
# # "context_window": 128000
|
| 153 |
+
# },
|
| 154 |
{
|
| 155 |
"name": ModelName.GPT_4.value,
|
| 156 |
"provider": "openai",
|
|
|
|
| 183 |
|
| 184 |
# Mistral AI models
|
| 185 |
mistral_models = [
|
| 186 |
+
{
|
| 187 |
+
"name": ModelName.MISTRAL_LARGE.value,
|
| 188 |
+
"provider": "mistralai",
|
| 189 |
+
"description": "Mistral Large",
|
| 190 |
+
"supports_streaming": True,
|
| 191 |
+
# "context_window": 32000
|
| 192 |
+
},
|
| 193 |
|
| 194 |
# {
|
| 195 |
# "name": ModelName.MISTRAL_MEDIUM.value,
|