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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.8.0
+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.8.0 → 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.8.0 → 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.8.0 → 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.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/config/__init__.py

@@ -50,6 +50,7 @@ from .constants import (
     GLOBAL_CONFIG_FILE,
     GLOBAL_CONFIG_DIR,
     DEFAULT_IMAGE_TAG,
+    DEFAULT_RUNTIME_PROJECT_NAME,
     DEFAULT_TOS_BUCKET_TEMPLATE_NAME,
     DEFAULT_CR_INSTANCE_TEMPLATE_NAME,
     DEFAULT_CR_NAMESPACE,
@@ -75,6 +76,7 @@ __all__ = [
     "AUTH_TYPE_KEY_AUTH",
     "AUTH_TYPE_CUSTOM_JWT",
     "DEFAULT_WORKSPACE_NAME",
+    "DEFAULT_RUNTIME_PROJECT_NAME",
     "DEFAULT_CR_NAMESPACE",
     "DEFAULT_CR_INSTANCE_TEMPLATE_NAME",
     "DEFAULT_TOS_BUCKET_TEMPLATE_NAME",

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/config/config_handler.py

@@ -98,6 +98,7 @@ class ConfigParamHandler:
         cr_namespace_name: Optional[str],
         cr_repo_name: Optional[str],
         cr_auto_create_instance_type: Optional[str],
+        project_name: Optional[str],
         runtime_name: Optional[str],
         runtime_role_name: Optional[str],
         runtime_apikey_name: Optional[str],
@@ -178,6 +179,8 @@ class ConfigParamHandler:
             strategy_params["cr_auto_create_instance_type"] = (
                 cr_auto_create_instance_type
             )
+        if project_name is not None:
+            strategy_params["project_name"] = project_name
         if runtime_name is not None:
             strategy_params["runtime_name"] = runtime_name
         if runtime_role_name is not None:

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/config/constants.py

@@ -24,6 +24,7 @@ from pathlib import Path
 # Auto-creation resource identifier
 AUTO_CREATE_VE = "Auto"
 DEFAULT_WORKSPACE_NAME = "agentkit-cli-workspace"
+DEFAULT_RUNTIME_PROJECT_NAME = "default"
 DEFAULT_CR_NAMESPACE = "agentkit"
 DEFAULT_CR_INSTANCE_TEMPLATE_NAME = "agentkit-platform-{{account_id}}"
 DEFAULT_TOS_BUCKET_TEMPLATE_NAME = "agentkit-platform-{{account_id}}"

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/config/strategy_configs.py

@@ -22,6 +22,7 @@ from .constants import (
     DEFAULT_CR_NAMESPACE,
     DEFAULT_IMAGE_TAG,
     DEFAULT_IMAGE_TAG_TEMPLATE,
+    DEFAULT_RUNTIME_PROJECT_NAME,
     DEFAULT_WORKSPACE_NAME,
     DEFAULT_CR_INSTANCE_TEMPLATE_NAME,
     DEFAULT_TOS_BUCKET_TEMPLATE_NAME,
@@ -213,6 +214,10 @@ class HybridStrategyConfig(AutoSerializableMixin):
     )
 
     # Runtime configuration
+    project_name: str = field(
+        default=DEFAULT_RUNTIME_PROJECT_NAME,
+        metadata={"description": "Volcano Engine project for the Runtime"},
+    )
     runtime_id: str = field(
         default="",
         metadata={
@@ -478,6 +483,10 @@ class CloudStrategyConfig(AutoSerializableMixin):
     )
 
     # Runtime configuration for deployed application
+    project_name: str = field(
+        default=DEFAULT_RUNTIME_PROJECT_NAME,
+        metadata={"description": "Volcano Engine project for the Runtime"},
+    )
     runtime_id: str = field(
         default=AUTO_CREATE_VE,
         metadata={

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/runners/ve_agentkit.py

@@ -26,13 +26,14 @@ from agentkit.toolkit.config import (
     AUTO_CREATE_VE,
     AUTH_TYPE_KEY_AUTH,
     AUTH_TYPE_CUSTOM_JWT,
+    DEFAULT_RUNTIME_PROJECT_NAME,
     is_valid_config,
     is_invalid_config,
 )
 from agentkit.toolkit.config.dataclass_utils import AutoSerializableMixin
 from agentkit.toolkit.models import DeployResult, InvokeResult, StatusResult
 from agentkit.toolkit.reporter import Reporter
-from agentkit.toolkit.errors import ErrorCode
+from agentkit.toolkit.errors import ConfigError, ErrorCode
 from agentkit.utils.misc import (
     generate_runtime_name,
     generate_runtime_role_name,
@@ -49,7 +50,6 @@ import agentkit.sdk.runtime.types as runtime_types
 
 ARTIFACT_TYPE_DOCKER_IMAGE = "image"
 API_KEY_LOCATION = "HEADER"
-PROJECT_NAME_DEFAULT = "default"
 RUNTIME_STATUS_READY = "Ready"
 RUNTIME_STATUS_ERROR = "Error"
 RUNTIME_STATUS_UPDATING = "Updating"
@@ -67,6 +67,10 @@ class VeAgentkitRunnerConfig(AutoSerializableMixin):
     )
 
     # Runtime configuration
+    project_name: str = field(
+        default=DEFAULT_RUNTIME_PROJECT_NAME,
+        metadata={"description": "Volcano Engine project for the Runtime"},
+    )
     runtime_id: str = field(
         default=AUTO_CREATE_VE,
         metadata={"description": "Runtime ID; 'Auto' means auto-create"},
@@ -169,6 +173,21 @@ class VeAgentkitRuntimeRunner(Runner):
     def _get_runtime_client(self, region: str = "") -> AgentkitRuntimeClient:
         return AgentkitRuntimeClient(region=region or "")
 
+    @staticmethod
+    def _get_runtime_for_project(
+        client: AgentkitRuntimeClient, runtime_id: str, project_name: str
+    ) -> runtime_types.GetRuntimeResponse:
+        runtime = client.get_runtime(
+            runtime_types.GetRuntimeRequest(runtime_id=runtime_id)
+        )
+        actual_project_name = getattr(runtime, "project_name", None)
+        if actual_project_name and actual_project_name != project_name:
+            raise ConfigError(
+                f"Runtime {runtime_id} belongs to project '{actual_project_name}', "
+                f"not '{project_name}'."
+            )
+        return runtime
+
     def deploy(self, config: VeAgentkitRunnerConfig) -> DeployResult:
         """Deploy Runtime.
 
@@ -240,6 +259,9 @@ class VeAgentkitRuntimeRunner(Runner):
                 return True
 
             client = self._get_runtime_client(config.region)
+            self._get_runtime_for_project(
+                client, runner_config.runtime_id, runner_config.project_name
+            )
             client.delete_runtime(
                 runtime_types.DeleteRuntimeRequest(runtime_id=runner_config.runtime_id)
             )
@@ -280,8 +302,8 @@ class VeAgentkitRuntimeRunner(Runner):
                 )
 
             client = self._get_runtime_client(config.region)
-            runtime = client.get_runtime(
-                runtime_types.GetRuntimeRequest(runtime_id=runner_config.runtime_id)
+            runtime = self._get_runtime_for_project(
+                client, runner_config.runtime_id, runner_config.project_name
             )
 
             # Only fetch API key for key_auth mode
@@ -367,6 +389,7 @@ class VeAgentkitRuntimeRunner(Runner):
                 ),
                 metadata={
                     "runtime_id": runner_config.runtime_id,
+                    "project_name": runner_config.project_name,
                     "runtime_name": runtime.name
                     if hasattr(runtime, "name")
                     else runner_config.runtime_name,
@@ -379,6 +402,14 @@ class VeAgentkitRuntimeRunner(Runner):
                 },
             )
 
+        except ConfigError as e:
+            return StatusResult(
+                success=False,
+                status="error",
+                error=str(e),
+                error_code=e.error_code,
+                metadata={"runtime_id": runner_config.runtime_id},
+            )
         except Exception as e:
             logger.error(f"Failed to get Runtime status: {str(e)}")
             if "InvalidAgentKitRuntime.NotFound" in str(e):
@@ -441,10 +472,10 @@ class VeAgentkitRuntimeRunner(Runner):
                 # Auto-fetch Runtime information if not cached
                 try:
                     client = self._get_runtime_client(config.region)
-                    runtime = client.get_runtime(
-                        runtime_types.GetRuntimeRequest(
-                            runtime_id=runner_config.runtime_id
-                        )
+                    runtime = self._get_runtime_for_project(
+                        client,
+                        runner_config.runtime_id,
+                        runner_config.project_name,
                     )
                 except Exception as e:
                     if "NotFound" in str(e):
@@ -551,6 +582,8 @@ class VeAgentkitRuntimeRunner(Runner):
                     success=False, error=error_msg, error_code=ErrorCode.INVOKE_FAILED
                 )
 
+        except ConfigError as e:
+            return InvokeResult(success=False, error=str(e), error_code=e.error_code)
         except Exception as e:
             error_msg = f"Runtime invocation failed: {str(e)}"
             logger.exception("Runtime invocation failed with exception")
@@ -773,7 +806,7 @@ class VeAgentkitRuntimeRunner(Runner):
                 ),
                 network_configuration=network_configuration,
                 envs=envs,
-                project_name=PROJECT_NAME_DEFAULT,
+                project_name=config.project_name,
                 authorizer_configuration=authorizer_config,
                 client_token=generate_client_token(),
                 apmplus_enable=True,
@@ -861,6 +894,7 @@ class VeAgentkitRuntimeRunner(Runner):
                 deploy_timestamp=datetime.now(),
                 metadata={
                     "runtime_id": config.runtime_id,
+                    "project_name": config.project_name,
                     "runtime_name": config.runtime_name,
                     "runtime_apikey": config.runtime_apikey,
                     "runtime_apikey_name": config.runtime_apikey_name,
@@ -1132,8 +1166,8 @@ class VeAgentkitRuntimeRunner(Runner):
 
             # Get existing Runtime information
             try:
-                runtime = client.get_runtime(
-                    runtime_types.GetRuntimeRequest(runtime_id=config.runtime_id)
+                runtime = self._get_runtime_for_project(
+                    client, config.runtime_id, config.project_name
                 )
             except Exception as e:
                 if "InvalidAgentKitRuntime.NotFound" in str(e):
@@ -1182,6 +1216,7 @@ class VeAgentkitRuntimeRunner(Runner):
                     deploy_timestamp=datetime.now(),
                     metadata={
                         "runtime_id": config.runtime_id,
+                        "project_name": config.project_name,
                         "runtime_name": config.runtime_name,
                         "runtime_apikey": config.runtime_apikey,
                         "runtime_auth_type": config.runtime_auth_type,
@@ -1327,6 +1362,7 @@ class VeAgentkitRuntimeRunner(Runner):
                 deploy_timestamp=datetime.now(),
                 metadata={
                     "runtime_id": config.runtime_id,
+                    "project_name": config.project_name,
                     "runtime_name": runtime.name
                     if hasattr(runtime, "name")
                     else config.runtime_name,
@@ -1340,6 +1376,13 @@ class VeAgentkitRuntimeRunner(Runner):
                 },
             )
 
+        except ConfigError as e:
+            return DeployResult(
+                success=False,
+                error=str(e),
+                error_code=e.error_code,
+                service_id=config.runtime_id,
+            )
         except Exception as e:
             error_msg = f"Failed to update Runtime: {str(e)}"
             logger.exception("Runtime update failed with exception")

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/strategies/cloud_strategy.py

@@ -342,6 +342,7 @@ class CloudStrategy(Strategy):
 
         return VeAgentkitRunnerConfig(
             common_config=common_config,
+            project_name=strategy_config.project_name,
             runtime_id=strategy_config.runtime_id,
             runtime_name=strategy_config.runtime_name,
             runtime_role_name=strategy_config.runtime_role_name,

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/toolkit/strategies/hybrid_strategy.py

@@ -508,6 +508,7 @@ class HybridStrategy(Strategy):
 
         return VeAgentkitRunnerConfig(
             common_config=common_config,
+            project_name=strategy_config.project_name,
             runtime_id=strategy_config.runtime_id or AUTO_CREATE_VE,
             runtime_name=strategy_config.runtime_name,
             runtime_role_name=strategy_config.runtime_role_name,

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit/version.py

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

{agentkit_sdk_python-0.8.0 → agentkit_sdk_python-0.8.1}/agentkit_sdk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.8.0
+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.8.0 → agentkit_sdk_python-0.8.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentkit-sdk-python"
-version = "0.8.0"
+version = "0.8.1"
 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"