This commit is contained in:
Yeuoly
2024-09-11 15:12:37 +08:00
parent 99d3a46fd7
commit 70db8bcb8e
25 changed files with 51 additions and 52 deletions
@@ -7,7 +7,7 @@ from dify_plugin.entities.tool import ToolInvokeMessage
from dify_plugin.entities.model.message import SystemPromptMessage, UserPromptMessage
class LLMTool(Tool):
def _invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:
def invoke(self, tool_parameters: dict[str, Any]) -> Generator[ToolInvokeMessage]:
response = self.session.model.llm.invoke(
model_config=LLMModelConfig(
provider='openai',
@@ -4,5 +4,5 @@ from dify_plugin import Tool
class ParameterExtractorTool(Tool):
def _invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
def invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
return super()._invoke(tool_parameters)
@@ -4,5 +4,5 @@ from dify_plugin import Tool
class QuestionClassifierTool(Tool):
def _invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
def invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
return super()._invoke(tool_parameters)
@@ -4,5 +4,5 @@ from dify_plugin import Tool
class ToolTool(Tool):
def _invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
return super()._invoke(tool_parameters)
def invoke(self, tool_parameters: dict) -> Generator[ToolInvokeMessage, None, None]:
return super().invoke(tool_parameters)