mirror of
https://gitee.com/wanwujie/deer-flow
synced 2026-04-03 06:12:14 +08:00
7414947cc6
Add image viewing capability for vision-enabled models with ViewImageMiddleware and view_image_tool. Limit web_fetch tool output to 4096 characters to prevent excessive content. Update model config to support vision capability flag. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
22 lines
1.0 KiB
Python
22 lines
1.0 KiB
Python
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",
|
|
)
|
|
supports_vision: bool = Field(default_factory=lambda: False, description="Whether the model supports vision/image inputs")
|