agentkit-sdk-python 0.6.1__tar.gz → 0.6.2__tar.gz

{agentkit_sdk_python-0.6.1 → agentkit_sdk_python-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.6.1
+Version: 0.6.2
 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.6.1 → agentkit_sdk_python-0.6.2}/agentkit/apps/agent_server_app/agent_server_app.py

@@ -156,7 +156,9 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
         @asynccontextmanager
         async def lifespan(app: FastAPI):
             # trigger A2A server app startup
-            logger.info("Triggering A2A server app startup within API server...")
+            logger.info(
+                "Triggering A2A server app startup within API server..."
+            )
             for handler in _a2a_server_app.router.on_startup:
                 await handler()
             yield
@@ -278,6 +280,78 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                 routes.insert(0, routes.pop(i))
                 break
 
+        @self.app.post("/run_sse")
+        async def run_agent_sse(req: RunAgentRequest) -> StreamingResponse:
+            print("my run sse !!!")
+            # 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:
+                raise HTTPException(status_code=404, detail="Session not found")
+
+            # 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)
+                    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",
+            )
+
         # Attach ASGI middleware for unified telemetry across all routes
         self.app.add_middleware(AgentkitTelemetryHTTPMiddleware)
 
@@ -353,7 +427,9 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                             user_id=user_id,
                             session_id=session_id,
                             new_message=content,
-                            run_config=RunConfig(streaming_mode=StreamingMode.SSE),
+                            run_config=RunConfig(
+                                streaming_mode=StreamingMode.SSE
+                            ),
                         )
                     ) as agen:
                         async for event in agen:

agentkit_sdk_python-0.6.2/agentkit/sdk/identity/client.py

@@ -0,0 +1,79 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from __future__ import annotations
+
+from typing import Dict
+
+from agentkit.client import BaseAgentkitClient
+from agentkit.sdk.identity.types import (
+    CreateInboundAuthConfigRequest,
+    CreateInboundAuthConfigResponse,
+    DeleteInboundAuthConfigRequest,
+    DeleteInboundAuthConfigResponse,
+    ListInboundAuthConfigsRequest,
+    ListInboundAuthConfigsResponse,
+)
+
+
+class AgentkitIdentityClient(BaseAgentkitClient):
+    """AgentKit Identity / Inbound Auth Config Service."""
+
+    API_ACTIONS: Dict[str, str] = {
+        "CreateInboundAuthConfig": "CreateInboundAuthConfig",
+        "ListInboundAuthConfigs": "ListInboundAuthConfigs",
+        "DeleteInboundAuthConfig": "DeleteInboundAuthConfig",
+    }
+
+    def __init__(
+        self,
+        access_key: str = "",
+        secret_key: str = "",
+        region: str = "",
+        session_token: str = "",
+    ) -> None:
+        super().__init__(
+            access_key=access_key,
+            secret_key=secret_key,
+            region=region,
+            session_token=session_token,
+            service_name="identity",
+        )
+
+    def create_inbound_auth_config(
+        self, request: CreateInboundAuthConfigRequest
+    ) -> CreateInboundAuthConfigResponse:
+        return self._invoke_api(
+            api_action="CreateInboundAuthConfig",
+            request=request,
+            response_type=CreateInboundAuthConfigResponse,
+        )
+
+    def list_inbound_auth_configs(
+        self, request: ListInboundAuthConfigsRequest
+    ) -> ListInboundAuthConfigsResponse:
+        return self._invoke_api(
+            api_action="ListInboundAuthConfigs",
+            request=request,
+            response_type=ListInboundAuthConfigsResponse,
+        )
+
+    def delete_inbound_auth_config(
+        self, request: DeleteInboundAuthConfigRequest
+    ) -> DeleteInboundAuthConfigResponse:
+        return self._invoke_api(
+            api_action="DeleteInboundAuthConfig",
+            request=request,
+            response_type=DeleteInboundAuthConfigResponse,
+        )

agentkit_sdk_python-0.6.2/agentkit/sdk/identity/types.py

