mirror of
https://github.com/langgenius/dify.git
synced 2025-11-25 17:31:56 +00:00
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
|
|
|
|
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: str | None = None,
|
|
app_id: str | None = None,
|
|
message_id: str | None = None,
|
|
credentials: InvokeCredentials | None = 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: str | None = None,
|
|
app_id: str | None = None,
|
|
message_id: str | None = None,
|
|
credentials: InvokeCredentials | None = None,
|
|
) -> Generator[AgentInvokeMessage, None, None]:
|
|
pass
|