mirror of
https://github.com/langgenius/dify.git
synced 2026-03-27 01:00:13 +08:00
56593f20b0
Signed-off-by: -LAN- <laipz8200@outlook.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: WH-2099 <wh2099@pm.me>
106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from enum import StrEnum, auto
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field
|
|
|
|
|
|
class _ToolRuntimeModel(BaseModel):
|
|
model_config = ConfigDict(extra="forbid")
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ToolRuntimeHandle:
|
|
"""Opaque graph-owned handle for a workflow-layer tool runtime.
|
|
|
|
Workflow-specific execution context must stay behind `raw` so the graph
|
|
contract does not absorb application-owned concepts.
|
|
"""
|
|
|
|
raw: object
|
|
|
|
|
|
@dataclass(frozen=True, slots=True)
|
|
class ToolRuntimeParameter:
|
|
"""Graph-owned parameter shape used by tool nodes."""
|
|
|
|
name: str
|
|
required: bool = False
|
|
|
|
|
|
class ToolRuntimeMessage(_ToolRuntimeModel):
|
|
"""Graph-owned tool invocation message DTO."""
|
|
|
|
class TextMessage(_ToolRuntimeModel):
|
|
text: str
|
|
|
|
class JsonMessage(_ToolRuntimeModel):
|
|
json_object: dict[str, Any] | list[Any]
|
|
suppress_output: bool = Field(default=False)
|
|
|
|
class BlobMessage(_ToolRuntimeModel):
|
|
blob: bytes
|
|
|
|
class BlobChunkMessage(_ToolRuntimeModel):
|
|
id: str
|
|
sequence: int
|
|
total_length: int
|
|
blob: bytes
|
|
end: bool
|
|
|
|
class FileMessage(_ToolRuntimeModel):
|
|
file_marker: str = Field(default="file_marker")
|
|
|
|
class VariableMessage(_ToolRuntimeModel):
|
|
variable_name: str
|
|
variable_value: dict[str, Any] | list[Any] | str | int | float | bool | None
|
|
stream: bool = Field(default=False)
|
|
|
|
class LogMessage(_ToolRuntimeModel):
|
|
class LogStatus(StrEnum):
|
|
START = auto()
|
|
ERROR = auto()
|
|
SUCCESS = auto()
|
|
|
|
id: str
|
|
label: str
|
|
parent_id: str | None = None
|
|
error: str | None = None
|
|
status: LogStatus
|
|
data: dict[str, Any]
|
|
metadata: dict[str, Any] = Field(default_factory=dict)
|
|
|
|
class RetrieverResourceMessage(_ToolRuntimeModel):
|
|
retriever_resources: list[dict[str, Any]]
|
|
context: str
|
|
|
|
class MessageType(StrEnum):
|
|
TEXT = auto()
|
|
IMAGE = auto()
|
|
LINK = auto()
|
|
BLOB = auto()
|
|
JSON = auto()
|
|
IMAGE_LINK = auto()
|
|
BINARY_LINK = auto()
|
|
VARIABLE = auto()
|
|
FILE = auto()
|
|
LOG = auto()
|
|
BLOB_CHUNK = auto()
|
|
RETRIEVER_RESOURCES = auto()
|
|
|
|
type: MessageType = MessageType.TEXT
|
|
message: (
|
|
JsonMessage
|
|
| TextMessage
|
|
| BlobChunkMessage
|
|
| BlobMessage
|
|
| LogMessage
|
|
| FileMessage
|
|
| None
|
|
| VariableMessage
|
|
| RetrieverResourceMessage
|
|
)
|
|
meta: dict[str, Any] | None = None
|