hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-11 03:46:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(entities): add validator for VisionConfig to handle None values (#9598)

Browse Source
This commit is contained in:
-LAN-
2024-10-22 10:49:03 +08:00
committed by GitHub
parent 3f1c84f65a
commit 5838345f48
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
api/core/workflow/nodes/llm/entities.py
View File

@@ -1,7 +1,7 @@
from collections.abc import Sequence from collections.abc import Sequence
from typing import Any, Optional 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.model_runtime.entities import ImagePromptMessageContent
from core.prompt.entities.advanced_prompt_entities import ChatModelMessage, CompletionModelPromptTemplate, MemoryConfig from core.prompt.entities.advanced_prompt_entities import ChatModelMessage, CompletionModelPromptTemplate, MemoryConfig
@@ -30,6 +30,13 @@ class VisionConfig(BaseModel):
enabled: bool = False enabled: bool = False
configs: VisionConfigOptions = Field(default_factory=VisionConfigOptions) 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): class PromptConfig(BaseModel):
jinja2_variables: Optional[list[VariableSelector]] = None jinja2_variables: Optional[list[VariableSelector]] = None