mirror of
https://github.com/Sea-Haven-Industries/open-swe.git
synced 2026-10-01 07:23:14 +00:00
38 lines
908 B
Python
38 lines
908 B
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-7",
|
||
|
|
"label": "Opus 4.7",
|
||
|
|
"efforts": ["low", "medium", "high", "xhigh", "max"],
|
||
|
|
"default_effort": "high",
|
||
|
|
},
|
||
|
|
{
|
||
|
|
"id": "openai:gpt-5.5",
|
||
|
|
"label": "GPT-5.5",
|
||
|
|
"efforts": ["low", "medium", "high", "xhigh"],
|
||
|
|
"default_effort": "xhigh",
|
||
|
|
},
|
||
|
|
]
|
||
|
|
|
||
|
|
SUPPORTED_MODEL_IDS: frozenset[str] = frozenset(m["id"] for m in SUPPORTED_MODELS)
|
||
|
|
|
||
|
|
|
||
|
|
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
|