mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 18:35:29 -04:00
Yeuoly
117 lines
4.2 KiB
Python
117 lines
4.2 KiB
Python
from enum import Enum
|
|
from typing import Any, Mapping, Optional, Union
|
|
from pydantic import BaseModel, Field, field_validator
|
|
|
|
from dify_plugin.core.utils.yaml_loader import load_yaml_file
|
|
from dify_plugin.entities import I18nObject
|
|
from dify_plugin.entities.tool import (
|
|
CommonParameterType,
|
|
ParameterAutoGenerate,
|
|
ParameterTemplate,
|
|
ToolIdentity,
|
|
ToolInvokeMessage,
|
|
ToolParameterOption,
|
|
ToolProviderIdentity,
|
|
)
|
|
|
|
|
|
class AgentStrategyProviderIdentity(ToolProviderIdentity):
|
|
pass
|
|
|
|
|
|
class AgentStrategyIdentity(ToolIdentity):
|
|
pass
|
|
|
|
|
|
class AgentStrategyParameter(BaseModel):
|
|
class ToolParameterType(str, Enum):
|
|
STRING = CommonParameterType.STRING.value
|
|
NUMBER = CommonParameterType.NUMBER.value
|
|
BOOLEAN = CommonParameterType.BOOLEAN.value
|
|
SELECT = CommonParameterType.SELECT.value
|
|
SECRET_INPUT = CommonParameterType.SECRET_INPUT.value
|
|
FILE = CommonParameterType.FILE.value
|
|
FILES = CommonParameterType.FILES.value
|
|
MODEL_SELECTOR = CommonParameterType.MODEL_SELECTOR.value
|
|
APP_SELECTOR = CommonParameterType.APP_SELECTOR.value
|
|
TOOLS_SELECTOR = CommonParameterType.TOOLS_SELECTOR.value
|
|
# TOOL_SELECTOR = CommonParameterType.TOOL_SELECTOR.value
|
|
|
|
name: str = Field(..., description="The name of the parameter")
|
|
label: I18nObject = Field(..., description="The label presented to the user")
|
|
type: ToolParameterType = Field(..., description="The type of the parameter")
|
|
auto_generate: Optional[ParameterAutoGenerate] = Field(
|
|
default=None, description="The auto generate of the parameter"
|
|
)
|
|
template: Optional[ParameterTemplate] = Field(default=None, description="The template of the parameter")
|
|
scope: str | None = None
|
|
required: Optional[bool] = False
|
|
default: Optional[Union[int, float, str]] = None
|
|
min: Optional[Union[float, int]] = None
|
|
max: Optional[Union[float, int]] = None
|
|
precision: Optional[int] = None
|
|
options: Optional[list[ToolParameterOption]] = None
|
|
|
|
|
|
class AgentStrategyConfigurationExtra(BaseModel):
|
|
class Python(BaseModel):
|
|
source: str
|
|
|
|
python: Python
|
|
|
|
|
|
class AgentStrategyConfiguration(BaseModel):
|
|
identity: AgentStrategyIdentity
|
|
parameters: list[AgentStrategyParameter] = Field(default=[], description="The parameters of the agent")
|
|
description: I18nObject
|
|
extra: AgentStrategyConfigurationExtra
|
|
has_runtime_parameters: bool = Field(default=False, description="Whether the tool has runtime parameters")
|
|
output_schema: Optional[Mapping[str, Any]] = None
|
|
|
|
|
|
class AgentProviderConfigurationExtra(BaseModel):
|
|
class Python(BaseModel):
|
|
source: str
|
|
|
|
python: Python
|
|
|
|
|
|
class AgentStrategyProviderConfiguration(BaseModel):
|
|
identity: AgentStrategyProviderIdentity
|
|
strategies: list[AgentStrategyConfiguration] = Field(default=[], description="The strategies of the agent provider")
|
|
|
|
@field_validator("strategies", mode="before")
|
|
@classmethod
|
|
def validate_strategies(cls, value) -> list[AgentStrategyConfiguration]:
|
|
if not isinstance(value, list):
|
|
raise ValueError("strategies should be a list")
|
|
|
|
strategies: list[AgentStrategyConfiguration] = []
|
|
|
|
for strategy in value:
|
|
# read from yaml
|
|
if not isinstance(strategy, str):
|
|
raise ValueError("strategy path should be a string")
|
|
try:
|
|
file = load_yaml_file(strategy)
|
|
strategies.append(
|
|
AgentStrategyConfiguration(
|
|
**{
|
|
"identity": AgentStrategyIdentity(**file["identity"]),
|
|
"parameters": [
|
|
AgentStrategyParameter(**param) for param in file.get("parameters", []) or []
|
|
],
|
|
"description": I18nObject(**file["description"]),
|
|
"extra": AgentStrategyConfigurationExtra(**file.get("extra", {})),
|
|
}
|
|
)
|
|
)
|
|
except Exception as e:
|
|
raise ValueError(f"Error loading agent strategy configuration: {str(e)}") from e
|
|
|
|
return strategies
|
|
|
|
|
|
class AgentInvokeMessage(ToolInvokeMessage):
|
|
pass
|