mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-23 02:45:37 -04:00
Yeuoly
503db4ced5
- Introduced ParameterOption class for parameter options with value, label, and optional icon. - Updated ToolParameterOption to inherit from ParameterOption. - Added DYNAMIC_SELECT to CommonParameterType enum. - Implemented fetch_parameter_options method in Tool class for fetching parameter options. - Created DynamicSelectProtocol for dynamic select functionality. - Added tests for Tool construction and parameter options fetching.
56 lines
2.0 KiB
Python
56 lines
2.0 KiB
Python
from collections.abc import Generator, Mapping
|
|
from concurrent.futures import ThreadPoolExecutor
|
|
|
|
from dify_plugin.core.runtime import Session
|
|
from dify_plugin.core.server.stdio.request_reader import StdioRequestReader
|
|
from dify_plugin.core.server.stdio.response_writer import StdioResponseWriter
|
|
from dify_plugin.entities import I18nObject, ParameterOption
|
|
from dify_plugin.entities.tool import ToolInvokeMessage, ToolRuntime
|
|
from dify_plugin.interfaces.tool import Tool
|
|
|
|
|
|
def test_construct_tool():
|
|
"""
|
|
Test the constructor of Tool
|
|
NOTE:
|
|
- This test is to ensure that the constructor of Tool is not overridden.
|
|
- And ensure a breaking change will be detected by CI.
|
|
"""
|
|
|
|
class ToolImpl(Tool):
|
|
def _invoke(self, tool_parameters: Mapping) -> Generator[ToolInvokeMessage, None, None]:
|
|
yield self.create_text_message("Hello, world!")
|
|
|
|
session = Session(
|
|
session_id="test",
|
|
executor=ThreadPoolExecutor(max_workers=1),
|
|
reader=StdioRequestReader(),
|
|
writer=StdioResponseWriter(),
|
|
)
|
|
|
|
tool = ToolImpl(runtime=ToolRuntime(credentials={}, user_id="test", session_id="test"), session=session)
|
|
assert tool is not None
|
|
|
|
|
|
def test_fetch_parameter_options():
|
|
"""
|
|
Test that the Tool can fetch the parameter options
|
|
"""
|
|
|
|
class ToolImpl(Tool):
|
|
def _invoke(self, tool_parameters: Mapping) -> Generator[ToolInvokeMessage, None, None]:
|
|
yield self.create_text_message("Hello, world!")
|
|
|
|
def _fetch_parameter_options(self, parameter: str) -> list[ParameterOption]:
|
|
return [ParameterOption(value="test", label=I18nObject(en_US="test"))]
|
|
|
|
session = Session(
|
|
session_id="test",
|
|
executor=ThreadPoolExecutor(max_workers=1),
|
|
reader=StdioRequestReader(),
|
|
writer=StdioResponseWriter(),
|
|
)
|
|
|
|
tool = ToolImpl(runtime=ToolRuntime(credentials={}, user_id="test", session_id="test"), session=session)
|
|
assert tool.fetch_parameter_options("test") == [ParameterOption(value="test", label=I18nObject(en_US="test"))]
|