agentkit-sdk-python 0.7.13__tar.gz → 0.8.1__tar.gz

{agentkit_sdk_python-0.7.13 → agentkit_sdk_python-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.7.13
+Version: 0.8.1
 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.7.13 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/cli/cli_config.py

@@ -131,6 +131,12 @@ def config_command(
         help="CR instance type when auto-creating: Micro/Enterprise",
     ),
     # Runtime configuration parameters
+    project_name: Optional[str] = typer.Option(
+        None,
+        "--project_name",
+        "--project-name",
+        help="Volcano Engine project for the Runtime",
+    ),
     runtime_name: Optional[str] = typer.Option(
         None, "--runtime_name", "--ve_runtime_name", help="Runtime instance name"
     ),
@@ -323,6 +329,7 @@ def config_command(
             cr_namespace_name=cr_namespace_name,
             cr_repo_name=cr_repo_name,
             cr_auto_create_instance_type=cr_auto_create_instance_type,
+            project_name=project_name,
             runtime_name=runtime_name,
             runtime_role_name=runtime_role_name,
             runtime_apikey_name=runtime_apikey_name,

{agentkit_sdk_python-0.7.13 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/cli/cli_memory.py

@@ -31,6 +31,7 @@ from rich.console import Console
 from rich.table import Table
 from rich.panel import Panel
 
+from agentkit.platform import CloudProvider, VolcConfiguration
 from agentkit.sdk.memory.client import AgentkitMemoryClient
 from agentkit.sdk.memory import types as memory_types
 
@@ -44,6 +45,8 @@ memory_app = typer.Typer(
 
 
 # ---- ProviderType normalization and validation ----
+DEFAULT_PROVIDER_TYPE = "MEM0"
+BYTEPLUS_DEFAULT_PROVIDER_TYPE = "VIKINGDB_MEMORY"
 ALLOWED_PROVIDER_TYPES = {"MEM0", "VIKINGDB_MEMORY"}
 PROVIDER_TYPE_ALIASES = {
     "mem0": "MEM0",
@@ -53,6 +56,10 @@ PROVIDER_TYPE_ALIASES = {
     "vikingdb-memory": "VIKINGDB_MEMORY",
     "VIKINGDB_MEMORY": "VIKINGDB_MEMORY",
 }
+CREATE_PROVIDER_TYPE_HELP = (
+    "Provider type: MEM0 | VIKINGDB_MEMORY. Defaults to MEM0; when cloud provider "
+    "is byteplus, defaults to VIKINGDB_MEMORY and MEM0 is not supported."
+)
 
 
 def _normalize_provider_type(value: Optional[str]) -> str:
@@ -61,7 +68,7 @@ def _normalize_provider_type(value: Optional[str]) -> str:
     Accepts case-insensitive aliases and raises a helpful error on invalid values.
     """
     if not value:
-        return "MEM0"
+        return DEFAULT_PROVIDER_TYPE
     raw = value.strip()
     if raw in ALLOWED_PROVIDER_TYPES:
         return raw
@@ -79,6 +86,22 @@ def _normalize_provider_type(value: Optional[str]) -> str:
     )
 
 
+def _normalize_create_provider_type(
+    value: Optional[str],
+    cloud_provider: CloudProvider,
+) -> str:
+    if not value and cloud_provider == CloudProvider.BYTEPLUS:
+        return BYTEPLUS_DEFAULT_PROVIDER_TYPE
+
+    normalized = _normalize_provider_type(value)
+    if cloud_provider == CloudProvider.BYTEPLUS and normalized == "MEM0":
+        raise typer.BadParameter(
+            "MEM0 provider type is not supported for BytePlus yet. "
+            "Use --provider-type vikingdb."
+        )
+    return normalized
+
+
 def _validate_collection_name(name: Optional[str]):
     """Validate memory collection name: 1-48 chars, [A-Za-z0-9_]."""
     if not name:
@@ -163,7 +186,7 @@ def create_command(
     provider_type: Optional[str] = typer.Option(
         None,
         "--provider-type",
-        help="Provider type: MEM0 | VIKINGDB_MEMORY (default: MEM0)",
+        help=CREATE_PROVIDER_TYPE_HELP,
     ),
     vpc_id: Optional[str] = typer.Option(None, "--vpc-id", help="VPC ID"),
     subnet_ids: Optional[str] = typer.Option(
@@ -202,7 +225,8 @@ def create_command(
 ):
     """Create a managed memory collection."""
     try:
-        client = AgentkitMemoryClient(region=(region or "").strip())
+        resolved_region = (region or "").strip()
+        cloud_provider = VolcConfiguration(region=resolved_region).provider
 
         if json_body:
             payload = json.loads(json_body)
@@ -210,8 +234,9 @@ def create_command(
             if provider_type is not None:
                 payload["ProviderType"] = provider_type
             # Default and normalize if missing or alias
-            payload["ProviderType"] = _normalize_provider_type(
-                payload.get("ProviderType")
+            payload["ProviderType"] = _normalize_create_provider_type(
+                payload.get("ProviderType"),
+                cloud_provider,
             )
             # Validate name and add default strategies if missing
             _validate_collection_name(payload.get("Name"))
@@ -329,12 +354,16 @@ def create_command(
                 name=name,
                 description=description,
                 project_name=project_name,
-                provider_type=_normalize_provider_type(provider_type),
+                provider_type=_normalize_create_provider_type(
+                    provider_type,
+                    cloud_provider,
+                ),
                 long_term_configuration=long_term,
                 vpc_config=vpc,
                 tags=tags,
             )
 
+        client = AgentkitMemoryClient(region=resolved_region)
         resp = client.create_memory_collection(req)
         console.print(
             Panel.fit(
@@ -445,7 +474,9 @@ def provider_types_command():
         table.add_row("VIKINGDB_MEMORY", "vikingdb, vikingdb_memory, vikingdb-memory")
         console.print(table)
         console.print(
-            "Default for create is [bold]MEM0[/bold]. Use --provider-type to override."
+            "Default for create is [bold]MEM0[/bold], or "
+            "[bold]VIKINGDB_MEMORY[/bold] when cloud provider is byteplus. "
+            "MEM0 is not supported on BytePlus."
         )
     except Exception as e:
         console.print(f"[red]Failed to list provider types: {e}[/red]")

{agentkit_sdk_python-0.7.13 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/cli/cli_runtime.py

@@ -92,12 +92,25 @@ def _build_network_for_create_runtime(
     )
 
 
+def _validate_runtime_create_authorizer_options(
+    api_key_name: Optional[str],
+    jwt_discovery_url: Optional[str],
+) -> None:
+    has_api_key_name = bool((api_key_name or "").strip())
+    has_jwt_discovery_url = bool((jwt_discovery_url or "").strip())
+
+    if has_api_key_name == has_jwt_discovery_url:
+        raise ValueError(
+            "Exactly one of --apikey-name or --jwt-discovery-url is required."
+        )
+
+
 @runtime_app.command("create")
 def create_runtime_command(
     name: str = typer.Option(..., "--name", help="Runtime name"),
     role_name: str = typer.Option(..., "--role-name", help="IAM role name"),
     artifact_type: str = typer.Option(
-        ..., "--artifact-type", help="Artifact type (e.g., DockerImage)"
+        ..., "--artifact-type", help="Artifact type (e.g., image)"
     ),
     artifact_url: str = typer.Option(
         ..., "--artifact-url", help="Artifact URL (image URL)"
@@ -137,13 +150,17 @@ def create_runtime_command(
         help="Enable shared internet egress for private network (effective for private/both)",
     ),
     api_key_name: Optional[str] = typer.Option(
-        None, "--apikey-name", help="API key name"
+        None,
+        "--apikey-name",
+        help="API key name (required unless --jwt-discovery-url is provided)",
     ),
     api_key_location: Optional[str] = typer.Option(
         None, "--apikey-location", help="API key location"
     ),
     jwt_discovery_url: Optional[str] = typer.Option(
-        None, "--jwt-discovery-url", help="JWT discovery URL"
+        None,
+        "--jwt-discovery-url",
+        help="JWT discovery URL (required unless --apikey-name is provided)",
     ),
     jwt_allowed_clients: Optional[str] = typer.Option(
         None, "--jwt-allowed-clients", help="JWT allowed clients (comma-separated)"
@@ -168,11 +185,15 @@ def create_runtime_command(
 ):
     """Create a Runtime."""
     try:
-        client = AgentkitRuntimeClient(region=(region or "").strip())
         if json_body:
             payload = json.loads(json_body)
             req = rt.CreateRuntimeRequest(**payload)
         else:
+            _validate_runtime_create_authorizer_options(
+                api_key_name=api_key_name,
+                jwt_discovery_url=jwt_discovery_url,
+            )
+
             authorizer = None
             if any([api_key_name, api_key_location, jwt_discovery_url]):
                 authorizer = rt.AuthorizerForCreateRuntime(
@@ -237,6 +258,7 @@ def create_runtime_command(
                 envs=envs,
                 tags=tags,
             )
+        client = AgentkitRuntimeClient(region=(region or "").strip())
         resp = client.create_runtime(req)
         console.print(
             Panel.fit(

{agentkit_sdk_python-0.7.13 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/cli/sandbox/__init__.py

@@ -18,9 +18,9 @@ from agentkit.toolkit.cli.sandbox.cli import sandbox_app
 from agentkit.toolkit.cli.sandbox.cli_build import build_command
 from agentkit.toolkit.cli.sandbox.cli_create import create_command
 from agentkit.toolkit.cli.sandbox.cli_exec import exec_command
-from agentkit.toolkit.cli.sandbox.cli_file import file_command
-from agentkit.toolkit.cli.sandbox.cli_get import get_command
-from agentkit.toolkit.cli.sandbox.cli_init_dockerfile import init_dockerfile_command
+from agentkit.toolkit.cli.sandbox.cli_scp import scp_command
+from agentkit.toolkit.cli.sandbox.cli_list import list_command
+from agentkit.toolkit.cli.sandbox.cli_init import init_command
 from agentkit.toolkit.cli.sandbox.cli_run import run_command
 from agentkit.toolkit.cli.sandbox.cli_shell import shell_command
 from agentkit.toolkit.cli.sandbox.cli_web import web_command
@@ -29,9 +29,9 @@ __all__ = [
     "build_command",
     "create_command",
     "exec_command",
-    "file_command",
-    "get_command",
-    "init_dockerfile_command",
+    "scp_command",
+    "init_command",
+    "list_command",
     "run_command",
     "sandbox_app",
     "shell_command",

{agentkit_sdk_python-0.7.13 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/cli/sandbox/agentkit_client.py

@@ -28,6 +28,8 @@ from agentkit.sdk.tools.client import AgentkitToolsClient as _OpenapiAgentkitToo
 from agentkit.sdk.tools.types import (
     CreateSessionRequest,
     CreateSessionResponse,
+    DeleteSessionRequest,
+    DeleteSessionResponse,
     GetSessionRequest,
     GetSessionResponse,
     ListSessionsRequest,
@@ -128,6 +130,15 @@ class TipAgentkitToolsClient(_OpenapiAgentkitToolsClient):
         if self.uses_tip_auth:
             self._tip_session = requests.Session()
             return
+        if not region:
+            try:
+                from agentkit.toolkit.cli.sandbox.config_store import (
+                    config_default_str,
+                )
+
+                region = config_default_str("region") or region
+            except Exception:
+                pass
 
         super().__init__(
             access_key=access_key,
@@ -216,6 +227,13 @@ class TipAgentkitToolsClient(_OpenapiAgentkitToolsClient):
             response_type=ListSessionsResponse,
         )
 
+    def delete_session(self, request: DeleteSessionRequest) -> DeleteSessionResponse:
+        return self._invoke_tip_api(
+            api_action="DeleteSession",
+            request=request,
+            response_type=DeleteSessionResponse,
+        )
+
     def _raise_tip_unsupported(self, api_action: str) -> None:
         raise Exception(
             f"{api_action} is not available with TIP sandbox auth. "
@@ -238,6 +256,11 @@ class TipAgentkitToolsClient(_OpenapiAgentkitToolsClient):
             self._raise_tip_unsupported("ListTools")
         return super().list_tools(request)
 
+    def delete_tool(self, request: Any) -> Any:
+        if self.uses_tip_auth:
+            self._raise_tip_unsupported("DeleteTool")
+        return super().delete_tool(request)
+
 
 def is_tip_agentkit_client(client: object) -> bool:
     return bool(getattr(client, "uses_tip_auth", False))

{agentkit_sdk_python-0.7.13 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/cli/sandbox/cli.py

@@ -19,12 +19,14 @@ from __future__ import annotations
 import typer
 
 from agentkit.toolkit.cli.sandbox.cli_build import build_command
+from agentkit.toolkit.cli.sandbox.cli_config import config_command
 from agentkit.toolkit.cli.sandbox.cli_create import create_command
+from agentkit.toolkit.cli.sandbox.cli_delete import delete_command
 from agentkit.toolkit.cli.sandbox.cli_exec import exec_command
-from agentkit.toolkit.cli.sandbox.cli_file import file_command
-from agentkit.toolkit.cli.sandbox.cli_get import get_command
+from agentkit.toolkit.cli.sandbox.cli_scp import scp_command
+from agentkit.toolkit.cli.sandbox.cli_list import list_command
 from agentkit.toolkit.cli.sandbox.cli_invoke import invoke_command
-from agentkit.toolkit.cli.sandbox.cli_init_dockerfile import init_dockerfile_command
+from agentkit.toolkit.cli.sandbox.cli_init import init_command
 from agentkit.toolkit.cli.sandbox.cli_model_login import codex_login_command
 from agentkit.toolkit.cli.sandbox.cli_mount import mount_command
 from agentkit.toolkit.cli.sandbox.cli_run import run_command
@@ -38,12 +40,13 @@ sandbox_app = typer.Typer(
 )
 
 sandbox_app.command(name="build")(build_command)
-sandbox_app.command(name="init-dockerfile")(init_dockerfile_command)
+sandbox_app.command(name="init")(init_command)
 sandbox_app.command(
     name="create",
     context_settings={"allow_extra_args": True, "ignore_unknown_options": True},
 )(create_command)
-sandbox_app.command(name="get")(get_command)
+sandbox_app.command(name="delete")(delete_command)
+sandbox_app.command(name="list")(list_command)
 sandbox_app.command(name="mount")(mount_command)
 sandbox_app.command(
     name="exec",
@@ -61,4 +64,5 @@ sandbox_app.command(
 sandbox_app.command(name="web")(web_command)
 sandbox_app.command(name="codex-login")(codex_login_command)
 sandbox_app.command(name="model-login")(codex_login_command)  # provider-agnostic alias
-sandbox_app.add_typer(file_command, name="file")
+sandbox_app.command(name="config")(config_command)
+sandbox_app.command(name="scp")(scp_command)

agentkit_sdk_python-0.8.1/agentkit/toolkit/cli/sandbox/cli_config.py

@@ -0,0 +1,139 @@
+# 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.
+
+"""Config command for sandbox CLI."""
+
+from __future__ import annotations
+
+import typer
+import yaml
+
+from agentkit.toolkit.cli.sandbox.config_store import (
+    SandboxConfigError,
+    configured_sandbox_config,
+    effective_sandbox_config,
+    ensure_sandbox_config_initialized,
+    get_sandbox_config_path,
+    redact_sandbox_config,
+    set_config_value,
+    unset_config_value,
+    write_sandbox_config,
+)
+from agentkit.toolkit.cli.sandbox.sandbox_client import error
+
+
+def _dump_yaml(payload: object) -> None:
+    typer.echo(
+        yaml.safe_dump(
+            payload,
+            sort_keys=False,
+            allow_unicode=True,
+        ).rstrip()
+    )
+
+
+def _format_config_value(canonical: str, value: object) -> object:
+    if canonical in {"model-api-key", "websearch-apikey"}:
+        return "<redacted>"
+    return value
+
+
+def _parse_set_field(field: str) -> tuple[str, str]:
+    if "=" not in field:
+        raise SandboxConfigError("--set value must use KEY=VALUE format")
+    key, value = field.split("=", 1)
+    if not key.strip():
+        raise SandboxConfigError("--set key must not be empty")
+    return key, value
+
+
+def _apply_config_options(
+    set_fields: list[str],
+    unset_keys: list[str],
+    list_config: bool,
+) -> None:
+    path = get_sandbox_config_path()
+    data = None
+    wrote = False
+
+    if set_fields:
+        try:
+            path, data, _created = ensure_sandbox_config_initialized()
+            for field in set_fields:
+                key, value = _parse_set_field(field)
+                canonical, parsed = set_config_value(data, key, value)
+                typer.echo(
+                    f"Set {canonical}: {_format_config_value(canonical, parsed)}"
+                )
+        except SandboxConfigError as exc:
+            error(str(exc))
+
+    if unset_keys:
+        if data is None:
+            if not path.exists():
+                error(f"Sandbox config not found: {path}")
+            try:
+                data = configured_sandbox_config()
+            except SandboxConfigError as exc:
+                error(str(exc))
+        try:
+            for key in unset_keys:
+                canonical, removed = unset_config_value(data, key)
+                if removed:
+                    typer.echo(f"Unset {canonical}")
+                else:
+                    typer.echo(f"{canonical} was not set")
+        except SandboxConfigError as exc:
+            error(str(exc))
+
+    if data is not None:
+        write_sandbox_config(data, path)
+        wrote = True
+
+    if wrote:
+        typer.echo(f"Wrote {path}")
+
+    if list_config:
+        try:
+            _dump_yaml(redact_sandbox_config(effective_sandbox_config()))
+        except SandboxConfigError as exc:
+            error(str(exc))
+
+
+def config_command(
+    ctx: typer.Context,
+    set_fields: list[str] = typer.Option(
+        [],
+        "--set",
+        metavar="KEY=VALUE",
+        help="Set a config value. Can be repeated.",
+    ),
+    unset_keys: list[str] = typer.Option(
+        [],
+        "--unset",
+        metavar="KEY",
+        help="Unset a config value. Can be repeated.",
+    ),
+    list_config: bool = typer.Option(
+        False,
+        "--list",
+        help="List the current effective sandbox config.",
+    ),
+) -> None:
+    """Configure default values for sandbox commands."""
+    has_options = bool(set_fields or unset_keys or list_config)
+    if not has_options:
+        typer.echo(ctx.get_help())
+        raise typer.Exit()
+    _apply_config_options(set_fields, unset_keys, list_config)