agentkit-sdk-python 0.9.1__py3-none-any.whl → 0.9.2__py3-none-any.whl

agentkit/apps/agent_server_app/agent_server_app.py

@@ -59,9 +59,6 @@ except ImportError:  # pragma: no cover - default migration path avoids veadk.
     ShortTermMemory = None  # type: ignore[assignment,misc]
     VeadkRunner = None  # type: ignore[assignment]
 
-from agentkit.apps.agent_server_app.middleware import (
-    AgentkitTelemetryHTTPMiddleware,
-)
 from agentkit.apps.agent_server_app.credential_service import (
     AgentkitCredentialService,
 )
@@ -70,7 +67,9 @@ from agentkit.apps.agent_server_app.inbound_auth_adapter import (
     build_a2a_inbound_auth_executor_factory,
     save_inbound_auth,
 )
-from agentkit.apps.auth.inbound import redact_inbound_auth_headers
+from agentkit.apps.agent_server_app.middleware import (
+    AgentkitTelemetryHTTPMiddleware,
+)
 from agentkit.apps.agent_server_app.origin import (
     add_cors_compat_middleware,
     adk_supports_regex_origins,
@@ -79,6 +78,7 @@ from agentkit.apps.agent_server_app.origin import (
     supports_get_fast_api_kwarg,
 )
 from agentkit.apps.agent_server_app.telemetry import telemetry
+from agentkit.apps.auth.inbound import redact_inbound_auth_headers
 from agentkit.apps.base_app import BaseAgentkitApp
 
 if TYPE_CHECKING:
@@ -124,6 +124,34 @@ def _resolve_memory_service(root_agent: BaseAgent) -> Any:
     return InMemoryMemoryService()
 
 
+class _SsePayloadEventIdRewriter:
+    def __init__(self) -> None:
+        self._emitted_ids: set[str] = set()
+        self._next_id = 0
+
+    def prepare(self, event: Any) -> Any:
+        event_id = str(getattr(event, "id", "") or "")
+        should_rewrite = (
+            bool(getattr(event, "partial", False))
+            or not event_id
+            or event_id in self._emitted_ids
+        )
+        if should_rewrite:
+            event = event.model_copy(deep=True)
+            unique_id = (
+                f"{event_id}:veid:{self._next_id}"
+                if event_id
+                else f"veid:{self._next_id}"
+            )
+            self._next_id += 1
+            event.id = unique_id
+
+        emitted_id = str(getattr(event, "id", "") or "")
+        if emitted_id:
+            self._emitted_ids.add(emitted_id)
+        return event
+
+
 def _create_a2a_runner(
     *,
     entry: BaseAgent | App,
@@ -425,6 +453,7 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
             # Convert the events to properly formatted SSE
             async def event_generator():
                 error_payload = None
+                event_id_rewriter = _SsePayloadEventIdRewriter()
                 try:
                     stream_mode = (
                         StreamingMode.SSE if req.streaming else StreamingMode.NONE
@@ -460,6 +489,9 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                                     artifact_event,
                                 ]
                             for event_to_stream in events_to_stream:
+                                event_to_stream = event_id_rewriter.prepare(
+                                    event_to_stream
+                                )
                                 sse_event = event_to_stream.model_dump_json(
                                     exclude_none=True,
                                     by_alias=True,
@@ -596,6 +628,7 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
 
             async def event_generator():
                 error_payload = None
+                event_id_rewriter = _SsePayloadEventIdRewriter()
                 try:
                     runner = await self.server.get_runner_async(app_name)
                     async with Aclosing(
@@ -607,6 +640,7 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                         )
                     ) as agen:
                         async for event in agen:
+                            event = event_id_rewriter.prepare(event)
                             yield (
                                 "data: "
                                 + event.model_dump_json(

agentkit/plugins/agentsentry_v3.py

@@ -152,24 +152,27 @@ class AgentSentryV3:
                 return UNAVAILABLE
             data = response.json()
             result = data.get("Result") or {}
+            metadata = data.get("ResponseMetadata") or {}
+            request_id = metadata.get("RequestId")
+            if metadata.get("Error"):
+                logger.warning("AgentSentry V3 API error request_id=%s", request_id)
+                return UNAVAILABLE
+            decision = (result.get("Decision") or {}).get("DecisionType")
             logger.info(
                 "AgentSentry V3 hook=%s request_id=%s decision=%s degraded=%s",
                 hook_name,
-                (data.get("ResponseMetadata") or {}).get("RequestId"),
-                (result.get("Decision") or {}).get("DecisionType"),
+                request_id,
+                decision,
                 result.get("Degraded"),
             )
-            if (data.get("ResponseMetadata") or {}).get("Error"):
-                return UNAVAILABLE
-            if result.get("Degraded") is not False:
-                return UNAVAILABLE
-            decision = (result.get("Decision") or {}).get("DecisionType")
-            if type(decision) is int and decision == 1:
+            try:
+                decision = int(decision)
+            except (ValueError, TypeError, OverflowError):
                 return None
-            if type(decision) is int and decision == 2:
+            if decision == 2:
                 return BLOCK
-            # REPLACE/unknown decisions are stopped rather than executing unreviewed args.
-            return UNAVAILABLE
+            # Only an explicit BLOCK decision stops a successful moderation response.
+            return None
         except (requests.RequestException, ValueError, TypeError, AttributeError):
             logger.warning("AgentSentry V3 transport or response error")
             return UNAVAILABLE

agentkit/sdk/gateway/__init__.py

@@ -0,0 +1,29 @@
+# Copyright (c) 2026 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 .client import AgentkitGatewayClient
+from .types import (
+    CreateGatewayPluginBindingRequest,
+    CreateGatewayPluginBindingResponse,
+    CreateGatewayPluginRequest,
+    CreateGatewayPluginResponse,
+)
+
+__all__ = [
+    "AgentkitGatewayClient",
+    "CreateGatewayPluginBindingRequest",
+    "CreateGatewayPluginBindingResponse",
+    "CreateGatewayPluginRequest",
+    "CreateGatewayPluginResponse",
+]

agentkit/sdk/gateway/client.py

@@ -0,0 +1,70 @@
+# Copyright (c) 2026 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 ClassVar
+
+from agentkit.client import BaseAgentkitClient
+
+from .types import (
+    CreateGatewayPluginBindingRequest,
+    CreateGatewayPluginBindingResponse,
+    CreateGatewayPluginRequest,
+    CreateGatewayPluginResponse,
+)
+
+
+class AgentkitGatewayClient(BaseAgentkitClient):
+    """Manage plugins on dedicated AgentKit gateways and bind them to MCP services."""
+
+    API_ACTIONS: ClassVar[dict[str, str]] = {
+        "CreateGatewayPlugin": "CreateGatewayPlugin",
+        "CreateGatewayPluginBinding": "CreateGatewayPluginBinding",
+    }
+
+    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="gateway",
+        )
+
+    def create_gateway_plugin(
+        self, request: CreateGatewayPluginRequest
+    ) -> CreateGatewayPluginResponse:
+        """Install a plugin; create a binding separately to apply it to requests."""
+        return self._invoke_api(
+            api_action="CreateGatewayPlugin",
+            request=request,
+            response_type=CreateGatewayPluginResponse,
+        )
+
+    def create_gateway_plugin_binding(
+        self, request: CreateGatewayPluginBindingRequest
+    ) -> CreateGatewayPluginBindingResponse:
+        """Bind an installed gateway plugin to a target MCP service."""
+        return self._invoke_api(
+            api_action="CreateGatewayPluginBinding",
+            request=request,
+            response_type=CreateGatewayPluginBindingResponse,
+        )

agentkit/sdk/gateway/types.py

@@ -0,0 +1,45 @@
+# Copyright (c) 2026 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 pydantic import BaseModel, Field
+
+
+class GatewayBaseModel(BaseModel):
+    model_config = {"populate_by_name": True}
+
+
+class CreateGatewayPluginRequest(GatewayBaseModel):
+    gateway_instance_id: str = Field(..., alias="GatewayInstanceId")
+    gateway_plugin_name: str = Field(..., alias="GatewayPluginName")
+    gateway_plugin_config: str = Field(..., alias="GatewayPluginConfig")
+
+
+class CreateGatewayPluginResponse(GatewayBaseModel):
+    gateway_plugin_id: str | None = Field(default=None, alias="GatewayPluginId")
+
+
+class CreateGatewayPluginBindingRequest(GatewayBaseModel):
+    gateway_plugin_id: str = Field(..., alias="GatewayPluginId")
+    target_id: str = Field(..., alias="TargetId")
+    gateway_plugin_binding_config: str = Field(..., alias="GatewayPluginBindingConfig")
+    enable: bool | None = Field(default=None, alias="Enable")
+    comments: str | None = Field(default=None, alias="Comments")
+
+
+class CreateGatewayPluginBindingResponse(GatewayBaseModel):
+    gateway_plugin_binding_id: str | None = Field(
+        default=None, alias="GatewayPluginBindingId"
+    )

agentkit/version.py

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

{agentkit_sdk_python-0.9.1.dist-info → agentkit_sdk_python-0.9.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.9.1
+Version: 0.9.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.9.1.dist-info → agentkit_sdk_python-0.9.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 agentkit/__init__.py,sha256=l27ZMDslc3VhmmnPZJyrqVvTDoZ0LqhCtM5hw0caHcU,1021
 agentkit/errors.py,sha256=UWlXv0ZsXd_oLTMlxuf4u9-XkTdRYeVZueE--Z3JsHU,1214
-agentkit/version.py,sha256=xdaOmClla_D6rxyWhfTBx5FMq3JRKpWmIZhqZYUaCSY,653
+agentkit/version.py,sha256=S-JFp6zp3F5qnumt6_WB_5ZZ686sIhab2ZIUhSDOP7k,653
 agentkit/apps/__init__.py,sha256=_Na1A4NSaMbVfPbsFKj2U-yI5mytD7W5skMaAOGbauY,2328
 agentkit/apps/base_app.py,sha256=3hZZExL1wyTGWveJEZZoqXN086MzmkVS_WU1vyulIWg,754
 agentkit/apps/utils.py,sha256=IzimIDmT6FS6PV9MLimPh46mq5zT-EjVmnYPxBdyEq0,1934
@@ -8,7 +8,7 @@ agentkit/apps/a2a_app/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7
 agentkit/apps/a2a_app/a2a_app.py,sha256=mFuPFuqRlGdQ2VIjQq2_gtQKWwnkwSjSmIMRnammWJU,7605
 agentkit/apps/a2a_app/telemetry.py,sha256=vR8tf7EHLor62cxs5PBx4mrfHeOaV1a7Xryqlu8nGCU,4290
 agentkit/apps/agent_server_app/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7qWbo,634
-agentkit/apps/agent_server_app/agent_server_app.py,sha256=Iu4_9OXpsigS7mqvBnXFLU6R6Q29c9f_rCMUIofhMBw,27410
+agentkit/apps/agent_server_app/agent_server_app.py,sha256=ZNhmBtazxQKyfHsLPl8GDJsausziXtYfAsdO3be06jQ,28620
 agentkit/apps/agent_server_app/credential_service.py,sha256=jBSIY6-1yc01rRRVYN8wuahTIs9AEY4zH5JGSAWIzUU,3468
 agentkit/apps/agent_server_app/inbound_auth.py,sha256=FPCfFdg13Z9E-kasyZ-in3COdXHG_exKHAFThwXFnS8,1394
 agentkit/apps/agent_server_app/inbound_auth_adapter.py,sha256=n0ji_wKVmBOabepqTqWR1_0mrcYjH2xZlQJx-tGgNcs,6087
@@ -76,11 +76,14 @@ agentkit/platform/constants.py,sha256=52XqqOEBcvGQFgEb2rALyaDhdx1MOT9YSyS1s8VSYV
 agentkit/platform/context.py,sha256=eb6ub4nUhoV1hXawjSVdY7gfc5XcgHJjFIhrzOuQ2G0,1662
 agentkit/platform/provider.py,sha256=Yy5bYveVYQOJiR8_s_DW7rEDtK4cXZQSxLaUifDozuY,2245
 agentkit/plugins/__init__.py,sha256=f8WA2b4D44jM42QGY7OPAG-3EwV8URnqigwABG8JLT0,129
-agentkit/plugins/agentsentry_v3.py,sha256=fNucjUc8gLgCtFcOAEdwos8iNtij50q2DW9wA_HtuEU,6908
+agentkit/plugins/agentsentry_v3.py,sha256=mirwC-xvQ_X-oA6JcuHqZfolc7AlFz83Zxa_CDECUzM,6925
 agentkit/plugins/llm_shield.py,sha256=1z6GgZPgyFYxq9Uyf3HAkFsU2xEGqlROLlutdHirsKM,20722
 agentkit/sdk/__init__.py,sha256=lT6PKXd1N-Z4kMYIe2fdNLq9Z_apddIcwkS1DOGZSZI,635
 agentkit/sdk/account/client.py,sha256=hOxqpiI0H7qlsBZR3aUNKi8G7eJOr46GtAS-J9on_Uk,3919
 agentkit/sdk/account/types.py,sha256=RQdVaeQxd_ntlyG2HZIx4q24m1aDvttxncXtrSamILE,1540
+agentkit/sdk/gateway/__init__.py,sha256=Ujt-tNcwI4CKELMZ4Z0LCVT_cjvt9CpOw_FyYou1DOI,1040
+agentkit/sdk/gateway/client.py,sha256=KXf_hKkVe6pHfKwf_UQVlScLGnXn4Upj4ZjObr6pZqs,2384
+agentkit/sdk/gateway/types.py,sha256=jcf7YUWxT7-sQZm8mfOwHdrKd791oKCVr5zkAoAWVvQ,1756
 agentkit/sdk/identity/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7qWbo,634
 agentkit/sdk/identity/auth.py,sha256=zRX5ViIiEJZedBx4dctb7zEGYsOme70UmOPZFkBXvKk,2808
 agentkit/sdk/identity/client.py,sha256=LY0qds5y2p0b3_y1c7xa_0EnGCOub7c-D7VMMCEv-ew,2915
@@ -291,9 +294,9 @@ agentkit_identity/py.typed,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 agentkit_identity/runtime.py,sha256=hQHLvbAK760teFPTGZ0RG9Ed5DQAXAqIdaAm9ojzS34,14423
 agentkit_identity/transport.py,sha256=kwGHSEQ5Qcn1kIu6N6DVid7MdDUMj5uaIRIUA_tJ3cM,7693
 agentkit_identity/types.py,sha256=lT8KQz1pG0rgdy5Gh6WnM0mq9VqCsWSE41Z8Fw3DHZo,7358
-agentkit_sdk_python-0.9.1.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agentkit_sdk_python-0.9.1.dist-info/METADATA,sha256=gMsFpCpKN3GzKx7U2cTE-dSlLED1FBeuD2npf3FsESk,21544
-agentkit_sdk_python-0.9.1.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
-agentkit_sdk_python-0.9.1.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
-agentkit_sdk_python-0.9.1.dist-info/top_level.txt,sha256=Kw8DVx7vzg-oUh7xC_hBZw7OwKW3Y_fo_8rkWOL1d5o,27
-agentkit_sdk_python-0.9.1.dist-info/RECORD,,
+agentkit_sdk_python-0.9.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agentkit_sdk_python-0.9.2.dist-info/METADATA,sha256=WKCRL6ReEdjQImf2gHyuauwZO1jHNj9cMmC3qlaEGJs,21544
+agentkit_sdk_python-0.9.2.dist-info/WHEEL,sha256=mOdXR9C3wMVov7cSY8PY-XCzo3SmmWCHvGQr5csJoao,91
+agentkit_sdk_python-0.9.2.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
+agentkit_sdk_python-0.9.2.dist-info/top_level.txt,sha256=Kw8DVx7vzg-oUh7xC_hBZw7OwKW3Y_fo_8rkWOL1d5o,27
+agentkit_sdk_python-0.9.2.dist-info/RECORD,,