mirror of
https://github.com/langgenius/dify.git
synced 2026-07-15 01:17:04 +08:00
0923ebaf88
Co-authored-by: yunlu.wen <yunlu.wen@dify.ai> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com> Co-authored-by: Yunlu Wen <wylswz@163.com> Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com> Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Yanli 盐粒 <yanli@dify.ai> Co-authored-by: 盐粒 Yanli <beautyyuyanli@gmail.com> Co-authored-by: zyssyz123 <916125788@qq.com> Co-authored-by: 盐粒 Yanli <mail@yanli.one>
60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""Request/response DTOs for the Agent tool inner invoke API."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import ClassVar, Literal
|
|
|
|
from pydantic import BaseModel, ConfigDict, Field, JsonValue
|
|
|
|
type AgentToolProviderType = Literal["plugin", "builtin", "api", "workflow", "mcp"]
|
|
|
|
|
|
class AgentToolInvokeCaller(BaseModel):
|
|
tenant_id: str
|
|
user_id: str
|
|
user_from: str
|
|
app_id: str
|
|
invoke_from: str
|
|
conversation_id: str | None = None
|
|
workflow_id: str | None = None
|
|
workflow_run_id: str | None = None
|
|
node_id: str | None = None
|
|
node_execution_id: str | None = None
|
|
agent_id: str | None = None
|
|
agent_config_version_id: str | None = None
|
|
|
|
model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
|
|
|
|
|
|
class AgentToolInvokeTarget(BaseModel):
|
|
"""One tool invocation payload.
|
|
|
|
`runtime_parameters` are API-prepared hidden/form/runtime values forwarded
|
|
into `ToolManager` runtime construction. `tool_parameters` are the live
|
|
LLM/user invocation arguments supplied at call time.
|
|
"""
|
|
|
|
provider_type: AgentToolProviderType
|
|
provider_id: str
|
|
tool_name: str
|
|
credential_id: str | None = None
|
|
tool_parameters: dict[str, JsonValue] = Field(default_factory=dict)
|
|
runtime_parameters: dict[str, JsonValue] = Field(default_factory=dict)
|
|
|
|
model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
|
|
|
|
|
|
class AgentToolInvokeRequest(BaseModel):
|
|
caller: AgentToolInvokeCaller
|
|
tool: AgentToolInvokeTarget
|
|
|
|
model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
|
|
|
|
|
|
class AgentToolInvokeResponse(BaseModel):
|
|
messages: list[dict[str, JsonValue]] = Field(default_factory=list)
|
|
observation: str
|
|
metadata: dict[str, JsonValue] = Field(default_factory=dict)
|
|
|
|
model_config: ClassVar[ConfigDict] = ConfigDict(extra="forbid")
|