agentkit-sdk-python 0.8.5__tar.gz → 0.8.7__tar.gz

{agentkit_sdk_python-0.8.5 → agentkit_sdk_python-0.8.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.8.5
+Version: 0.8.7
 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.8.5 → agentkit_sdk_python-0.8.7}/agentkit/auth/admin.py

@@ -47,9 +47,15 @@ POLICY_NAME = "agentkit_cli_sandbox_access"
 SANDBOX_ACCESS_POLICY = "AgentKitSandboxAccess"
 WELL_KNOWN_KEY = ".well-known/agentkit-cli"
 ROLE_ACTIONS = (
-    "agentkit:CreateSession", "agentkit:GetSession", "agentkit:DeleteSession",
-    "agentkit:GetSessionLogs", "agentkit:SetSessionTtl",
-    "agentkit:ListTools", "agentkit:GetTool",
+    "agentkit:CreateSession",
+    "agentkit:GetSession",
+    "agentkit:DeleteSession",
+    "agentkit:PauseSession",
+    "agentkit:ResumeSession",
+    "agentkit:GetSessionLogs",
+    "agentkit:SetSessionTtl",
+    "agentkit:ListTools",
+    "agentkit:GetTool",
 )
 
 

{agentkit_sdk_python-0.8.5 → agentkit_sdk_python-0.8.7}/agentkit/frameworks/model_replacement.py

@@ -7,10 +7,12 @@ import logging
 import os
 from typing import Any
 
-
 logger = logging.getLogger(__name__)
 ARK_DEFAULT_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3"
+VEADK_SOURCE_HEADER = "veadk-source"
+VEADK_SOURCE_VALUE = "veadk"
 _DISABLED_VALUES = {"0", "false", "off", "none", "disabled"}
+_HEADER_PATCH_MARKER = "__agentkit_veadk_source_header__"
 _MODEL_PARAM_KEYS = {
     "frequency_penalty",
     "max_tokens",
@@ -49,6 +51,55 @@ def _replacement_enabled() -> bool:
     return value not in _DISABLED_VALUES
 
 
+def _merge_veadk_source_header(headers: Any = None) -> dict[str, Any]:
+    merged = {
+        key: value
+        for key, value in dict(headers or {}).items()
+        if key.lower() != VEADK_SOURCE_HEADER
+    }
+    merged[VEADK_SOURCE_HEADER] = VEADK_SOURCE_VALUE
+    return merged
+
+
+def _header_kwarg_model_cls(base_cls: type[Any], header_kwarg: str) -> type[Any]:
+    if getattr(base_cls, _HEADER_PATCH_MARKER, False):
+        return base_cls
+
+    class AgentKitHeaderModel(base_cls):
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            kwargs[header_kwarg] = _merge_veadk_source_header(kwargs.get(header_kwarg))
+            super().__init__(*args, **kwargs)
+
+    AgentKitHeaderModel.__name__ = base_cls.__name__
+    AgentKitHeaderModel.__qualname__ = base_cls.__qualname__
+    AgentKitHeaderModel.__module__ = base_cls.__module__
+    setattr(AgentKitHeaderModel, _HEADER_PATCH_MARKER, True)
+    return AgentKitHeaderModel
+
+
+def _openai_client_args_model_cls(base_cls: type[Any]) -> type[Any]:
+    if getattr(base_cls, _HEADER_PATCH_MARKER, False):
+        return base_cls
+
+    class AgentKitHeaderModel(base_cls):
+        def __init__(self, *args: Any, **kwargs: Any) -> None:
+            if args or kwargs.get("client") is not None:
+                super().__init__(*args, **kwargs)
+                return
+            client_args = dict(kwargs.get("client_args") or {})
+            client_args["default_headers"] = _merge_veadk_source_header(
+                client_args.get("default_headers")
+            )
+            kwargs["client_args"] = client_args
+            super().__init__(*args, **kwargs)
+
+    AgentKitHeaderModel.__name__ = base_cls.__name__
+    AgentKitHeaderModel.__qualname__ = base_cls.__qualname__
+    AgentKitHeaderModel.__module__ = base_cls.__module__
+    setattr(AgentKitHeaderModel, _HEADER_PATCH_MARKER, True)
+    return AgentKitHeaderModel
+
+
 def _replacement_model_config(config: dict[str, Any]) -> dict[str, Any]:
     model_id = _target_model_id()
     if not model_id:
@@ -90,6 +141,7 @@ def _agentkit_openai_model_cls() -> type[Any]:
                 client_args={
                     "base_url": _target_base_url(),
                     "api_key": api_key,
+                    "default_headers": _merge_veadk_source_header(),
                 },
                 **_replacement_model_config(kwargs),
             )
