35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""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)
|
|
)
|