mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-12 14:18:18 +08:00
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, field_validator
|
|
|
|
from .types import SegmentType
|
|
|
|
|
|
class Segment(BaseModel):
|
|
model_config = ConfigDict(frozen=True)
|
|
|
|
value_type: SegmentType
|
|
value: Any
|
|
|
|
@field_validator('value_type')
|
|
def validate_value_type(cls, value):
|
|
"""
|
|
This validator checks if the provided value is equal to the default value of the 'value_type' field.
|
|
If the value is different, a ValueError is raised.
|
|
"""
|
|
if value != cls.model_fields['value_type'].default:
|
|
raise ValueError("Cannot modify 'value_type'")
|
|
return value
|
|
|
|
@property
|
|
def text(self) -> str:
|
|
return str(self.value)
|
|
|
|
@property
|
|
def log(self) -> str:
|
|
return str(self.value)
|
|
|
|
@property
|
|
def markdown(self) -> str:
|
|
return str(self.value)
|
|
|
|
def to_object(self) -> Any:
|
|
if isinstance(self.value, Segment):
|
|
return self.value.to_object()
|
|
if isinstance(self.value, list):
|
|
return [v.to_object() for v in self.value]
|
|
if isinstance(self.value, dict):
|
|
return {k: v.to_object() for k, v in self.value.items()}
|
|
return self.value
|
|
|
|
|
|
class StringSegment(Segment):
|
|
value_type: SegmentType = SegmentType.STRING
|
|
value: str
|