@@ -101,7 +153,7 @@ def _agentkit_openai_model_cls() -> type[Any]:
 def _patch_model_attr(module_name: str, attr: str, replacement: type[Any]) -> bool:
     try:
         module = importlib.import_module(module_name)
-    except Exception:
+    except Exception:  # noqa: BLE001 - optional framework imports are best effort
         return False
     if hasattr(module, attr):
         setattr(module, attr, replacement)
@@ -109,6 +161,49 @@ def _patch_model_attr(module_name: str, attr: str, replacement: type[Any]) -> bo
     return False
 
 
+def _patch_langchain_model_headers() -> bool:
+    try:
+        module = importlib.import_module("langchain_openai.chat_models.base")
+        replacement = _header_kwarg_model_cls(module.ChatOpenAI, "default_headers")
+    except Exception:  # noqa: BLE001 - optional framework imports are best effort
+        return False
+
+    patched = False
+    for module_name in (
+        "langchain_openai",
+        "langchain_openai.chat_models",
+        "langchain_openai.chat_models.base",
+    ):
+        patched = _patch_model_attr(module_name, "ChatOpenAI", replacement) or patched
+    return patched
+
+
+def _patch_adk_model_headers() -> bool:
+    try:
+        module = importlib.import_module("google.adk.models.lite_llm")
+        replacement = _header_kwarg_model_cls(module.LiteLlm, "extra_headers")
+    except Exception:  # noqa: BLE001 - optional framework imports are best effort
+        return False
+
+    patched = False
+    for module_name in ("google.adk.models", "google.adk.models.lite_llm"):
+        patched = _patch_model_attr(module_name, "LiteLlm", replacement) or patched
+    return patched
+
+
+def _patch_strands_openai_model_headers() -> bool:
+    try:
+        module = importlib.import_module("strands.models.openai")
+        replacement = _openai_client_args_model_cls(module.OpenAIModel)
+    except Exception:  # noqa: BLE001 - optional framework imports are best effort
+        return False
+
+    patched = False
+    for module_name in ("strands.models", "strands.models.openai"):
+        patched = _patch_model_attr(module_name, "OpenAIModel", replacement) or patched
+    return patched
+
+
 def _apply_env_aliases() -> bool:
     changed = False
     model_id = _target_model_id()
@@ -138,7 +233,7 @@ def _apply_env_aliases() -> bool:
 def _patch_strands_model_classes() -> bool:
     try:
         replacement = _agentkit_openai_model_cls()
-    except Exception:
+    except Exception:  # noqa: BLE001 - optional framework imports are best effort
         return False
 
     patched = False
@@ -155,21 +250,33 @@ def apply_agentkit_model_replacement() -> bool:
     """Apply explicit target-model replacement for generated migration wrappers.
 
     The helper intentionally stays low-intrusion: it normalizes common model
-    environment variables and patches only stable Strands Bedrock/Anthropic
-    model classes before user code constructs them. It does not rewrite project
-    source code or chase framework-specific model SDK constructor details.
+    environment variables and patches stable framework model constructors before
+    user code constructs them. It does not rewrite project source code or patch
+    global HTTP/OpenAI clients.
     """
 
     if not _replacement_enabled():
         logger.info("AgentKit model replacement disabled by ARK_MODEL_REPLACEMENT.")
         return False
     env_changed = _apply_env_aliases()
+    langchain_headers_patched = _patch_langchain_model_headers()
+    adk_headers_patched = _patch_adk_model_headers()
+    strands_headers_patched = _patch_strands_openai_model_headers()
     patched = _patch_strands_model_classes()
     logger.info(
-        "AgentKit model replacement enabled: env_aliases=%s strands_model_patch=%s target_model_configured=%s base_url_configured=%s",
+        "AgentKit model replacement enabled: env_aliases=%s langchain_headers=%s adk_headers=%s strands_headers=%s strands_model_patch=%s target_model_configured=%s base_url_configured=%s",
         env_changed,
+        langchain_headers_patched,
+        adk_headers_patched,
+        strands_headers_patched,
         patched,
         bool(_target_model_id()),
         bool(_target_base_url()),
     )
-    return env_changed or patched
+    return (
+        env_changed
+        or langchain_headers_patched
+        or adk_headers_patched
+        or strands_headers_patched
+        or patched
+    )

