refactor: invoke_from_executor -> invoke, invoke -> _invoke

This commit is contained in:
Yeuoly
2024-09-12 23:36:44 +08:00
parent 12935909a9
commit d8dd877c14
24 changed files with 55 additions and 55 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]:
return super().invoke(tool_parameters)
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]:
return super().invoke(tool_parameters)
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)