agentkit-sdk-python 0.5.2__tar.gz → 0.5.4__tar.gz

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.5.2
+Version: 0.5.4
 Summary: Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure.
 Author-email: Xiangrui Cheng <innsdcc@gmail.com>, Yumeng Bao <baoyumeng.123@gmail.com>, Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>
 License:                                  Apache License

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/apps/agent_server_app/agent_server_app.py

@@ -31,7 +31,7 @@ from google.adk.artifacts.in_memory_artifact_service import (
 from google.adk.auth.credential_service.in_memory_credential_service import (
     InMemoryCredentialService,
 )
-from google.adk.cli.adk_web_server import AdkWebServer
+from google.adk.cli.adk_web_server import AdkWebServer, RunAgentRequest
 from google.adk.cli.utils.base_agent_loader import BaseAgentLoader
 from google.adk.evaluation.local_eval_set_results_manager import (
     LocalEvalSetResultsManager,
@@ -149,6 +149,92 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
 
         self.app = self.server.get_fast_api_app(lifespan=lifespan)
 
+        @self.app.post("/run_sse")
+        async def run_agent_sse(req: RunAgentRequest) -> StreamingResponse:
+            logger.info("Overriding run_agent_sse endpoint...")
+            # SSE endpoint
+            session = await self.server.session_service.get_session(
+                app_name=req.app_name,
+                user_id=req.user_id,
+                session_id=req.session_id,
+            )
+            if not session:
+                e = HTTPException(status_code=404, detail="Session not found")
+                telemetry.trace_agent_server_finish(
+                    path="/run_sse", func_result="", exception=e
+                )
+                raise e
+
+            # Convert the events to properly formatted SSE
+            async def event_generator():
+                try:
+                    stream_mode = (
+                        StreamingMode.SSE if req.streaming else StreamingMode.NONE
+                    )
+                    runner = await self.server.get_runner_async(req.app_name)
+                    async with Aclosing(
+                        runner.run_async(
+                            user_id=req.user_id,
+                            session_id=req.session_id,
+                            new_message=req.new_message,
+                            state_delta=req.state_delta,
+                            run_config=RunConfig(streaming_mode=stream_mode),
+                            invocation_id=req.invocation_id,
+                        )
+                    ) as agen:
+                        async for event in agen:
+                            # ADK Web renders artifacts from `actions.artifactDelta`
+                            # during part processing *and* during action processing
+                            # 1) the original event with `artifactDelta` cleared (content)
+                            # 2) a content-less "action-only" event carrying `artifactDelta`
+                            events_to_stream = [event]
+                            if (
+                                event.actions.artifact_delta
+                                and event.content
+                                and event.content.parts
+                            ):
+                                content_event = event.model_copy(deep=True)
+                                content_event.actions.artifact_delta = {}
+                                artifact_event = event.model_copy(deep=True)
+                                artifact_event.content = None
+                                events_to_stream = [
+                                    content_event,
+                                    artifact_event,
+                                ]
+                            for event_to_stream in events_to_stream:
+                                sse_event = event_to_stream.model_dump_json(
+                                    exclude_none=True,
+                                    by_alias=True,
+                                )
+                                logger.debug(
+                                    "Generated event in agent run streaming: %s",
+                                    sse_event,
+                                )
+                                yield f"data: {sse_event}\n\n"
+                except Exception as e:
+                    logger.exception("Error in event_generator: %s", e)
+                    telemetry.trace_agent_server_finish(
+                        path="/run_sse", func_result="", exception=e
+                    )
+                    yield f"data: {json.dumps({'error': str(e)})}\n\n"
+                # Returns a streaming response with the proper media type for SSE
+
+            return StreamingResponse(
+                event_generator(),
+                media_type="text/event-stream",
+            )
+
+        # Move the custom /run_sse route to the beginning of the routes list for priority matching (without deleting the ADK default route)
+        routes = self.app.router.routes
+        for i, r in enumerate(routes):
+            if (
+                getattr(r, "path", None) == "/run_sse"
+                and "POST" in getattr(r, "methods", set())
+                and getattr(r, "endpoint", None) == run_agent_sse
+            ):
+                routes.insert(0, routes.pop(i))
+                break
+
         # Attach ASGI middleware for unified telemetry across all routes
         self.app.add_middleware(AgentkitTelemetryHTTPMiddleware)
 
@@ -163,13 +249,27 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                 for k, v in dict(headers).items()
                 if k.lower() not in {"authorization", "token"}
             }
+            # trace request attributes on current span
+            telemetry.trace_agent_server(
+                func_name="_invoke_compat",
+                span=span,
+                headers=telemetry_headers,
+                text="",
+            )
+
             user_id = headers.get("user_id") or "agentkit_user"
             session_id = headers.get("session_id") or ""
 
             # Determine app_name from loader
             app_names = self.server.agent_loader.list_agents()
             if not app_names:
-                raise HTTPException(status_code=404, detail="No agents configured")
+                exception = HTTPException(
+                    status_code=404, detail="No agents configured"
+                )
+                telemetry.trace_agent_server_finish(
+                    path="/invoke", func_result="", exception=exception
+                )
+                raise exception
             app_name = app_names[0]
 
             # Parse payload and convert to ADK Content
@@ -193,14 +293,6 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                         text = ""
             content = types.UserContent(parts=[types.Part(text=text or "")])
 
-            # trace request attributes on current span
-            telemetry.trace_agent_server(
-                func_name="_invoke_compat",
-                span=span,
-                headers=telemetry_headers,
-                text=text or "",
-            )
-
             # Ensure session exists
             session = await self.server.session_service.get_session(
                 app_name=app_name, user_id=user_id, session_id=session_id
@@ -232,10 +324,10 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                     # finish span on successful end of stream handled by middleware
                     pass
                 except Exception as e:
-                    yield f'data: {{"error": "{str(e)}"}}\n\n'
                     telemetry.trace_agent_server_finish(
                         path="/invoke", func_result="", exception=e
                     )
+                    yield f'data: {{"error": "{str(e)}"}}\n\n'
 
             return StreamingResponse(
                 event_generator(),

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/apps/agent_server_app/telemetry.py

@@ -105,8 +105,12 @@ class Telemetry:
             }
             if exception:
                 self.handle_exception(span, exception)
-                attributes["error_type"] = exception.__class__.__name__
-
+                if getattr(exception, "status_code", None):
+                    attributes["error_type"] = (
+                        f"{exception.__class__.__name__}_{exception.status_code}"
+                    )
+                else:
+                    attributes["error_type"] = exception.__class__.__name__
             # only record invoke request latency metrics
             if (
                 hasattr(span, "start_time")

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/platform/__init__.py

@@ -17,6 +17,7 @@ from __future__ import annotations
 from typing import Optional
 
 from agentkit.platform.configuration import VolcConfiguration, Endpoint, Credentials
+from agentkit.platform.console_urls import agentkit_enable_services_url
 from agentkit.platform.provider import CloudProvider
 from agentkit.platform.constants import DEFAULT_REGION_RULES
 
@@ -27,6 +28,7 @@ __all__ = [
     "CloudProvider",
     "resolve_endpoint",
     "resolve_credentials",
+    "agentkit_enable_services_url",
     "DEFAULT_REGION_RULES",
 ]
 

agentkit_sdk_python-0.5.4/agentkit/platform/console_urls.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+import os
+from typing import Optional
+from urllib.parse import urlencode
+
+from agentkit.platform.configuration import VolcConfiguration
+from agentkit.platform.provider import CloudProvider
+
+
+def agentkit_enable_services_url(
+    *,
+    region: Optional[str] = None,
+    project_name: Optional[str] = None,
+    platform_config: Optional[VolcConfiguration] = None,
+) -> str:
+    if platform_config is not None and region is not None:
+        raise ValueError("Only one of 'region' or 'platform_config' can be provided.")
+    cfg = platform_config or VolcConfiguration(region=region)
+    ep = cfg.get_service_endpoint("agentkit")
+
+    base = (
+        "https://console.byteplus.com"
+        if cfg.provider == CloudProvider.BYTEPLUS
+        else "https://console.volcengine.com"
+    )
+
+    path = f"/agentkit/region:agentkit+{ep.region}/auth"
+
+    if cfg.provider != CloudProvider.BYTEPLUS:
+        return f"{base}{path}"
+
+    project = project_name or os.getenv("AGENTKIT_CONSOLE_PROJECT_NAME") or "default"
+    query = urlencode({"projectName": project})
+    return f"{base}{path}?{query}"

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/toolkit/builders/ve_pipeline.py

@@ -64,6 +64,10 @@ class VeCPCRBuilderConfig(AutoSerializableMixin):
         default=None,
         metadata={"system": True, "description": "Resolved cloud provider"},
     )
+    agentkit_region: str = field(
+        default="",
+        metadata={"system": True, "description": "AgentKit service region"},
+    )
 
     tos_bucket: str = field(
         default=AUTO_CREATE_VE,
@@ -1054,8 +1058,28 @@ class VeCPCRBuilder(Builder):
 
         except Exception as e:
             if "AccountDisable" in str(e):
+                from agentkit.platform import (
+                    VolcConfiguration,
+                    agentkit_enable_services_url,
+                )
+
+                provider = getattr(config, "cloud_provider", None) or getattr(
+                    getattr(config, "common_config", None), "cloud_provider", None
+                )
+                region_hint = (
+                    getattr(config, "agentkit_region", None)
+                    or getattr(config, "cp_region", None)
+                    or getattr(config, "cr_region", None)
+                    or getattr(config, "tos_region", None)
+                )
+                url = agentkit_enable_services_url(
+                    platform_config=VolcConfiguration(
+                        region=region_hint or None, provider=provider or None
+                    )
+                )
                 raise Exception(
-                    "Tos Service is not enabled, please enable it in the console. Enable services at: https://console.volcengine.com/agentkit/region:agentkit+cn-beijing/auth"
+                    "Tos Service is not enabled, please enable it in the console. "
+                    f"Enable services at: {url}"
                 )
             if "TooManyBuckets" in str(e):
                 raise Exception(

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/toolkit/executors/base_executor.py

@@ -295,7 +295,13 @@ class BaseExecutor:
 
             if missing:
                 self.logger.warning(f"Services not enabled: {missing}")
-                return PreflightResult(passed=False, missing_services=missing)
+                from agentkit.platform import agentkit_enable_services_url
+
+                return PreflightResult(
+                    passed=False,
+                    missing_services=missing,
+                    auth_url=agentkit_enable_services_url(region=region),
+                )
 
             self.logger.debug(f"All required services are enabled: {required_services}")
             return PreflightResult(passed=True, missing_services=[])

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/toolkit/executors/lifecycle_executor.py

@@ -102,7 +102,13 @@ class LifecycleExecutor(BaseExecutor):
 
             if missing:
                 self.logger.warning(f"Services not enabled: {missing}")
-                return PreflightResult(passed=False, missing_services=missing)
+                from agentkit.platform import agentkit_enable_services_url
+
+                return PreflightResult(
+                    passed=False,
+                    missing_services=missing,
+                    auth_url=agentkit_enable_services_url(region=region),
+                )
 
             self.logger.debug(f"All required services are enabled: {required_list}")
             return PreflightResult(passed=True, missing_services=[])

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/toolkit/models.py

@@ -200,9 +200,7 @@ class PreflightResult:
 
     passed: bool
     missing_services: List[str] = field(default_factory=list)
-    auth_url: str = (
-        "https://console.volcengine.com/agentkit/region:agentkit+cn-beijing/auth"
-    )
+    auth_url: str = ""
 
     @property
     def message(self) -> str:

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/toolkit/strategies/cloud_strategy.py

@@ -301,6 +301,7 @@ class CloudStrategy(Strategy):
         return VeCPCRBuilderConfig(
             common_config=common_config,
             cloud_provider=resolved_provider,
+            agentkit_region=resolver.resolve("agentkit"),
             cp_region=resolver.resolve("cp"),
             tos_bucket=strategy_config.tos_bucket,
             tos_region=resolver.resolve("tos"),

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.5.2"
+VERSION = "0.5.4"

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit_sdk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.5.2
+Version: 0.5.4
 Summary: Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure.
 Author-email: Xiangrui Cheng <innsdcc@gmail.com>, Yumeng Bao <baoyumeng.123@gmail.com>, Yaozheng Fang <fangyozheng@gmail.com>, Guodong Li <cu.eric.lee@gmail.com>
 License:                                  Apache License

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/agentkit_sdk_python.egg-info/SOURCES.txt

@@ -27,6 +27,7 @@ agentkit/client/base_iam_client.py
 agentkit/client/base_service_client.py
 agentkit/platform/__init__.py
 agentkit/platform/configuration.py
+agentkit/platform/console_urls.py
 agentkit/platform/constants.py
 agentkit/platform/context.py
 agentkit/platform/provider.py

{agentkit_sdk_python-0.5.2 → agentkit_sdk_python-0.5.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentkit-sdk-python"
-version = "0.5.2"
+version = "0.5.4"
 description = "Python SDK for transforming any AI agent into a production-ready application. Framework-agnostic primitives for runtime, memory, authentication, and tools with volcengine-managed infrastructure."
 readme = "README.md"
 requires-python = ">=3.10"