소스 검색

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

tags/0.10.1
-LAN- 1 년 전
부모
커밋
5838345f48
No account linked to committer's email address
1개의 변경된 파일8개의 추가작업 그리고 1개의 파일을 삭제
  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

Loading…
취소
저장