mirror of
https://github.com/langgenius/dify.git
synced 2026-11-11 05:41:09 +08:00
55f95dbc36
Co-authored-by: Copilot <copilot@github.com> Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
"""Runtime Dify plugin context layer.
|
|
|
|
The public config identifies tenant/plugin/user context only. Plugin daemon URL
|
|
and API key are server-side settings injected by the provider factory. The layer
|
|
is intentionally config/settings-only under Agenton's state-only core: it does
|
|
not open, cache, close, or snapshot HTTP clients, and its lifecycle hooks remain
|
|
the inherited no-op hooks. Runtime code passes the FastAPI lifespan-owned shared
|
|
``httpx.AsyncClient`` into ``create_daemon_provider`` for each model adapter.
|
|
Business model-provider names belong to the LLM layer/model request, not this
|
|
daemon context layer.
|
|
"""
|
|
|
|
from dataclasses import dataclass
|
|
|
|
import httpx
|
|
from typing_extensions import Self, override
|
|
|
|
from agenton.layers import EmptyRuntimeState, NoLayerDeps, PlainLayer
|
|
from dify_agent.adapters.llm import DifyPluginDaemonProvider
|
|
from dify_agent.layers.dify_plugin.configs import DIFY_PLUGIN_LAYER_TYPE_ID, DifyPluginLayerConfig
|
|
|
|
|
|
@dataclass(slots=True)
|
|
class DifyPluginLayer(PlainLayer[NoLayerDeps, DifyPluginLayerConfig, EmptyRuntimeState]):
|
|
"""Layer that carries plugin daemon identity without owning live resources."""
|
|
|
|
type_id = DIFY_PLUGIN_LAYER_TYPE_ID
|
|
|
|
config: DifyPluginLayerConfig
|
|
daemon_url: str
|
|
daemon_api_key: str
|
|
|
|
@classmethod
|
|
@override
|
|
def from_config(cls, config: DifyPluginLayerConfig) -> Self:
|
|
"""Reject construction without server-injected daemon settings."""
|
|
del config
|
|
raise TypeError("DifyPluginLayer requires server-side daemon settings and must use a provider factory.")
|
|
|
|
@classmethod
|
|
def from_config_with_settings(
|
|
cls,
|
|
config: DifyPluginLayerConfig,
|
|
*,
|
|
daemon_url: str,
|
|
daemon_api_key: str,
|
|
) -> Self:
|
|
"""Create a plugin layer from public config plus server-only daemon settings."""
|
|
return cls(config=config, daemon_url=daemon_url, daemon_api_key=daemon_api_key)
|
|
|
|
def create_daemon_provider(self, *, http_client: httpx.AsyncClient) -> DifyPluginDaemonProvider:
|
|
"""Return a daemon provider backed by the shared plugin daemon client.
|
|
|
|
Raises:
|
|
RuntimeError: if ``http_client`` has already been closed.
|
|
"""
|
|
if http_client.is_closed:
|
|
raise RuntimeError("DifyPluginLayer.create_daemon_provider() requires an open shared HTTP client.")
|
|
return DifyPluginDaemonProvider(
|
|
tenant_id=self.config.tenant_id,
|
|
plugin_id=self.config.plugin_id,
|
|
plugin_daemon_url=self.daemon_url,
|
|
plugin_daemon_api_key=self.daemon_api_key,
|
|
user_id=self.config.user_id,
|
|
http_client=http_client,
|
|
)
|
|
|
|
|
|
__all__ = ["DifyPluginLayer"]
|