agentkit_sdk_python-0.8.7/agentkit/sdk/runtime/gateway.py

@@ -0,0 +1,90 @@
+# 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 Any, Optional
+
+RUNTIME_GATEWAY_MODES = {"Shared", "Exclusive"}
+
+
+def normalize_runtime_gateway_mode(
+    value: Any,
+    label: str = "runtime.gateway_mode",
+) -> Optional[str]:
+    if value is None or value == "":
+        return None
+    if not isinstance(value, str):
+        raise ValueError(f"{label} must be Shared or Exclusive.")
+
+    normalized = value.strip().lower()
+    if normalized == "shared":
+        return "Shared"
+    if normalized == "exclusive":
+        return "Exclusive"
+    raise ValueError(f"{label} must be Shared or Exclusive.")
+
+
+def effective_runtime_gateway_mode(mode: Optional[str]) -> str:
+    return str(mode).strip() if mode and str(mode).strip() else "Shared"
+
+
+def validate_runtime_gateway_create_config(
+    *,
+    gateway_mode: Optional[str],
+    gateway_instance_id: Optional[str],
+    has_network_configuration: bool = False,
+    label: str = "runtime",
+) -> None:
+    mode = (
+        normalize_runtime_gateway_mode(gateway_mode, f"{label}.gateway_mode")
+        or "Shared"
+    )
+    gateway_instance_id = (gateway_instance_id or "").strip()
+
+    if mode == "Exclusive":
+        if not gateway_instance_id:
+            raise ValueError(
+                f"{label}.gateway_instance_id is required when gateway_mode is Exclusive."
+            )
+        if has_network_configuration:
+            raise ValueError(
+                f"{label}.network cannot be used when gateway_mode is Exclusive; "
+                "the Runtime follows the selected gateway instance network."
+            )
+        return
+
+    if gateway_instance_id:
+        raise ValueError(
+            f"{label}.gateway_instance_id is only valid when gateway_mode is Exclusive."
+        )
+
+
+def runtime_gateway_binding_changed(
+    *,
+    desired_mode: Optional[str],
+    desired_instance_id: Optional[str],
+    current_mode: Optional[str],
+    current_instance_id: Optional[str],
+) -> bool:
+    desired_mode = effective_runtime_gateway_mode(desired_mode)
+    current_mode = effective_runtime_gateway_mode(current_mode)
+
+    if desired_mode not in RUNTIME_GATEWAY_MODES or current_mode not in RUNTIME_GATEWAY_MODES:
+        return desired_mode != current_mode
+    if desired_mode != current_mode:
+        return True
+    if desired_mode != "Exclusive":
+        return False
+    return (desired_instance_id or "").strip() != (current_instance_id or "").strip()

{agentkit_sdk_python-0.8.5 → agentkit_sdk_python-0.8.7}/agentkit/sdk/runtime/types.py

@@ -40,6 +40,11 @@ class AgentKitRuntimeVersionsForListRuntimeVersions(RuntimeTypeBaseModel):
     created_at: Optional[str] = Field(default=None, alias="CreatedAt")
     description: Optional[str] = Field(default=None, alias="Description")
     envs: Optional[list[EnvsForListRuntimeVersions]] = Field(default=None, alias="Envs")
+    gateway_instance_id: Optional[str] = Field(default=None, alias="GatewayInstanceId")
+    gateway_instance_name: Optional[str] = Field(
+        default=None, alias="GatewayInstanceName"
+    )
+    gateway_mode: Optional[str] = Field(default=None, alias="GatewayMode")
     memory_mb: Optional[int] = Field(default=None, alias="MemoryMb")
     role_name: Optional[str] = Field(default=None, alias="RoleName")
     runtime_id: Optional[str] = Field(default=None, alias="RuntimeId")
@@ -63,6 +68,11 @@ class AgentKitRuntimesForListRuntimes(RuntimeTypeBaseModel):
     )
     description: Optional[str] = Field(default=None, alias="Description")
     envs: Optional[list[EnvsForListRuntimes]] = Field(default=None, alias="Envs")
+    gateway_instance_id: Optional[str] = Field(default=None, alias="GatewayInstanceId")
+    gateway_instance_name: Optional[str] = Field(
+        default=None, alias="GatewayInstanceName"
+    )
+    gateway_mode: Optional[str] = Field(default=None, alias="GatewayMode")
     knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
     mcp_toolset_id: Optional[str] = Field(default=None, alias="MCPToolsetId")
     memory_id: Optional[str] = Field(default=None, alias="MemoryId")