@@ -0,0 +1,134 @@
+# Copyright (c) 2025 Beijing Volcano Engine Technology Co., Ltd. and/or its affiliates.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# Request/response models for the InboundAuthConfig APIs.
+#
+# ``CreateInboundAuthConfig`` mirrors the published OpenAPI definition
+# (Version 2025-10-30). ``ListInboundAuthConfigs`` and
+# ``DeleteInboundAuthConfig`` follow the standard Volcengine list/delete
+# conventions; adjust the field names here if the published specs differ.
+
+from __future__ import annotations
+
+from typing import Optional
+from pydantic import BaseModel, Field
+
+
+class IdentityBaseModel(BaseModel):
+    """AgentKit auto-generated base model"""
+
+    model_config = {"populate_by_name": True, "arbitrary_types_allowed": True}
+
+
+# Data Types
+class ApiKeyMetadata(IdentityBaseModel):
+    location: Optional[str] = Field(default=None, alias="Location")
+    parameter_name: Optional[str] = Field(default=None, alias="ParameterName")
+
+
+class ApiKeyAuthConfig(IdentityBaseModel):
+    api_key_name: str = Field(..., alias="ApiKeyName")
+    api_key: Optional[str] = Field(default=None, alias="ApiKey")
+    api_key_metadata: Optional[list[ApiKeyMetadata]] = Field(
+        default=None, alias="ApiKeyMetadata"
+    )
+    expiry_timestamp: Optional[int] = Field(default=None, alias="ExpiryTimestamp")
+
+
+class JwtAuthConfig(IdentityBaseModel):
+    discovery_url: str = Field(..., alias="DiscoveryUrl")
+    allowed_audiences: Optional[list[str]] = Field(
+        default=None, alias="AllowedAudiences"
+    )
+    allowed_clients: Optional[list[str]] = Field(default=None, alias="AllowedClients")
+
+
+# CreateInboundAuthConfig - Request
+class CreateInboundAuthConfigRequest(IdentityBaseModel):
+    auth_type: str = Field(..., alias="AuthType")
+    api_key_auth_configs: Optional[list[ApiKeyAuthConfig]] = Field(
+        default=None, alias="ApiKeyAuthConfigs"
+    )
+    jwt_auth_config: Optional[JwtAuthConfig] = Field(
+        default=None, alias="JwtAuthConfig"
+    )
+    config_name: Optional[str] = Field(default=None, alias="ConfigName")
+    description: Optional[str] = Field(default=None, alias="Description")
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+
+
+# CreateInboundAuthConfig - Response
+class CreateInboundAuthConfigResponse(IdentityBaseModel):
+    inbound_auth_config_id: Optional[str] = Field(
+        default=None, alias="InboundAuthConfigId"
+    )
+    trn: Optional[str] = Field(default=None, alias="Trn")
+    config_name: Optional[str] = Field(default=None, alias="ConfigName")
+    description: Optional[str] = Field(default=None, alias="Description")
+    auth_type: Optional[str] = Field(default=None, alias="AuthType")
+    jwt_auth_config: Optional[JwtAuthConfig] = Field(
+        default=None, alias="JwtAuthConfig"
+    )
+    api_key_auth_configs: Optional[list[ApiKeyAuthConfig]] = Field(
+        default=None, alias="ApiKeyAuthConfigs"
+    )
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+    created_at: Optional[str] = Field(default=None, alias="CreatedAt")
+    updated_at: Optional[str] = Field(default=None, alias="UpdatedAt")
+
+
+# ListInboundAuthConfigs - Request
+class ListInboundAuthConfigsRequest(IdentityBaseModel):
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+    max_results: Optional[int] = Field(default=None, alias="MaxResults")
+    next_token: Optional[str] = Field(default=None, alias="NextToken")
+
+
+# ListInboundAuthConfigs - Response
+class InboundAuthConfigForList(IdentityBaseModel):
+    inbound_auth_config_id: Optional[str] = Field(
+        default=None, alias="InboundAuthConfigId"
+    )
+    trn: Optional[str] = Field(default=None, alias="Trn")
+    config_name: Optional[str] = Field(default=None, alias="ConfigName")
+    description: Optional[str] = Field(default=None, alias="Description")
+    auth_type: Optional[str] = Field(default=None, alias="AuthType")
+    jwt_auth_config: Optional[JwtAuthConfig] = Field(
+        default=None, alias="JwtAuthConfig"
+    )
+    api_key_auth_configs: Optional[list[ApiKeyAuthConfig]] = Field(
+        default=None, alias="ApiKeyAuthConfigs"
+    )
+    instance_id: Optional[str] = Field(default=None, alias="InstanceId")
+    created_at: Optional[str] = Field(default=None, alias="CreatedAt")
+    updated_at: Optional[str] = Field(default=None, alias="UpdatedAt")
+
+
+class ListInboundAuthConfigsResponse(IdentityBaseModel):
+    inbound_auth_configs: Optional[list[InboundAuthConfigForList]] = Field(
+        default=None, alias="InboundAuthConfigs"
+    )
+    next_token: Optional[str] = Field(default=None, alias="NextToken")
+
+
+# DeleteInboundAuthConfig - Request
+class DeleteInboundAuthConfigRequest(IdentityBaseModel):
+    inbound_auth_config_id: str = Field(..., alias="InboundAuthConfigId")
+
+
+# DeleteInboundAuthConfig - Response
+class DeleteInboundAuthConfigResponse(IdentityBaseModel):
+    inbound_auth_config_id: Optional[str] = Field(
+        default=None, alias="InboundAuthConfigId"
+    )

{agentkit_sdk_python-0.6.1 → agentkit_sdk_python-0.6.2}/agentkit/toolkit/cli/cli.py

@@ -23,7 +23,7 @@ from agentkit.utils.logging_config import setup_cli_logging
 
 # Import command modules
 from agentkit.toolkit.cli.cli_init import init_command, show_logo
-from agentkit.toolkit.cli.cli_invoke import invoke_command
+from agentkit.toolkit.cli.cli_invoke import invoke_app
 from agentkit.toolkit.cli.cli_config import config_command
 from agentkit.toolkit.cli.cli_version import version_command, get_package_version
 from agentkit.toolkit.cli.cli_build import build_command
@@ -44,6 +44,9 @@ from agentkit.toolkit.cli.cli_auth import (
     logout_command,
     whoami_command,
 )
+from agentkit.toolkit.cli.cli_add import add_app
+from agentkit.toolkit.cli.cli_list import list_app
+from agentkit.toolkit.cli.cli_delete import delete_app
 
 # Note: Avoid importing heavy packages at the top to keep CLI startup fast
 
@@ -99,7 +102,6 @@ def main(
 
 # Register commands
 app.command(name="init")(init_command)
-app.command(name="invoke")(invoke_command)
 app.command(name="config")(config_command)
 app.command(name="version")(version_command)
 app.command(name="build")(build_command)
@@ -122,6 +124,10 @@ app.add_typer(tools_app, name="tools")
 app.add_typer(runtime_app, name="runtime")
 app.add_typer(skills_app, name="skills")
 app.add_typer(sandbox_app, name="sandbox")
+app.add_typer(invoke_app, name="invoke")
+app.add_typer(add_app, name="add")
+app.add_typer(list_app, name="list")
+app.add_typer(delete_app, name="delete")
 
 
 if __name__ == "__main__":