mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 18:35:29 -04:00
Yeuoly
4bc3d18364
- Introduced new `Trigger` and `TriggerProvider` classes for handling trigger operations. - Added `TriggerResponse`, `TriggerRuntime`, and `TriggerParameter` models to manage trigger data. - Implemented `DynamicSelectProtocol` for dynamic parameter options. - Updated `ToolParameterOption` to inherit from `ParameterOption`. - Enhanced `CommonParameterType` with `DYNAMIC_SELECT` option. - Added tests for `TriggerProvider` and `Trigger` to ensure proper functionality.
79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
from collections.abc import Mapping
|
|
from enum import StrEnum
|
|
from typing import Any, Optional, Union
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from dify_plugin.core.documentation.schema_doc import docs
|
|
from dify_plugin.entities import I18nObject, ParameterOption
|
|
from dify_plugin.entities.provider_config import CommonParameterType
|
|
from dify_plugin.entities.tool import ParameterAutoGenerate, ParameterTemplate
|
|
|
|
|
|
class TriggerRuntime(BaseModel):
|
|
credentials: dict[str, Any]
|
|
session_id: Optional[str]
|
|
|
|
|
|
@docs(
|
|
description="The response of the trigger",
|
|
)
|
|
class TriggerResponse(BaseModel):
|
|
"""
|
|
The response of the trigger
|
|
"""
|
|
|
|
variables: Mapping[str, Any] = Field(
|
|
..., description="The variables of the trigger, must have the same schema as defined in the YAML"
|
|
)
|
|
|
|
|
|
@docs(
|
|
description="The option of the trigger parameter",
|
|
)
|
|
class TriggerParameterOption(ParameterOption):
|
|
"""
|
|
The option of the trigger parameter
|
|
"""
|
|
|
|
|
|
@docs(
|
|
description="The type of the parameter",
|
|
)
|
|
class TriggerParameter(BaseModel):
|
|
"""
|
|
The parameter of the trigger
|
|
"""
|
|
|
|
class TriggerParameterType(StrEnum):
|
|
STRING = CommonParameterType.STRING.value
|
|
NUMBER = CommonParameterType.NUMBER.value
|
|
BOOLEAN = CommonParameterType.BOOLEAN.value
|
|
SELECT = CommonParameterType.SELECT.value
|
|
FILE = CommonParameterType.FILE.value
|
|
FILES = CommonParameterType.FILES.value
|
|
MODEL_SELECTOR = CommonParameterType.MODEL_SELECTOR.value
|
|
APP_SELECTOR = CommonParameterType.APP_SELECTOR.value
|
|
# TOOL_SELECTOR = CommonParameterType.TOOL_SELECTOR.value
|
|
# ANY = CommonParameterType.ANY.value
|
|
# MCP object and array type parameters
|
|
OBJECT = CommonParameterType.OBJECT.value
|
|
ARRAY = CommonParameterType.ARRAY.value
|
|
DYNAMIC_SELECT = CommonParameterType.DYNAMIC_SELECT.value
|
|
|
|
name: str = Field(..., description="The name of the parameter")
|
|
label: I18nObject = Field(..., description="The label presented to the user")
|
|
type: TriggerParameterType = 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[TriggerParameterOption]] = None
|
|
description: Optional[I18nObject] = None
|