@@ -326,6 +336,8 @@ class CreateRuntimeRequest(RuntimeTypeBaseModel):
     client_token: Optional[str] = Field(default=None, alias="ClientToken")
     cpu_milli: Optional[int] = Field(default=None, alias="CpuMilli")
     description: Optional[str] = Field(default=None, alias="Description")
+    gateway_instance_id: Optional[str] = Field(default=None, alias="GatewayInstanceId")
+    gateway_mode: Optional[str] = Field(default=None, alias="GatewayMode")
     knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
     mcp_toolset_id: Optional[str] = Field(default=None, alias="MCPToolsetId")
     max_concurrency: Optional[int] = Field(default=None, alias="MaxConcurrency")
@@ -388,6 +400,11 @@ class GetRuntimeResponse(RuntimeTypeBaseModel):
     description: Optional[str] = Field(default=None, alias="Description")
     envs: Optional[list[EnvsForGetRuntime]] = Field(default=None, alias="Envs")
     failed_log_file_url: Optional[str] = Field(default=None, alias="FailedLogFileUrl")
+    gateway_instance_id: Optional[str] = Field(default=None, alias="GatewayInstanceId")
+    gateway_instance_name: Optional[str] = Field(
+        default=None, alias="GatewayInstanceName"
+    )
+    gateway_mode: Optional[str] = Field(default=None, alias="GatewayMode")
     knowledge_id: Optional[str] = Field(default=None, alias="KnowledgeId")
     mcp_toolset_id: Optional[str] = Field(default=None, alias="MCPToolsetId")
     max_concurrency: Optional[int] = Field(default=None, alias="MaxConcurrency")
@@ -455,6 +472,11 @@ class GetRuntimeVersionResponse(RuntimeTypeBaseModel):
     description: Optional[str] = Field(default=None, alias="Description")
     endpoint: Optional[str] = Field(default=None, alias="Endpoint")
     envs: Optional[list[EnvsForGetRuntimeVersion]] = Field(default=None, alias="Envs")
+    gateway_instance_id: Optional[str] = Field(default=None, alias="GatewayInstanceId")
+    gateway_instance_name: Optional[str] = Field(
+        default=None, alias="GatewayInstanceName"
+    )
+    gateway_mode: Optional[str] = Field(default=None, alias="GatewayMode")
     max_concurrency: Optional[int] = Field(default=None, alias="MaxConcurrency")
     memory_mb: Optional[int] = Field(default=None, alias="MemoryMb")
     model_agent_name: Optional[str] = Field(default=None, alias="ModelAgentName")

{agentkit_sdk_python-0.8.5 → agentkit_sdk_python-0.8.7}/agentkit/sdk/tools/client.py

@@ -46,6 +46,10 @@ from .types import (
     ListSessionsResponse,
     ListToolsRequest,
     ListToolsResponse,
+    PauseSessionRequest,
+    PauseSessionResponse,
+    ResumeSessionRequest,
+    ResumeSessionResponse,
     ResumeSessionFromSnapshotRequest,
     ResumeSessionFromSnapshotResponse,
     SetSessionTtlRequest,
@@ -72,6 +76,8 @@ class AgentkitToolsClient(BaseAgentkitClient):
         "ListSessionSnapshots": "ListSessionSnapshots",
         "ListSessions": "ListSessions",
         "ListTools": "ListTools",
+        "PauseSession": "PauseSession",
+        "ResumeSession": "ResumeSession",
         "ResumeSessionFromSnapshot": "ResumeSessionFromSnapshot",
         "SetSessionTtl": "SetSessionTtl",
         "UpdateTool": "UpdateTool",
@@ -193,6 +199,20 @@ class AgentkitToolsClient(BaseAgentkitClient):
             response_type=ListToolsResponse,
         )
 
+    def pause_session(self, request: PauseSessionRequest) -> PauseSessionResponse:
+        return self._invoke_api(
+            api_action="PauseSession",
+            request=request,
+            response_type=PauseSessionResponse,
+        )
+
+    def resume_session(self, request: ResumeSessionRequest) -> ResumeSessionResponse:
+        return self._invoke_api(
+            api_action="ResumeSession",
+            request=request,
+            response_type=ResumeSessionResponse,
+        )
+
     def resume_session_from_snapshot(
         self, request: ResumeSessionFromSnapshotRequest
     ) -> ResumeSessionFromSnapshotResponse:

{agentkit_sdk_python-0.8.5 → agentkit_sdk_python-0.8.7}/agentkit/sdk/tools/types.py

@@ -133,14 +133,31 @@ class SessionMetaForListSessions(ToolsBaseModel):
     webshell_url: Optional[str] = Field(default=None, alias="WebshellUrl")
 
 
+class SessionMetadataForGetSessionSnapshot(ToolsBaseModel):
+    key: Optional[str] = Field(default=None, alias="Key")
+    type: Optional[str] = Field(default=None, alias="Type")
+    value: Optional[str] = Field(default=None, alias="Value")
+
+
+class SessionMetadataForListSessionSnapshots(ToolsBaseModel):
+    key: Optional[str] = Field(default=None, alias="Key")
+    type: Optional[str] = Field(default=None, alias="Type")
+    value: Optional[str] = Field(default=None, alias="Value")
+
+
 class SnapshotForGetSessionSnapshot(ToolsBaseModel):
     created_at: Optional[str] = Field(default=None, alias="CreatedAt")
     reason: Optional[str] = Field(default=None, alias="Reason")
     sandbox_id: Optional[str] = Field(default=None, alias="SandboxId")
     session_id: Optional[str] = Field(default=None, alias="SessionId")
+    session_metadata: Optional[list[SessionMetadataForGetSessionSnapshot]] = Field(
+        default=None, alias="SessionMetadata"
+    )
     snapshot_id: Optional[str] = Field(default=None, alias="SnapshotId")
+    snapshot_type: Optional[str] = Field(default=None, alias="SnapshotType")
     status: Optional[str] = Field(default=None, alias="Status")
     tool_id: Optional[str] = Field(default=None, alias="ToolId")
+    user_session_id: Optional[str] = Field(default=None, alias="UserSessionId")
 
 
 class SnapshotsForListSessionSnapshots(ToolsBaseModel):
@@ -148,7 +165,11 @@ class SnapshotsForListSessionSnapshots(ToolsBaseModel):
     reason: Optional[str] = Field(default=None, alias="Reason")
     sandbox_id: Optional[str] = Field(default=None, alias="SandboxId")
     session_id: Optional[str] = Field(default=None, alias="SessionId")
+    session_metadata: Optional[list[SessionMetadataForListSessionSnapshots]] = Field(
+        default=None, alias="SessionMetadata"
+    )
     snapshot_id: Optional[str] = Field(default=None, alias="SnapshotId")
+    snapshot_type: Optional[str] = Field(default=None, alias="SnapshotType")
     status: Optional[str] = Field(default=None, alias="Status")
     tool_id: Optional[str] = Field(default=None, alias="ToolId")
     user_session_id: Optional[str] = Field(default=None, alias="UserSessionId")
@@ -186,6 +207,7 @@ class ToolsForListTools(ToolsBaseModel):
     command: Optional[str] = Field(default=None, alias="Command")
     created_at: Optional[str] = Field(default=None, alias="CreatedAt")
     description: Optional[str] = Field(default=None, alias="Description")
+    enable_mcp: Optional[bool] = Field(default=None, alias="EnableMcp")
     envs: Optional[list[EnvsForListTools]] = Field(default=None, alias="Envs")
     image_url: Optional[str] = Field(default=None, alias="ImageUrl")
     model_agent_name: Optional[str] = Field(default=None, alias="ModelAgentName")
@@ -293,6 +315,7 @@ class CreateSessionResponse(ToolsBaseModel):
 
 # CreateSessionSnapshot - Request
 class CreateSessionSnapshotRequest(ToolsBaseModel):
+    retention_count: Optional[int] = Field(default=None, alias="RetentionCount")
     session_id: str = Field(..., alias="SessionId")
     tool_id: str = Field(..., alias="ToolId")
 
@@ -396,6 +419,7 @@ class CreateToolRequest(ToolsBaseModel):
     command: Optional[str] = Field(default=None, alias="Command")
     cpu_milli: Optional[int] = Field(default=None, alias="CpuMilli")
     description: Optional[str] = Field(default=None, alias="Description")
+    enable_mcp: Optional[bool] = Field(default=None, alias="EnableMcp")
     enable_object_set_isolation: Optional[bool] = Field(
         default=None, alias="EnableObjectSetIsolation"
     )
@@ -533,6 +557,7 @@ class GetToolResponse(ToolsBaseModel):
     cpu_milli: Optional[int] = Field(default=None, alias="CpuMilli")
     created_at: Optional[str] = Field(default=None, alias="CreatedAt")
     description: Optional[str] = Field(default=None, alias="Description")
