mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-09-30 10:23:14 +00:00
a26d5d32cb
* feat: add Fireworks model provider with accurate reasoning levels Adds Kimi K2.6, DeepSeek V4 Pro, Nemotron 3 Ultra, GLM 5.1 via fireworks: provider, mapping per-model reasoning_effort. Surfaces each model's real effort range (incl. openai none, deepseek xhigh/max). * fix: drop serverless-unsupported Nemotron 3 Ultra, document FIREWORKS_API_KEY --------- Co-authored-by: open-swe[bot] <215916821+open-swe[bot]@users.noreply.github.com>
94 lines
3 KiB
Python
94 lines
3 KiB
Python
"""Supported models and reasoning efforts surfaced in the profile editor."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TypedDict
|
|
|
|
|
|
class ModelOption(TypedDict):
|
|
id: str
|
|
label: str
|
|
efforts: list[str]
|
|
default_effort: str
|
|
|
|
|
|
SUPPORTED_MODELS: list[ModelOption] = [
|
|
{
|
|
"id": "anthropic:claude-opus-4-8",
|
|
"label": "Opus 4.8",
|
|
"efforts": ["low", "medium", "high", "xhigh", "max"],
|
|
"default_effort": "high",
|
|
},
|
|
{
|
|
"id": "openai:gpt-5.5",
|
|
"label": "GPT-5.5",
|
|
"efforts": ["none", "low", "medium", "high", "xhigh"],
|
|
"default_effort": "xhigh",
|
|
},
|
|
{
|
|
"id": "fireworks:accounts/fireworks/models/kimi-k2p6",
|
|
"label": "Kimi K2.6",
|
|
"efforts": ["none", "low", "medium", "high"],
|
|
"default_effort": "high",
|
|
},
|
|
{
|
|
"id": "fireworks:accounts/fireworks/models/deepseek-v4-pro",
|
|
"label": "DeepSeek V4 Pro",
|
|
"efforts": ["none", "low", "medium", "high", "xhigh", "max"],
|
|
"default_effort": "high",
|
|
},
|
|
{
|
|
"id": "fireworks:accounts/fireworks/models/glm-5p1",
|
|
"label": "GLM 5.1",
|
|
"efforts": ["none", "low", "medium", "high"],
|
|
"default_effort": "high",
|
|
},
|
|
]
|
|
|
|
SUPPORTED_MODEL_IDS: frozenset[str] = frozenset(m["id"] for m in SUPPORTED_MODELS)
|
|
|
|
DEFAULT_MODEL_ID: str = "openai:gpt-5.5"
|
|
DEFAULT_MODEL_EFFORT: str = "medium"
|
|
|
|
|
|
def model_supports_effort(model_id: str, effort: str) -> bool:
|
|
for m in SUPPORTED_MODELS:
|
|
if m["id"] == model_id:
|
|
return effort in m["efforts"]
|
|
return False
|
|
|
|
|
|
def _provider_of(model_id: str) -> str | None:
|
|
provider, _, rest = model_id.partition(":")
|
|
return provider if rest else None
|
|
|
|
|
|
def provider_fallback_pair(model_id: object, effort: object = None) -> tuple[str, str] | None:
|
|
"""Newest supported ``(model_id, effort)`` for the same provider as ``model_id``.
|
|
|
|
Keeps a stored selection on its original provider when its exact id has
|
|
dropped out of the supported set (e.g. an Opus minor-version bump), instead
|
|
of falling through to the cross-provider global default. Preserves ``effort``
|
|
when the fallback model supports it, otherwise uses that model's default
|
|
effort. Returns ``None`` when no supported model shares the provider.
|
|
"""
|
|
if not isinstance(model_id, str):
|
|
return None
|
|
provider = _provider_of(model_id)
|
|
if provider is None:
|
|
return None
|
|
for m in SUPPORTED_MODELS:
|
|
if _provider_of(m["id"]) == provider:
|
|
new_effort = effort if (isinstance(effort, str) and effort in m["efforts"]) else None
|
|
return m["id"], new_effort or m["default_effort"]
|
|
return None
|
|
|
|
|
|
def default_model_pair() -> tuple[str, str]:
|
|
"""Hardcoded fallback (model_id, reasoning_effort) used when no team default is set."""
|
|
if DEFAULT_MODEL_ID in SUPPORTED_MODEL_IDS and model_supports_effort(
|
|
DEFAULT_MODEL_ID, DEFAULT_MODEL_EFFORT
|
|
):
|
|
return DEFAULT_MODEL_ID, DEFAULT_MODEL_EFFORT
|
|
first = SUPPORTED_MODELS[0]
|
|
return first["id"], first["default_effort"]
|