feat: add config modules

This commit is contained in:
Henry Li
2026-01-14 07:15:58 +08:00
parent c2a62a2266
commit 88ed3841c7
5 changed files with 202 additions and 0 deletions

View File

@@ -0,0 +1,26 @@
from pydantic import BaseModel, ConfigDict, Field
class ModelConfig(BaseModel):
"""Config section for a model"""
name: str = Field(..., description="Unique name for the model")
display_name: str | None = Field(
..., default_factory=lambda: None, description="Display name for the model"
)
description: str | None = Field(
..., default_factory=lambda: None, description="Description for the model"
)
use: str = Field(
...,
description="Class path of the model provider(e.g. langchain_openai.ChatOpenAI)",
)
model: str = Field(..., description="Model name")
model_config = ConfigDict(extra="allow")
supports_thinking: bool = Field(
default_factory=lambda: False, description="Whether the model supports thinking"
)
when_thinking_enabled: dict | None = Field(
default_factory=lambda: None,
description="Extra settings to be passed to the model when thinking is enabled",
)