+    enable_mcp: Optional[bool] = Field(default=None, alias="EnableMcp")
     enable_object_set_isolation: Optional[bool] = Field(
         default=None, alias="EnableObjectSetIsolation"
     )
@@ -576,6 +601,7 @@ class ListSessionSnapshotsRequest(ToolsBaseModel):
     page_number: Optional[int] = Field(default=None, alias="PageNumber")
     page_size: Optional[int] = Field(default=None, alias="PageSize")
     session_id: Optional[str] = Field(default=None, alias="SessionId")
+    snapshot_type: Optional[str] = Field(default=None, alias="SnapshotType")
     tool_id: str = Field(..., alias="ToolId")
     user_session_id: Optional[str] = Field(default=None, alias="UserSessionId")
 
@@ -657,6 +683,33 @@ class ListToolsResponse(ToolsBaseModel):
     tools: Optional[list[ToolsForListTools]] = Field(default=None, alias="Tools")
 
 
+# PauseSession - Request
+class PauseSessionRequest(ToolsBaseModel):
+    session_id: str = Field(..., alias="SessionId")
+    tool_id: str = Field(..., alias="ToolId")
+
+
+# PauseSession - Response
+class PauseSessionResponse(ToolsBaseModel):
+    session_id: Optional[str] = Field(default=None, alias="SessionId")
+    status: Optional[str] = Field(default=None, alias="Status")
+
+
+# ResumeSession - Request
+class ResumeSessionRequest(ToolsBaseModel):
+    session_id: str = Field(..., alias="SessionId")
+    tool_id: str = Field(..., alias="ToolId")
+    ttl: Optional[int] = Field(default=None, alias="Ttl")
+    ttl_unit: Optional[str] = Field(default=None, alias="TtlUnit")
+
+
+# ResumeSession - Response
+class ResumeSessionResponse(ToolsBaseModel):
+    expire_at: Optional[str] = Field(default=None, alias="ExpireAt")
+    session_id: Optional[str] = Field(default=None, alias="SessionId")
+    status: Optional[str] = Field(default=None, alias="Status")
+
+
 # ResumeSessionFromSnapshot - Request
 class ResumeSessionFromSnapshotRequest(ToolsBaseModel):
     create_new_instance: Optional[bool] = Field(default=None, alias="CreateNewInstance")
@@ -722,6 +775,7 @@ class UpdateToolRequest(ToolsBaseModel):
     command: Optional[str] = Field(default=None, alias="Command")
     cpu_milli: Optional[int] = Field(default=None, alias="CpuMilli")
     description: Optional[str] = Field(default=None, alias="Description")
+    enable_mcp: Optional[bool] = Field(default=None, alias="EnableMcp")
     image_url: Optional[str] = Field(default=None, alias="ImageUrl")
     lark_app_id: Optional[str] = Field(default=None, alias="LarkAppId")
     lark_app_secret: Optional[str] = Field(default=None, alias="LarkAppSecret")

{agentkit_sdk_python-0.8.5 → agentkit_sdk_python-0.8.7}/agentkit/toolkit/cli/cli_config.py

@@ -192,6 +192,18 @@ def config_command(
         "--runtime_binding_mcp_toolset_id",
         help="Bind Runtime to an MCP Toolset ID (cloud/hybrid only)",
     ),
+    runtime_gateway_mode: Optional[str] = typer.Option(
+        None,
+        "--runtime_gateway_mode",
+        "--runtime-gateway-mode",
+        help="Runtime gateway mode (cloud/hybrid, CreateRuntime only): Shared|Exclusive",
+    ),
+    runtime_gateway_instance_id: Optional[str] = typer.Option(
+        None,
+        "--runtime_gateway_instance_id",
+        "--runtime-gateway-instance-id",
+        help="Exclusive Runtime gateway instance ID (cloud/hybrid, CreateRuntime only)",
+    ),
     # Runtime network configuration (advanced, CreateRuntime only)
     runtime_network_mode: Optional[str] = typer.Option(
         None,
@@ -344,6 +356,8 @@ def config_command(
             runtime_vpc_id=runtime_vpc_id,
             runtime_subnet_ids=runtime_subnet_ids,
             runtime_enable_shared_internet_access=runtime_enable_shared_internet_access,
+            runtime_gateway_mode=runtime_gateway_mode,
+            runtime_gateway_instance_id=runtime_gateway_instance_id,
         )
 
         has_cli_params = ConfigParamHandler.has_cli_params(cli_params)