| @@ -1,7 +1,7 @@ | |||
| from collections.abc import Sequence | |||
| from typing import Any, Optional | |||
| from pydantic import BaseModel, Field | |||
| from pydantic import BaseModel, Field, field_validator | |||
| from core.model_runtime.entities import ImagePromptMessageContent | |||
| from core.prompt.entities.advanced_prompt_entities import ChatModelMessage, CompletionModelPromptTemplate, MemoryConfig | |||
| @@ -30,6 +30,13 @@ class VisionConfig(BaseModel): | |||
| enabled: bool = False | |||
| configs: VisionConfigOptions = Field(default_factory=VisionConfigOptions) | |||
| @field_validator("configs", mode="before") | |||
| @classmethod | |||
| def convert_none_configs(cls, v: Any): | |||
| if v is None: | |||
| return VisionConfigOptions() | |||
| return v | |||
| class PromptConfig(BaseModel): | |||
| jinja2_variables: Optional[list[VariableSelector]] = None | |||