symphony-core 0.0.1__py3-none-any.whl

core_ai/__init__.py

@@ -0,0 +1,81 @@
+from core_ai.content import (
+    Content,
+    IMAGE_MIME_BY_SUFFIX,
+    estimate_content_tokens,
+    image_part,
+    image_part_from_bytes,
+    normalize_content,
+    sniff_image_media_type,
+    split_text_and_images,
+    text_from_content,
+    to_anthropic_blocks,
+    to_gemini_parts,
+    to_openai_chat_content,
+    to_openai_responses_content,
+)
+from core_ai.models import ModelAPI, ModelCatalog, ModelInfo, get_model, list_models
+from core_ai.providers.anthropic import AnthropicProvider
+from core_ai.providers.base import BaseProvider
+from core_ai.providers.catalog import (
+    PROVIDERS,
+    MissingProviderCredentials,
+    ProviderSpec,
+    configured_provider_ids,
+    find_provider,
+    get_provider,
+    has_configured_provider,
+    provider_is_configured,
+)
+from core_ai.providers.defaults import build_default_registry, default_model_id
+from core_ai.providers.gemini import GeminiProvider
+from core_ai.providers.grok import GrokProvider
+from core_ai.providers.local import LocalProvider
+from core_ai.providers.ollama import OllamaProvider
+from core_ai.providers.openai import OpenAIProvider
+from core_ai.providers.openrouter import OpenRouterProvider
+from core_ai.providers.vercel import VercelProvider
+from core_ai.registry import ModelRegistry
+from core_ai.types import Message, StreamEvent
+
+__all__ = [
+    "AnthropicProvider",
+    "BaseProvider",
+    "Content",
+    "GeminiProvider",
+    "GrokProvider",
+    "IMAGE_MIME_BY_SUFFIX",
+    "LocalProvider",
+    "Message",
+    "MissingProviderCredentials",
+    "ModelAPI",
+    "ModelCatalog",
+    "ModelInfo",
+    "ModelRegistry",
+    "OllamaProvider",
+    "OpenAIProvider",
+    "OpenRouterProvider",
+    "PROVIDERS",
+    "ProviderSpec",
+    "StreamEvent",
+    "VercelProvider",
+    "build_default_registry",
+    "configured_provider_ids",
+    "default_model_id",
+    "estimate_content_tokens",
+    "find_provider",
+    "get_model",
+    "get_provider",
+    "has_configured_provider",
+    "image_part",
+    "image_part_from_bytes",
+    "list_models",
+    "normalize_content",
+    "provider_is_configured",
+    "sniff_image_media_type",
+    "split_text_and_images",
+    "text_from_content",
+    "to_anthropic_blocks",
+    "to_gemini_parts",
+    "to_openai_chat_content",
+    "to_openai_responses_content",
+]

core_ai/content.py

