agentengine-sdk-python 0.6.9__py3-none-any.whl → 0.8.0__py3-none-any.whl

ksadk/agui/config.py

@@ -0,0 +1,86 @@
+"""Configuration and dependency gates for the optional AG-UI transport."""
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+from importlib.metadata import PackageNotFoundError, version
+from typing import Any
+
+AGUI_PROTOCOL_VERSION = "0.1.19"
+AGUI_LANGGRAPH_VERSION = "0.0.42"
+COPILOTKIT_VERSION = "0.1.94"
+
+_REQUIRED_DISTRIBUTIONS = {
+    "ag-ui-protocol": AGUI_PROTOCOL_VERSION,
+    "ag-ui-langgraph": AGUI_LANGGRAPH_VERSION,
+    "copilotkit": COPILOTKIT_VERSION,
+}
+
+
+@dataclass(frozen=True)
+class AGUIConfig:
+    """Optional per-app AG-UI endpoint configuration."""
+
+    enabled: bool = False
+    path: str = "/agentengine/agui"
+    agent_name: str = "ksadk"
+    runtime_type: str = "langgraph"
+    preferred: bool = True
+
+
+def agui_dependency_errors() -> list[str]:
+    """Return exact-version dependency mismatches without importing AG-UI eagerly."""
+    errors: list[str] = []
+    for distribution, expected in _REQUIRED_DISTRIBUTIONS.items():
+        try:
+            actual = version(distribution)
+        except PackageNotFoundError:
+            errors.append(f"{distribution}=={expected} is not installed")
+            continue
+        if actual != expected:
+            errors.append(f"{distribution}=={expected} required, found {actual}")
+    return errors
+
+
+def agui_dependencies_available() -> bool:
+    return not agui_dependency_errors()
+
+
+def default_agui_config(runner: Any) -> AGUIConfig:
+    """Build the production AG-UI config without loading the user's agent.
+
+    Goal 24's first production vertical is LangGraph.  An optional AG-UI install
+    therefore enables the endpoint automatically only for a detected LangGraph
+    runner; explicit ``RuntimeAppConfig.agui`` remains available to harnesses and
+    future framework adapters.
+    """
+    detection = getattr(runner, "detection_result", None)
+    framework_type = getattr(detection, "type", None)
+    runtime_type = str(getattr(framework_type, "value", framework_type) or "").lower()
+    agent_name = str(getattr(detection, "name", None) or "ksadk")
+    return AGUIConfig(
+        enabled=runtime_type in ("langgraph", "codex") and agui_dependencies_available(),
+        agent_name=agent_name,
+        runtime_type=runtime_type or "unknown",
+    )
+
+
+def require_agui_dependencies() -> None:
+    errors = agui_dependency_errors()
+    if errors:
+        detail = "; ".join(errors)
+        raise RuntimeError(
+            f"AG-UI transport is enabled but unavailable: {detail}. Install ksadk[agui]."
+        )
+
+
+__all__ = [
+    "AGUIConfig",
+    "AGUI_LANGGRAPH_VERSION",
+    "AGUI_PROTOCOL_VERSION",
+    "COPILOTKIT_VERSION",
+    "agui_dependencies_available",
+    "agui_dependency_errors",
+    "default_agui_config",
+    "require_agui_dependencies",
+]

ksadk/agui/routes.py

@@ -0,0 +1,55 @@
+"""Factory wiring for the official AG-UI FastAPI endpoint helper."""
+
+from __future__ import annotations
+
+from importlib import import_module
+from typing import Any, Callable, cast
+
+from fastapi import FastAPI
+
+from ksadk.agui.agent import KsadkAGUIAgent
+from ksadk.agui.config import AGUIConfig, require_agui_dependencies
+from ksadk.runners.base_runner import BaseRunner
+from ksadk.runtime.framework_adapters import ADKRuntimeAdapter, LangGraphRuntimeAdapter
+from ksadk.runtime.runner_adapter import RunnerRuntimeAdapter
+
+
+def build_runtime_adapter(runner: BaseRunner, *, runtime_type: str):
+    if runtime_type == "langgraph":
+        return LangGraphRuntimeAdapter(runner)
+    if runtime_type == "adk":
+        return ADKRuntimeAdapter(runner)
+    return RunnerRuntimeAdapter(runner, runtime_type=runtime_type)
+
+
+def _fastapi_endpoint_helper() -> Callable[..., Any]:
+    """Resolve the optional untyped helper only after the dependency gate."""
+    module = import_module("ag_ui_langgraph")
+    helper = getattr(module, "add_langgraph_fastapi_endpoint", None)
+    if not callable(helper):
+        raise RuntimeError("ag-ui-langgraph does not expose add_langgraph_fastapi_endpoint")
+    return cast(Callable[..., Any], helper)
+
+
+def add_ksadk_agui_endpoint(
+    app: FastAPI,
+    runner: BaseRunner,
+    config: AGUIConfig,
+    *,
+    event_store_factory=None,
+    session_service_factory=None,
+) -> KsadkAGUIAgent:
+    require_agui_dependencies()
+
+    adapter = build_runtime_adapter(runner, runtime_type=config.runtime_type)
+    agent = KsadkAGUIAgent(
+        name=config.agent_name,
+        adapter=adapter,
+        event_store_factory=event_store_factory,
+        session_service_factory=session_service_factory,
+    )
+    _fastapi_endpoint_helper()(app, cast(Any, agent), path=config.path)
+    return agent
+
+
+__all__ = ["add_ksadk_agui_endpoint", "build_runtime_adapter"]