agentscope-runtime 1.0.4__py3-none-any.whl → 1.0.5__py3-none-any.whl

agentscope_runtime/engine/app/agent_app.py

@@ -1,4 +1,5 @@
 # -*- coding: utf-8 -*-
+# flake8: noqa: E501
 import logging
 import os
 import types
@@ -20,6 +21,7 @@ from ..deployers.adapter.a2a import (
 from ..deployers.adapter.responses.response_api_protocol_adapter import (
     ResponseAPIDefaultAdapter,
 )
+from ..deployers.adapter.agui import AGUIDefaultAdapter, AGUIAdaptorConfig
 from ..deployers.utils.deployment_modes import DeploymentMode
 from ..deployers.utils.service_utils.fastapi_factory import FastAPIAppFactory
 from ..runner import Runner
@@ -52,6 +54,7 @@ class AgentApp(BaseApp):
         runner: Optional[Runner] = None,
         enable_embedded_worker: bool = False,
         a2a_config: Optional["AgentCardWithRuntimeConfig"] = None,
+        agui_config: Optional[AGUIAdaptorConfig] = None,
         **kwargs,
     ):
         """
@@ -74,11 +77,14 @@ class AgentApp(BaseApp):
                 Must be an ``AgentCardWithRuntimeConfig`` instance, which
                 contains ``agent_card`` (AgentCard object or dict) and runtime
                 settings (host, port, registry, task_timeout, etc.).
-                Example:
-                    from a2a.types import AgentCard, AgentCapabilities
-                    from agentscope_runtime.engine.deployers.adapter.a2a import (  # noqa: E501
+
+                Example::
+
+                    from a2a.types import AgentCapabilities
+                    from agentscope_runtime.engine.deployers.adapter.a2a import (
                         AgentCardWithRuntimeConfig,
                     )
+
                     config = AgentCardWithRuntimeConfig(
                         agent_card={
                             "name": "MyAgent",
@@ -93,6 +99,7 @@ class AgentApp(BaseApp):
                         registry=[nacos_registry],
                         task_timeout=120,
                     )
+            agui_config: Config for AGUI adaptor.
             **kwargs: Additional keyword arguments passed to FastAPI app
         """
 
@@ -125,7 +132,12 @@ class AgentApp(BaseApp):
         )
 
         response_protocol = ResponseAPIDefaultAdapter()
