mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-20 04:44:46 +08:00
feat: add config modules
This commit is contained in:
26
backend/src/config/model_config.py
Normal file
26
backend/src/config/model_config.py
Normal 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",
|
||||
)
|
||||
Reference in New Issue
Block a user