mirror of
https://github.com/langgenius/dify.git
synced 2025-07-24 10:00:32 +00:00
a4ef900916
Co-authored-by: zxhlyh <jasonapring2015@outlook.com> Co-authored-by: Yeuoly <admin@srmxy.cn>
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
from abc import ABC, abstractmethod
|
|
from collections.abc import Generator, Sequence
|
|
from typing import Any, Optional
|
|
|
|
from core.agent.entities import AgentInvokeMessage
|
|
from core.agent.plugin_entities import AgentStrategyParameter
|
|
from core.plugin.entities.request import InvokeCredentials
|
|
|
|
|
|
class BaseAgentStrategy(ABC):
|
|
"""
|
|
Agent Strategy
|
|
"""
|
|
|
|
def invoke(
|
|
self,
|
|
params: dict[str, Any],
|
|
user_id: str,
|
|
conversation_id: Optional[str] = None,
|
|
app_id: Optional[str] = None,
|
|
message_id: Optional[str] = None,
|
|
credentials: Optional[InvokeCredentials] = None,
|
|
) -> Generator[AgentInvokeMessage, None, None]:
|
|
"""
|
|
Invoke the agent strategy.
|
|
"""
|
|
yield from self._invoke(params, user_id, conversation_id, app_id, message_id, credentials)
|
|
|
|
def get_parameters(self) -> Sequence[AgentStrategyParameter]:
|
|
"""
|
|
Get the parameters for the agent strategy.
|
|
"""
|
|
return []
|
|
|
|
@abstractmethod
|
|
def _invoke(
|
|
self,
|
|
params: dict[str, Any],
|
|
user_id: str,
|
|
conversation_id: Optional[str] = None,
|
|
app_id: Optional[str] = None,
|
|
message_id: Optional[str] = None,
|
|
credentials: Optional[InvokeCredentials] = None,
|
|
) -> Generator[AgentInvokeMessage, None, None]:
|
|
pass
|