Forráskód Böngészése

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

-LAN- 6 hónapja
szülő
commit
5838345f48
1 módosított fájl, 8 hozzáadás és 1 törlés
  1. 8 1
      api/core/workflow/nodes/llm/entities.py

+ 8 - 1
api/core/workflow/nodes/llm/entities.py

@@ -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