mirror of
https://github.com/langgenius/dify-plugin-sdks.git
synced 2026-07-22 18:35:29 -04:00
Yeuoly
100 lines
4.1 KiB
Python
100 lines
4.1 KiB
Python
from collections.abc import AsyncGenerator
|
|
from typing import Any
|
|
import uuid
|
|
|
|
from dify_plugin.core.runtime.abstract.request import AbstractRequestInterface
|
|
from dify_plugin.core.runtime.entities.dify import ToolProviderType
|
|
from dify_plugin.core.runtime.entities.invocation import InvokeType
|
|
from dify_plugin.core.runtime.entities.model import LLMResultChunk, RerankResult, TextEmbeddingResult
|
|
from dify_plugin.core.runtime.entities.tool import ToolInvokeMessage
|
|
from dify_plugin.core.runtime.entities.workflow import KnowledgeRetrievalNodeData, NodeResponse, ParameterExtractorNodeData, QuestionClassifierNodeData
|
|
from dify_plugin.utils.io_writer import PluginOutputStream
|
|
|
|
class RequestInterface(AbstractRequestInterface):
|
|
session_id: str
|
|
|
|
def __init__(self, session_id: str) -> None:
|
|
self.session_id = session_id
|
|
|
|
def _generate_request_id(self):
|
|
return uuid.uuid4().hex
|
|
|
|
async def invoke_tool(self, provider_type: ToolProviderType, provider: str, tool_name: str,
|
|
parameters: dict[str, Any]) -> AsyncGenerator[ToolInvokeMessage, None]:
|
|
"""
|
|
Invoke tool
|
|
"""
|
|
request_id = self._generate_request_id()
|
|
|
|
PluginOutputStream.session_message(
|
|
session_id=self.session_id,
|
|
data=PluginOutputStream.stream_invoke_object(data={
|
|
'type': InvokeType.Tool.value,
|
|
'request_id': request_id,
|
|
'request': {
|
|
'provider_type': provider_type.value,
|
|
'provider': provider,
|
|
'tool': tool_name,
|
|
'parameters': parameters
|
|
}
|
|
}).to_dict()
|
|
)
|
|
|
|
yield ToolInvokeMessage(
|
|
type=ToolInvokeMessage.MessageType.TEXT,
|
|
message=ToolInvokeMessage.TextMessage(text='123')
|
|
)
|
|
|
|
async def invoke_builtin_tool(self, provider: str, tool_name: str, parameters: dict[str, Any]) -> AsyncGenerator[ToolInvokeMessage, None]:
|
|
"""
|
|
Invoke builtin tool
|
|
"""
|
|
return await super().invoke_builtin_tool(provider, tool_name, parameters)
|
|
|
|
async def invoke_workflow_tool(self, provider: str, tool_name: str, parameters: dict[str, Any]) -> AsyncGenerator[ToolInvokeMessage, None]:
|
|
"""
|
|
Invoke workflow tool
|
|
"""
|
|
return await super().invoke_workflow_tool(provider, tool_name, parameters)
|
|
|
|
async def invoke_api_tool(self, provider: str, tool_name: str, parameters: dict[str, Any]) -> AsyncGenerator[ToolInvokeMessage, None]:
|
|
"""
|
|
Invoke api tool
|
|
"""
|
|
return await super().invoke_api_tool(provider, tool_name, parameters)
|
|
|
|
async def invoke_llm(self, provider: str, model_name: str, parameters: dict[str, Any]) -> AsyncGenerator[LLMResultChunk, None]:
|
|
"""
|
|
Invoke llm
|
|
"""
|
|
return await super().invoke_llm(provider, model_name, parameters)
|
|
|
|
async def invoke_text_embedding(self, provider: str, model_name: str, parameters: dict[str, Any]) -> TextEmbeddingResult:
|
|
"""
|
|
Invoke text embedding
|
|
"""
|
|
return await super().invoke_text_embedding(provider, model_name, parameters)
|
|
|
|
async def invoke_rerank(self, provider: str, model_name: str, parameters: dict[str, Any]) -> RerankResult:
|
|
"""
|
|
Invoke rerank
|
|
"""
|
|
return await super().invoke_rerank(provider, model_name, parameters)
|
|
|
|
async def invoke_question_classifier(self, node_data: QuestionClassifierNodeData, inputs: dict) -> NodeResponse:
|
|
"""
|
|
Invoke question classifier
|
|
"""
|
|
return await super().invoke_question_classifier(node_data, inputs)
|
|
|
|
async def invoke_parameter_extractor(self, node_data: ParameterExtractorNodeData, inputs: dict) -> NodeResponse:
|
|
"""
|
|
Invoke parameter extractor
|
|
"""
|
|
return await super().invoke_parameter_extractor(node_data, inputs)
|
|
|
|
async def invoke_knowledge_retrieval(self, node_data: KnowledgeRetrievalNodeData, inputs: dict) -> NodeResponse:
|
|
"""
|
|
Invoke knowledge retrieval
|
|
"""
|
|
return await super().invoke_knowledge_retrieval(node_data, inputs) |