refactor: code cleanup

providers/shared/model_capabilities.py

@@ -0,0 +1,34 @@
+"""Dataclass describing the feature set of a model exposed by a provider."""
+
+from dataclasses import dataclass, field
+
+from .provider_type import ProviderType
+from .temperature import RangeTemperatureConstraint, TemperatureConstraint
+
+__all__ = ["ModelCapabilities"]
+
+
+@dataclass
+class ModelCapabilities:
+    """Static capabilities and constraints for a provider-managed model."""
+
+    provider: ProviderType
+    model_name: str
+    friendly_name: str
+    context_window: int
+    max_output_tokens: int
+    supports_extended_thinking: bool = False
+    supports_system_prompts: bool = True
+    supports_streaming: bool = True
+    supports_function_calling: bool = False
+    supports_images: bool = False
+    max_image_size_mb: float = 0.0
+    supports_temperature: bool = True
+    description: str = ""
+    aliases: list[str] = field(default_factory=list)
+    supports_json_mode: bool = False
+    max_thinking_tokens: int = 0
+    is_custom: bool = False
+    temperature_constraint: TemperatureConstraint = field(
+        default_factory=lambda: RangeTemperatureConstraint(0.0, 2.0, 0.3)
+    )