Update config.py
Browse files
config.py
CHANGED
|
@@ -34,7 +34,8 @@ class AppConfig:
|
|
| 34 |
"supports_reasoning": True,
|
| 35 |
"supports_tool_calling": True,
|
| 36 |
"active_params": "5.1B",
|
| 37 |
-
"is_gpt_oss": True
|
|
|
|
| 38 |
},
|
| 39 |
"openai/gpt-oss-20b": {
|
| 40 |
"name": "GPT OSS 20B",
|
|
@@ -44,7 +45,8 @@ class AppConfig:
|
|
| 44 |
"supports_reasoning": True,
|
| 45 |
"supports_tool_calling": True,
|
| 46 |
"active_params": "3.6B",
|
| 47 |
-
"is_gpt_oss": True
|
|
|
|
| 48 |
},
|
| 49 |
"Qwen/Qwen3-VL-235B-A22B-Thinking": {
|
| 50 |
"name": "Qwen3-VL 235B A22B Thinking",
|
|
@@ -54,7 +56,8 @@ class AppConfig:
|
|
| 54 |
"supports_reasoning": True,
|
| 55 |
"supports_tool_calling": True,
|
| 56 |
"is_gpt_oss": False,
|
| 57 |
-
"active_params": "22B"
|
|
|
|
| 58 |
},
|
| 59 |
"Qwen/Qwen3-VL-235B-A22B-Instruct": {
|
| 60 |
"name": "Qwen3-VL 235B A22B Instruct",
|
|
@@ -64,7 +67,8 @@ class AppConfig:
|
|
| 64 |
"supports_reasoning": True,
|
| 65 |
"supports_tool_calling": True,
|
| 66 |
"is_gpt_oss": False,
|
| 67 |
-
"active_params": "22B"
|
|
|
|
| 68 |
}
|
| 69 |
}
|
| 70 |
|
|
@@ -190,6 +194,11 @@ class AppConfig:
|
|
| 190 |
if info.get("is_gpt_oss"):
|
| 191 |
return True
|
| 192 |
return model_id.startswith("openai/gpt-oss-")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 193 |
|
| 194 |
@classmethod
|
| 195 |
def get_model_endpoint(cls, model_id: str, provider_id: str) -> str:
|
|
|
|
| 34 |
"supports_reasoning": True,
|
| 35 |
"supports_tool_calling": True,
|
| 36 |
"active_params": "5.1B",
|
| 37 |
+
"is_gpt_oss": True,
|
| 38 |
+
"supports_vision": False
|
| 39 |
},
|
| 40 |
"openai/gpt-oss-20b": {
|
| 41 |
"name": "GPT OSS 20B",
|
|
|
|
| 45 |
"supports_reasoning": True,
|
| 46 |
"supports_tool_calling": True,
|
| 47 |
"active_params": "3.6B",
|
| 48 |
+
"is_gpt_oss": True,
|
| 49 |
+
"supports_vision": False
|
| 50 |
},
|
| 51 |
"Qwen/Qwen3-VL-235B-A22B-Thinking": {
|
| 52 |
"name": "Qwen3-VL 235B A22B Thinking",
|
|
|
|
| 56 |
"supports_reasoning": True,
|
| 57 |
"supports_tool_calling": True,
|
| 58 |
"is_gpt_oss": False,
|
| 59 |
+
"active_params": "22B",
|
| 60 |
+
"supports_vision": True
|
| 61 |
},
|
| 62 |
"Qwen/Qwen3-VL-235B-A22B-Instruct": {
|
| 63 |
"name": "Qwen3-VL 235B A22B Instruct",
|
|
|
|
| 67 |
"supports_reasoning": True,
|
| 68 |
"supports_tool_calling": True,
|
| 69 |
"is_gpt_oss": False,
|
| 70 |
+
"active_params": "22B",
|
| 71 |
+
"supports_vision": True
|
| 72 |
}
|
| 73 |
}
|
| 74 |
|
|
|
|
| 194 |
if info.get("is_gpt_oss"):
|
| 195 |
return True
|
| 196 |
return model_id.startswith("openai/gpt-oss-")
|
| 197 |
+
|
| 198 |
+
@classmethod
|
| 199 |
+
def supports_vision_model(cls, model_id: str) -> bool:
|
| 200 |
+
info = cls.AVAILABLE_MODELS.get(model_id, {})
|
| 201 |
+
return bool(info.get("supports_vision", False))
|
| 202 |
|
| 203 |
@classmethod
|
| 204 |
def get_model_endpoint(cls, model_id: str, provider_id: str) -> str:
|