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

ksadk/agui/config.py

@@ -46,15 +46,15 @@ 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.
+def agui_config_for_detection(detection: Any) -> AGUIConfig:
+    """Build the production AG-UI config from a detected framework.
 
     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.
+    runtime; explicit ``RuntimeAppConfig.agui`` remains available to harnesses and
+    future framework adapters.  CLI bootstraps have a detection result before a
+    runner is constructed, so they use this same policy directly.
     """
-    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")
@@ -65,6 +65,11 @@ def default_agui_config(runner: Any) -> AGUIConfig:
     )
 
 
+def default_agui_config(runner: Any) -> AGUIConfig:
+    """Build the production AG-UI config without loading the user's agent."""
+    return agui_config_for_detection(getattr(runner, "detection_result", None))
+
+
 def require_agui_dependencies() -> None:
     errors = agui_dependency_errors()
     if errors:
@@ -81,6 +86,7 @@ __all__ = [
     "COPILOTKIT_VERSION",
     "agui_dependencies_available",
     "agui_dependency_errors",
+    "agui_config_for_detection",
     "default_agui_config",
     "require_agui_dependencies",
 ]

ksadk/agui/routes.py

@@ -9,17 +9,8 @@ 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)
+from ksadk.runtime.adapter import RuntimeLaunchContext
+from ksadk.runtime.executor import RuntimeExecutor
 
 
 def _fastapi_endpoint_helper() -> Callable[..., Any]:
@@ -33,7 +24,8 @@ def _fastapi_endpoint_helper() -> Callable[..., Any]:
 
 def add_ksadk_agui_endpoint(
     app: FastAPI,
-    runner: BaseRunner,
+    executor: RuntimeExecutor,
+    launch_context: RuntimeLaunchContext,
     config: AGUIConfig,
     *,
     event_store_factory=None,
@@ -41,10 +33,10 @@ def add_ksadk_agui_endpoint(
 ) -> KsadkAGUIAgent:
     require_agui_dependencies()
 
-    adapter = build_runtime_adapter(runner, runtime_type=config.runtime_type)
     agent = KsadkAGUIAgent(
         name=config.agent_name,
-        adapter=adapter,
+        executor=executor,
+        launch_context=launch_context,
         event_store_factory=event_store_factory,
         session_service_factory=session_service_factory,
     )
@@ -52,4 +44,4 @@ def add_ksadk_agui_endpoint(
     return agent
 
 
-__all__ = ["add_ksadk_agui_endpoint", "build_runtime_adapter"]
+__all__ = ["add_ksadk_agui_endpoint"]