@@ -0,0 +1,384 @@
+"""Provider-neutral multimodal content helpers.
+
+Canonical user/assistant parts look like:
+
+    {"type": "text", "text": "hello"}
+    {"type": "image", "media_type": "image/png", "data": "<base64>", "filename": "shot.png"}
+    {"type": "image", "media_type": "image/jpeg", "url": "https://example.com/a.jpg"}
+
+Providers translate this shape into their native request payloads. Incoming
+OpenAI / Anthropic / Gemini image parts are normalized back to the canonical
+form so mixed histories keep working.
+"""
+
+from __future__ import annotations
+
+import base64
+import re
+from typing import Any, Dict, Iterable, List, Optional, Sequence
+from urllib.parse import unquote, urlparse
+
+from core_ai.types import Content
+
+ContentPart = Dict[str, Any]
+
+IMAGE_TOKEN_ESTIMATE = 768
+IMAGE_MIME_BY_SUFFIX = {
+    ".png": "image/png",
+    ".jpg": "image/jpeg",
+    ".jpeg": "image/jpeg",
+    ".gif": "image/gif",
+    ".webp": "image/webp",
+    ".bmp": "image/bmp",
+    ".tif": "image/tiff",
+    ".tiff": "image/tiff",
+}
+_DATA_URL_RE = re.compile(
+    r"^data:(?P<media>image/[A-Za-z0-9.+-]+);base64,(?P<data>[A-Za-z0-9+/=\s]+)$",
+    re.IGNORECASE,
+)
+_MAGIC_TYPES: tuple[tuple[bytes, str], ...] = (
+    (b"\x89PNG\r\n\x1a\n", "image/png"),
+    (b"\xff\xd8\xff", "image/jpeg"),
+    (b"GIF87a", "image/gif"),
+    (b"GIF89a", "image/gif"),
+    (b"BM", "image/bmp"),
+)
+
+
+def text_from_content(content: Optional[Content]) -> str:
+    """Return the human-readable text from a message body."""
+    if content is None:
+        return ""
+    if isinstance(content, str):
+        return content
+    chunks: List[str] = []
+    for part in _iter_parts(content):
+        kind = _part_type(part)
+        if kind == "text":
+            text = str(part.get("text") or "")
+            if text:
+                chunks.append(text)
+        elif kind == "image":
+            chunks.append(f"[image:{_image_filename(part)}]")
+    return "\n".join(chunks)
+
+
+def normalize_content(content: Optional[Content]) -> List[ContentPart]:
+    """Expand a message body into canonical text/image parts."""
+    if content is None or content == "":
+        return []
+    if isinstance(content, str):
+        return [{"type": "text", "text": content}]
+    parts: List[ContentPart] = []
+    for item in content:
+        part = _normalize_part(item)
+        if part is not None:
+            parts.append(part)
+    return parts
+
+
+def to_openai_chat_content(content: Optional[Content]) -> Content:
+    """Chat Completions content: string or text / image_url parts."""
+    if isinstance(content, str) or content is None:
+        return content or ""
+    parts: List[ContentPart] = []
+    for part in normalize_content(content):
+        if part["type"] == "text":
+            parts.append({"type": "text", "text": part.get("text") or ""})
+            continue
+        url = _openai_image_url(part)
+        if url:
+            parts.append({"type": "image_url", "image_url": {"url": url}})
+    return parts
+
+
+def to_openai_responses_content(
+    content: Optional[Content],
+    *,
+    role: str = "user",
+    typed_parts: bool = False,
+) -> Content:
+    """Responses API content: string, or input_text / input_image parts.
+
+    Codex rejects bare strings, so callers on that path pass `typed_parts=True`.
+    """
+    text_type = "output_text" if role == "assistant" else "input_text"
+    if isinstance(content, str) or content is None:
+        text = content or ""
+        if typed_parts:
+            return [{"type": text_type, "text": text}] if text else []
+        return text
+    parts: List[ContentPart] = []
+    for part in normalize_content(content):
+        if part["type"] == "text":
+            parts.append({"type": text_type, "text": part.get("text") or ""})
+            continue
+        url = _openai_image_url(part)
+        if url:
+            parts.append({"type": "input_image", "image_url": url})
+    return parts
+
+
+def to_anthropic_blocks(content: Optional[Content]) -> List[ContentPart]:
+    """Anthropic Messages content blocks."""
+    blocks: List[ContentPart] = []
+    for part in normalize_content(content):
+        if part["type"] == "text":
+            text = part.get("text") or ""
+            if text:
+                blocks.append({"type": "text", "text": text})
+            continue
+        source = _anthropic_image_source(part)
+        if source is not None:
+            blocks.append({"type": "image", "source": source})
+    return blocks
+
+
+def to_gemini_parts(content: Optional[Content]) -> List[ContentPart]:
+    """Gemini generateContent parts."""
+    parts: List[ContentPart] = []
+    for part in normalize_content(content):
+        if part["type"] == "text":
+            text = part.get("text") or ""
+            if text:
+                parts.append({"text": text})
+            continue
+        media_type = str(part.get("media_type") or "image/png")
+        data = part.get("data")
+        url = str(part.get("url") or "")
+        if data:
+            parts.append({"inlineData": {"mimeType": media_type, "data": data}})
+        elif url.startswith(("http://", "https://")):
+            parts.append({"fileData": {"mimeType": media_type, "fileUri": url}})
+        elif url:
+            parsed = _parse_data_url(url)
+            if parsed is not None:
+                parts.append(
+                    {
+                        "inlineData": {
+                            "mimeType": parsed[0],
+                            "data": parsed[1],
+                        }
+                    }
+                )
+    return parts
+
+
+def image_part(
+    *,
+    media_type: str,
+    data: Optional[str] = None,
+    url: Optional[str] = None,
+    filename: Optional[str] = None,
+) -> ContentPart:
+    part: ContentPart = {"type": "image", "media_type": media_type}
+    if data:
+        part["data"] = data
+    if url:
+        part["url"] = url
+    if filename:
+        part["filename"] = filename
+    return part
+
+
+def image_part_from_bytes(
+    payload: bytes,
+    *,
+    media_type: str,
+    filename: Optional[str] = None,
+) -> ContentPart:
+    return image_part(
+        media_type=media_type,
+        data=base64.b64encode(payload).decode("ascii"),
+        filename=filename,
+    )
+
+
+def sniff_image_media_type(data: bytes, *, filename: str = "") -> Optional[str]:
+    """Return an image media type from a filename suffix or magic bytes."""
+    name = filename.rsplit("/", 1)[-1].rsplit("\\", 1)[-1]
+    suffix = ""
+    if "." in name:
+        suffix = "." + name.rsplit(".", 1)[-1].lower()
+    if suffix in IMAGE_MIME_BY_SUFFIX:
+        return IMAGE_MIME_BY_SUFFIX[suffix]
+    if data.startswith(b"RIFF") and data[8:12] == b"WEBP":
+        return "image/webp"
+    if data[:4] in {b"II*\x00", b"MM\x00*"}:
+        return "image/tiff"
+    for magic, media_type in _MAGIC_TYPES:
+        if data.startswith(magic):
+            return media_type
+    return None
+
+
+def split_text_and_images(content: Optional[Content]) -> tuple[str, List[ContentPart]]:
+    """Split a message/tool body into text and canonical image parts."""
+    if content is None:
+        return "", []
+    if isinstance(content, str):
+        return content, []
+    texts: List[str] = []
+    images: List[ContentPart] = []
+    for part in normalize_content(content):
+        if part["type"] == "text":
+            text = str(part.get("text") or "")
+            if text:
+                texts.append(text)
+        elif part["type"] == "image":
+            images.append(part)
+    text = "\n".join(texts)
+    if not text and images:
+        text = text_from_content(images)
+    return text, images
+
+
+def estimate_content_tokens(content: Optional[Content]) -> int:
+    """Heuristic tokens for a message body, without dumping image bytes."""
+    if content is None:
+        return 0
+    if isinstance(content, str):
+        return _text_tokens(content)
+    total = 0
+    for part in normalize_content(content):
+        if part["type"] == "image":
+            total += IMAGE_TOKEN_ESTIMATE
+        else:
+            total += _text_tokens(str(part.get("text") or ""))
+    return total
+
+
+def _text_tokens(text: str) -> int:
+    if not text:
+        return 0
+    return max(1, (len(text) + 3) // 4)
+
+
+def _iter_parts(content: Sequence[Any]) -> Iterable[Dict[str, Any]]:
+    for item in content:
+        if isinstance(item, dict):
+            yield item
+
+
+def _part_type(part: Dict[str, Any]) -> str:
+    declared = str(part.get("type") or "")
+    if declared in {"text", "input_text", "output_text"}:
+        return "text"
+    if declared in {"image", "image_url", "input_image"}:
+        return "image"
+    if "inlineData" in part or "fileData" in part or "source" in part:
+        return "image"
+    if "text" in part:
+        return "text"
+    return declared or "unknown"
+
+
+def _normalize_part(item: Any) -> Optional[ContentPart]:
+    if not isinstance(item, dict):
+        if item is None or item == "":
+            return None
+        return {"type": "text", "text": str(item)}
+
+    kind = _part_type(item)
+    if kind == "text":
+        text = item.get("text")
+        return {"type": "text", "text": "" if text is None else str(text)}
+
+    if kind != "image":
+        return None
+
+    media_type = str(item.get("media_type") or item.get("mime_type") or "")
+    data = item.get("data")
+    url = item.get("url")
+    filename = item.get("filename") or item.get("name")
+
+    nested = item.get("image_url")
+    if isinstance(nested, dict):
+        url = nested.get("url") or url
+    elif isinstance(nested, str) and nested:
+        url = nested
+
+    source = item.get("source")
+    if isinstance(source, dict):
+        media_type = str(source.get("media_type") or media_type)
+        if source.get("type") == "base64" and source.get("data"):
+            data = source.get("data")
+        elif source.get("url"):
+            url = source.get("url")
+
+    inline = item.get("inlineData") or item.get("inline_data")
+    if isinstance(inline, dict):
+        media_type = str(inline.get("mimeType") or inline.get("mime_type") or media_type)
+        data = inline.get("data") or data
+
+    file_data = item.get("fileData") or item.get("file_data")
+    if isinstance(file_data, dict):
+        media_type = str(file_data.get("mimeType") or file_data.get("mime_type") or media_type)
+        url = file_data.get("fileUri") or file_data.get("file_uri") or url
+
+    if isinstance(url, str):
+        parsed = _parse_data_url(url)
+        if parsed is not None:
+            media_type = media_type or parsed[0]
+            data = data or parsed[1]
+            url = None
+
+    if not media_type:
+        media_type = "image/png"
+
+    part = image_part(
+        media_type=media_type,
+        data=str(data) if data else None,
+        url=str(url) if url else None,
+        filename=str(filename) if filename else None,
+    )
+    if "data" not in part and "url" not in part:
+        return None
+    return part
+
+
+def _image_filename(part: Dict[str, Any]) -> str:
+    name = part.get("filename")
+    if name:
+        return str(name)
+    url = str(part.get("url") or "")
+    if url:
+        path = unquote(urlparse(url).path)
+        leaf = path.rsplit("/", 1)[-1]
+        if leaf:
+            return leaf
+    media = str(part.get("media_type") or "image")
+    suffix = media.split("/", 1)[-1] or "png"
+    return f"image.{suffix}"
+
+
+def _openai_image_url(part: Dict[str, Any]) -> Optional[str]:
+    data = part.get("data")
+    media_type = str(part.get("media_type") or "image/png")
+    if data:
+        return f"data:{media_type};base64,{data}"
+    url = str(part.get("url") or "")
+    return url or None
+
+
+def _anthropic_image_source(part: Dict[str, Any]) -> Optional[Dict[str, Any]]:
+    data = part.get("data")
+    media_type = str(part.get("media_type") or "image/png")
+    if data:
+        return {"type": "base64", "media_type": media_type, "data": data}
+    url = str(part.get("url") or "")
+    if url.startswith(("http://", "https://")):
+        return {"type": "url", "url": url}
+    parsed = _parse_data_url(url) if url else None
+    if parsed is not None:
+        return {"type": "base64", "media_type": parsed[0], "data": parsed[1]}
+    return None
+
+
+def _parse_data_url(value: str) -> Optional[tuple[str, str]]:
+    match = _DATA_URL_RE.match(value.strip())
+    if match is None:
+        return None
+    data = re.sub(r"\s+", "", match.group("data"))
+    return match.group("media").lower(), data

core_ai/models/__init__.py

@@ -0,0 +1,18 @@
+from core_ai.models.registry import (
+    ModelCatalog,
+    get_model,
+    list_models,
+    register_model,
+    unregister_model,
+)
+from core_ai.models.types import ModelAPI, ModelInfo
+
+__all__ = [
+    "ModelAPI",
+    "ModelCatalog",
+    "ModelInfo",
+    "get_model",
+    "list_models",
+    "register_model",
+    "unregister_model",
+]

core_ai/models/generated.py

@@ -0,0 +1,110 @@
+# Generated by scripts/generate_models.py. Do not edit by hand.
+
+from core_ai.models.types import ModelInfo
+
+OPENAI_MODELS = (
+    ModelInfo(id="gpt-4", provider="openai", api="responses", context_limit=8192),
+    ModelInfo(id="gpt-4-turbo", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-4.1", provider="openai", api="responses", context_limit=1047576),
+    ModelInfo(id="gpt-4.1-mini", provider="openai", api="responses", context_limit=1047576),
+    ModelInfo(id="gpt-4.1-nano", provider="openai", api="responses", context_limit=1047576),
+    ModelInfo(id="gpt-4o", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-4o-2024-05-13", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-4o-2024-08-06", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-4o-2024-11-20", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-4o-mini", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-5", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5-mini", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5-nano", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5-pro", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', None), ('medium', None), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.1", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.2", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.2-chat-latest", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', None), ('medium', 'medium'), ('high', None), ('xhigh', None), ('max', None)), context_limit=128000),
+    ModelInfo(id="gpt-5.2-pro", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', None), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.3-chat-latest", provider="openai", api="responses", context_limit=128000),
+    ModelInfo(id="gpt-5.3-codex", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.3-codex-spark", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=128000),
+    ModelInfo(id="gpt-5.4", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=1050000),
+    ModelInfo(id="gpt-5.4-mini", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.4-nano", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=400000),
+    ModelInfo(id="gpt-5.4-pro", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', None), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=1050000),
+    ModelInfo(id="gpt-5.5", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=1050000),
+    ModelInfo(id="gpt-5.5-pro", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', None), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=1050000),
+    ModelInfo(id="gpt-5.6", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-5.6-luna", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-5.6-sol", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-5.6-terra", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-6-astra", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-6-luna", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-6-sol", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-6.1-sol", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-daybreak-blue-latest", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1050000),
+    ModelInfo(id="gpt-daybreak-red-latest", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=400000),
+    ModelInfo(id="gpt-image-1", provider="openai", api="responses", context_limit=0),
+    ModelInfo(id="gpt-image-1-mini", provider="openai", api="responses", context_limit=0),
+    ModelInfo(id="gpt-image-1.5", provider="openai", api="responses", context_limit=0),
+    ModelInfo(id="gpt-image-2", provider="openai", api="responses", context_limit=0),
+    ModelInfo(id="gpt-realtime-2.1", provider="openai", api="responses", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=128000),
+    ModelInfo(id="o1", provider="openai", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="o1-pro", provider="openai", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="o3", provider="openai", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="o3-mini", provider="openai", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="o3-pro", provider="openai", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="o4-mini", provider="openai", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+)
+
+ANTHROPIC_MODELS = (
+    ModelInfo(id="claude-fable-5", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-fable-5-1", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-haiku-4-5", provider="anthropic", api="messages", reasoning=True, context_limit=200000),
+    ModelInfo(id="claude-haiku-4-5-20251001", provider="anthropic", api="messages", reasoning=True, context_limit=200000),
+    ModelInfo(id="claude-opus-4-5", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="claude-opus-4-5-20251101", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=200000),
+    ModelInfo(id="claude-opus-4-6", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-opus-4-7", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-opus-4-8", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-opus-5", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-opus-5-5", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-sonnet-4-5", provider="anthropic", api="messages", reasoning=True, context_limit=1000000),
+    ModelInfo(id="claude-sonnet-4-5-20250929", provider="anthropic", api="messages", reasoning=True, context_limit=1000000),
+    ModelInfo(id="claude-sonnet-4-6", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-sonnet-5", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+    ModelInfo(id="claude-sonnet-5-5", provider="anthropic", api="messages", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', 'max')), context_limit=1000000),
+)
+
+GEMINI_MODELS = (
+    ModelInfo(id="deep-research-max-preview-04-2026", provider="gemini", api="generate_content", reasoning=True, context_limit=131072),
+    ModelInfo(id="deep-research-preview-04-2026", provider="gemini", api="generate_content", reasoning=True, context_limit=131072),
+    ModelInfo(id="gemini-2.5-computer-use-preview-10-2025", provider="gemini", api="generate_content", reasoning=True, context_limit=128000),
+    ModelInfo(id="gemini-2.5-flash", provider="gemini", api="generate_content", reasoning=True, context_limit=1048576),
+    ModelInfo(id="gemini-2.5-flash-lite", provider="gemini", api="generate_content", reasoning=True, context_limit=1048576),
+    ModelInfo(id="gemini-2.5-pro", provider="gemini", api="generate_content", reasoning=True, context_limit=1048576),
+    ModelInfo(id="gemini-3-flash-preview", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.1-flash-lite", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.1-flash-lite-image", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', None), ('medium', None), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=65536),
+    ModelInfo(id="gemini-3.1-flash-lite-preview", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.1-flash-live-preview", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=131072),
+    ModelInfo(id="gemini-3.1-pro-preview", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.1-pro-preview-customtools", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.5-flash", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.5-flash-lite", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.6-flash", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.7-flash", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-3.8-flash", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-flash-latest", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemini-flash-lite-latest", provider="gemini", api="generate_content", reasoning=True, thinking_level_map=(('off', None), ('minimal', 'minimal'), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1048576),
+    ModelInfo(id="gemma-4-26b-a4b-it", provider="gemini", api="generate_content", reasoning=True, context_limit=262144),
+    ModelInfo(id="gemma-4-31b-it", provider="gemini", api="generate_content", reasoning=True, context_limit=262144),
+)
+
+GROK_MODELS = (
+    ModelInfo(id="grok-4.20-0309-non-reasoning", provider="grok", api="chat_completions", context_limit=1000000),
+    ModelInfo(id="grok-4.20-0309-reasoning", provider="grok", api="chat_completions", reasoning=True, context_limit=1000000),
+    ModelInfo(id="grok-4.3", provider="grok", api="chat_completions", reasoning=True, thinking_level_map=(('off', 'none'), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=1000000),
+    ModelInfo(id="grok-4.5", provider="grok", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', None), ('max', None)), context_limit=500000),
+    ModelInfo(id="grok-4.6", provider="grok", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=500000),
+    ModelInfo(id="grok-4.7", provider="grok", api="chat_completions", reasoning=True, thinking_level_map=(('off', None), ('minimal', None), ('low', 'low'), ('medium', 'medium'), ('high', 'high'), ('xhigh', 'xhigh'), ('max', None)), context_limit=500000),
+    ModelInfo(id="grok-build-0.1", provider="grok", api="chat_completions", reasoning=True, context_limit=256000),
+)
+
+ALL_MODELS = OPENAI_MODELS + ANTHROPIC_MODELS + GEMINI_MODELS + GROK_MODELS