-        self.protocol_adapters = [a2a_protocol, response_protocol]
+        agui_protocol = AGUIDefaultAdapter(config=agui_config)
+        self.protocol_adapters = [
+            a2a_protocol,
+            response_protocol,
+            agui_protocol,
+        ]
 
         self._app_kwargs = {
             "title": "Agent Service",

agentscope_runtime/engine/deployers/__init__.py

@@ -1,32 +1,42 @@
 # -*- coding: utf-8 -*-
-from .base import DeployManager
-from .local_deployer import LocalDeployManager
-from .kubernetes_deployer import (
-    KubernetesDeployManager,
-    K8sConfig,
-)
-from .modelstudio_deployer import (
-    ModelstudioDeployManager,
-)
-from .knative_deployer import (
-    KnativeDeployManager,
-)
+from typing import TYPE_CHECKING
+from ...common.utils.lazy_loader import install_lazy_loader
 
-try:
-    from .agentrun_deployer import (
-        AgentRunDeployManager,
+if TYPE_CHECKING:
+    from .base import DeployManager
+    from .local_deployer import LocalDeployManager
+    from .kubernetes_deployer import (
+        KubernetesDeployManager,
+        K8sConfig,
     )
-except ImportError:
-    AgentRunDeployManager = None  # type: ignore
+    from .modelstudio_deployer import ModelstudioDeployManager
+    from .knative_deployer import KnativeDeployManager
+    from .agentrun_deployer import AgentRunDeployManager
+    from .fc_deployer import FCDeployManager
+
+install_lazy_loader(
+    globals(),
+    {
+        "DeployManager": ".base",
+        "LocalDeployManager": ".local_deployer",
+        "KubernetesDeployManager": ".kubernetes_deployer",
+        "K8sConfig": ".kubernetes_deployer",
+        "ModelstudioDeployManager": ".modelstudio_deployer",
+        "KnativeDeployManager": ".knative_deployer",
+        "AgentRunDeployManager": ".agentrun_deployer",
+        "FCDeployManager": ".fc_deployer",
+    },
+)
 
 try:
-    from .fc_deployer import (
-        FCDeployManager,
+    from .pai_deployer import (
+        PAIDeployManager,
     )
 except ImportError:
-    FCDeployManager = None  # type: ignore
+    PAIDeployManager = None  # type: ignore
 
 __all__ = [
+    "K8sConfig",
     "DeployManager",
     "LocalDeployManager",
     "KubernetesDeployManager",
@@ -34,4 +44,5 @@ __all__ = [
     "AgentRunDeployManager",
     "KnativeDeployManager",
     "FCDeployManager",
+    "PAIDeployManager",
 ]

agentscope_runtime/engine/deployers/adapter/__init__.py

@@ -1,2 +1,10 @@
 # -*- coding: utf-8 -*-
 from .a2a import A2AFastAPIDefaultAdapter
+from .agui import AGUIDefaultAdapter
+from .responses import ResponseAPIDefaultAdapter
+
+__all__ = [
+    "A2AFastAPIDefaultAdapter",
+    "AGUIDefaultAdapter",
+    "ResponseAPIDefaultAdapter",
+]

agentscope_runtime/engine/deployers/adapter/a2a/a2a_protocol_adapter.py

@@ -55,20 +55,21 @@ AGENT_VERSION = "1.0.0"
 def extract_a2a_config(
     a2a_config: Optional["AgentCardWithRuntimeConfig"] = None,
 ) -> "AgentCardWithRuntimeConfig":
-    """Normalize a2a_config to AgentCardWithRuntimeConfig object.
+    """Normalize ``a2a_config`` to an ``AgentCardWithRuntimeConfig`` object.
 
     Registry resolution priority:
-    1. Use registry from a2a_config if provided
-    2. Fallback to environment variables if a2a_config.registry is
-       None
-    3. If neither is available, registry remains None (user doesn't
-       want registry)
+
+    1. Use ``registry`` from ``a2a_config`` if provided.
+    2. Fallback to environment variables if ``a2a_config.registry`` is
+       ``None``.
+    3. If neither is available, registry remains ``None``
+       (user doesn't want it).
 
     Args:
-        a2a_config: Optional AgentCardWithRuntimeConfig instance.
+        a2a_config: Optional ``AgentCardWithRuntimeConfig`` instance.
 
     Returns:
-        Normalized AgentCardWithRuntimeConfig object.
+        Normalized ``AgentCardWithRuntimeConfig`` object.
     """
     if a2a_config is None:
         a2a_config = AgentCardWithRuntimeConfig()

agentscope_runtime/engine/deployers/adapter/a2a/nacos_a2a_registry.py

@@ -157,8 +157,26 @@ def create_nacos_registry_from_env() -> Optional[A2ARegistry]:
     if not _NACOS_SDK_AVAILABLE:
         return None
 
+    # Only create a Nacos registry if the user has explicitly configured
+    # at least one NACOS_* related environment variable. If none of these
+    # are present, we assume the user does NOT want to use any registry.
+    nacos_settings = get_nacos_settings()
+    env_keys = [
+        "NACOS_SERVER_ADDR",
+        "NACOS_USERNAME",
+        "NACOS_PASSWORD",
+        "NACOS_NAMESPACE_ID",
+        "NACOS_ACCESS_KEY",
+        "NACOS_SECRET_KEY",
+    ]
+    if not any(key in os.environ for key in env_keys):
+        logger.info(
+            "[A2A] No NACOS_* environment variables found; "
+            "registry will not be enabled.",
+        )
+        return None
+
     try:
-        nacos_settings = get_nacos_settings()
         nacos_client_config = _build_nacos_client_config(nacos_settings)
         registry = NacosRegistry(nacos_client_config=nacos_client_config)
 

agentscope_runtime/engine/deployers/adapter/agui/__init__.py

@@ -0,0 +1,8 @@
+# -*- coding: utf-8 -*-
+from .agui_protocol_adapter import (
+    AGUIDefaultAdapter,
+    FlexibleRunAgentInput,
+    AGUIAdaptorConfig,
+)
+
+__all__ = ["AGUIDefaultAdapter", "FlexibleRunAgentInput", "AGUIAdaptorConfig"]