agentkit-sdk-python 0.5.8__tar.gz → 0.5.10__tar.gz

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/PKG-INFO

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

@@ -49,6 +49,13 @@ from veadk.runner import Runner
 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,
+    resolve_agentkit_allow_origins,
+    split_allow_origins,
+    supports_get_fast_api_kwarg,
+)
 from agentkit.apps.agent_server_app.telemetry import telemetry
 from agentkit.apps.base_app import BaseAgentkitApp
 
@@ -101,6 +108,7 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
         *,
         app: App | None = None,
         allow_origins: list[str] | None = None,
+        allow_origin_regex: str | list[str] | None = None,
     ) -> None:
         super().__init__()
 
@@ -153,9 +161,36 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                 await handler()
             yield
 
-        self.app = self.server.get_fast_api_app(
-            lifespan=lifespan, allow_origins=allow_origins
+        resolved_allow_origins = resolve_agentkit_allow_origins(
+            allow_origins=allow_origins,
+            allow_origin_regex=allow_origin_regex,
+        )
+        get_fast_api_app = self.server.get_fast_api_app
+        get_fast_api_app_kwargs: dict[str, Any] = {"lifespan": lifespan}
+        supports_allow_origins = supports_get_fast_api_kwarg(
+            get_fast_api_app, "allow_origins"
         )
+        supports_regex_origins = adk_supports_regex_origins()
+        needs_cors_compat_middleware = False
+
+        if supports_allow_origins:
+            if supports_regex_origins:
+                get_fast_api_app_kwargs["allow_origins"] = (
+                    resolved_allow_origins or None
+                )
+            else:
+                literal_origins, combined_regex = split_allow_origins(
+                    resolved_allow_origins
+                )
+                get_fast_api_app_kwargs["allow_origins"] = literal_origins or None
+                needs_cors_compat_middleware = combined_regex is not None
+        else:
+            needs_cors_compat_middleware = bool(resolved_allow_origins)
+
+        self.app = get_fast_api_app(**get_fast_api_app_kwargs)
+
+        if needs_cors_compat_middleware:
+            add_cors_compat_middleware(self.app, resolved_allow_origins)
 
         @self.app.post("/run_sse")
         async def run_agent_sse(req: RunAgentRequest) -> StreamingResponse:

agentkit_sdk_python-0.5.10/agentkit/apps/agent_server_app/origin.py

@@ -0,0 +1,190 @@
+# 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.
+
+import inspect
+import os
+import re
+from collections.abc import Callable
+from typing import Any
+
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+
+_REGEX_PREFIX = "regex:"
+
+DEFAULT_AGENTKIT_ALLOW_ORIGINS = ["*"]
+
+STRICT_AGENTKIT_HOSTED_ORIGINS = [
+    "https://console.volcengine.com",
+    "https://console.byteplus.com",
+    "regex:https://.*\\.volcengine\\.com",
+    "regex:https://.*\\.volceapi\\.com",
+    "regex:https://.*\\.byteplus\\.com",
+    "regex:https://.*\\.byteplusapi\\.com",
+]
+
+
+def resolve_agentkit_allow_origins(
+    *,
+    allow_origins: list[str] | None,
+    allow_origin_regex: str | list[str] | None = None,
+) -> list[str]:
+    """Resolve AgentKit CORS origins with SDK defaults and env overrides."""
+
+    env_origins = _get_env_list("AGENTKIT_ALLOW_ORIGINS", "ADK_ALLOW_ORIGINS")
+    env_regexes = _get_env_list(
+        "AGENTKIT_ALLOW_ORIGIN_REGEX",
+        "ADK_ALLOW_ORIGIN_REGEX",
+    )
+    if allow_origins is not None:
+        origins = list(allow_origins)
+    elif env_origins is not None or env_regexes is not None:
+        origins = env_origins or []
+    elif _truthy_env("AGENTKIT_DISABLE_DEFAULT_ALLOW_ORIGINS"):
+        origins = []
+    else:
+        origins = list(DEFAULT_AGENTKIT_ALLOW_ORIGINS)
+
+    if allow_origin_regex is None:
+        regexes = env_regexes or []
+    else:
+        regexes = _as_list(allow_origin_regex)
+
+    resolved = [_normalize_origin(origin) for origin in origins]
+    resolved.extend(_normalize_regex(pattern) for pattern in regexes)
+
+    deduped = _dedupe(resolved)
+    _validate_regex_origins(deduped)
+    return deduped
+
+
+def split_allow_origins(allow_origins: list[str]) -> tuple[list[str], str | None]:
+    """Split ADK-style origins into literal origins and a combined regex."""
+
+    literal_origins: list[str] = []
+    regex_patterns: list[str] = []
+    for origin in allow_origins:
+        if origin.startswith(_REGEX_PREFIX):
+            pattern = origin[len(_REGEX_PREFIX) :]
+            if pattern:
+                regex_patterns.append(pattern)
+        else:
+            literal_origins.append(origin)
+
+    return literal_origins, "|".join(regex_patterns) if regex_patterns else None
+
+
+def supports_get_fast_api_kwarg(func: Callable[..., Any], kwarg_name: str) -> bool:
+    """Return whether a callable accepts a given keyword argument."""
+
+    try:
+        signature = inspect.signature(func)
+    except (TypeError, ValueError):
+        return True
+
+    for parameter in signature.parameters.values():
+        if parameter.kind == inspect.Parameter.VAR_KEYWORD:
+            return True
+    return kwarg_name in signature.parameters
+
+
+def adk_supports_regex_origins() -> bool:
+    """Return whether installed ADK understands `regex:` allow_origins entries."""
+
+    try:
+        from google.adk.cli import adk_web_server
+
+        return hasattr(adk_web_server, "_parse_cors_origins")
+    except Exception:
+        return False
+
+
+def add_cors_compat_middleware(app: FastAPI, allow_origins: list[str]) -> None:
+    """Add CORS middleware for ADK versions that cannot parse regex origins."""
+
+    literal_origins, allow_origin_regex = split_allow_origins(allow_origins)
+    app.add_middleware(
+        CORSMiddleware,
+        allow_origins=literal_origins,
+        allow_origin_regex=allow_origin_regex,
+        allow_credentials=True,
+        allow_methods=["*"],
+        allow_headers=["*"],
+    )
+
+
+def _normalize_origin(origin: str) -> str:
+    origin = origin.strip()
+    if not origin or origin == "*" or origin.startswith(_REGEX_PREFIX):
+        return origin
+    if "*" in origin:
+        return _normalize_regex(_glob_to_regex(origin))
+    return origin
+
+
+def _normalize_regex(pattern: str) -> str:
+    pattern = pattern.strip()
+    if not pattern:
+        return pattern
+    if pattern.startswith(_REGEX_PREFIX):
+        return pattern
+    return f"{_REGEX_PREFIX}{pattern}"
+
+
+def _glob_to_regex(origin: str) -> str:
+    return re.escape(origin).replace("\\*", ".*")
+
+
+def _validate_regex_origins(origins: list[str]) -> None:
+    for origin in origins:
+        if not origin.startswith(_REGEX_PREFIX):
+            continue
+        pattern = origin[len(_REGEX_PREFIX) :]
+        try:
+            re.compile(pattern)
+        except re.error as e:
+            raise ValueError(f"Invalid allow origin regex '{pattern}': {e}") from e
+
+
+def _get_env_list(*names: str) -> list[str] | None:
+    for name in names:
+        value = os.getenv(name)
+        if value is None:
+            continue
+        return [item.strip() for item in value.split(",") if item.strip()]
+    return None
+
+
+def _truthy_env(name: str) -> bool:
+    value = os.getenv(name, "")
+    return value.lower() in {"1", "true", "yes", "on"}
+
+
+def _as_list(value: str | list[str] | None) -> list[str]:
+    if value is None:
+        return []
+    if isinstance(value, str):
+        return [value]
+    return list(value)
+
+
+def _dedupe(values: list[str]) -> list[str]:
+    seen: set[str] = set()
+    deduped: list[str] = []
+    for value in values:
+        if not value or value in seen:
+            continue
+        seen.add(value)
+        deduped.append(value)
+    return deduped

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/agentkit/platform/constants.py

@@ -29,7 +29,6 @@ class ServiceMeta:
 DEFAULT_REGION_RULES = {
     "cn-shanghai": {
         "cp": "cn-beijing",
-        "tos": "cn-beijing",
     }
 }
 

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/agentkit/toolkit/builders/ve_pipeline.py

@@ -952,6 +952,7 @@ class VeCPCRBuilder(Builder):
             import time
 
             created_in_this_run = False
+            name_conflict_not_created = False
 
             # Step 1: ensure bucket exists / accessible
             if auto_created_bucket:
@@ -988,9 +989,7 @@ class VeCPCRBuilder(Builder):
                         created_in_this_run = True
                     except tos.exceptions.TosServerError as e:
                         if e.status_code == 409:
-                            # The bucket name is already taken (possibly by another account).
-                            # Ownership verification in step 2 will block the upload.
-                            pass
+                            name_conflict_not_created = True
                         else:
                             raise
 
@@ -1032,10 +1031,43 @@ class VeCPCRBuilder(Builder):
                 )
                 raise Exception(error_msg)
 
+            if name_conflict_not_created and owned:
+                actual_location = tos_service.get_bucket_location(bucket_name)
+                current_region = getattr(
+                    tos_service, "actual_region", config.tos_region
+                )
+                if actual_location:
+                    raise Exception(
+                        f"TOS bucket '{bucket_name}' exists in region '{actual_location}', "
+                        f"but the current configuration targets region '{current_region}'. "
+                        "TOS buckets are region-scoped and cannot be accessed across regions. "
+                        "Please either switch to the bucket's region or use a different bucket name "
+                        f"(e.g. '{bucket_name}-{current_region.split('-')[-1]}')."
+                    )
+                else:
+                    raise Exception(
+                        f"TOS bucket '{bucket_name}' is owned by your account but does not exist "
+                        f"in the current region '{current_region}'. It may reside in another region. "
+                        "TOS buckets are region-scoped and cannot be accessed across regions. "
+                        "Please either switch to the bucket's region or use a different bucket name."
+                    )
+
             self.reporter.success(
                 f"TOS bucket ownership verified for current account: {bucket_name}"
             )
 
+            if created_in_this_run:
+                data_plane_timeout_s = 30
+                data_plane_interval_s = 3
+                data_plane_deadline = time.time() + data_plane_timeout_s
+                while not tos_service.bucket_exists():
+                    if time.time() >= data_plane_deadline:
+                        raise Exception(
+                            f"TOS bucket '{bucket_name}' was created but is not yet "
+                            "available for uploads. Please retry in a few seconds."
+                        )
+                    time.sleep(data_plane_interval_s)
+
             # Update config with auto-generated bucket name if applicable
             if auto_created_bucket:
                 config.tos_bucket = bucket_name

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/agentkit/toolkit/volcengine/services/tos_service.py

@@ -266,6 +266,30 @@ class TOSService:
             return False
         return name in set(self.list_bucket_names())
 
+    def get_bucket_location(self, bucket_name: Optional[str] = None) -> Optional[str]:
+        """Return the region (location) of a bucket owned by this account.
+
+        Uses the already-available ListBuckets data so no extra API call is needed
+        beyond what ``bucket_is_owned`` / ``list_bucket_names`` already do.
+
+        Args:
+            bucket_name: Bucket name to look up. Defaults to configured bucket.
+
+        Returns:
+            The region string (e.g. ``"cn-beijing"``) or ``None`` if not found.
+        """
+        name = bucket_name or self.config.bucket
+        if not name:
+            return None
+        try:
+            out = self.client.list_buckets()
+            for b in getattr(out, "buckets", None) or []:
+                if getattr(b, "name", None) == name:
+                    return getattr(b, "location", None)
+        except Exception as e:
+            logger.warning(f"Failed to get bucket location: {str(e)}")
+        return None
+
     def bucket_exists(self) -> bool:
         """Check if the configured bucket exists.
 

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/agentkit/version.py

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

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/agentkit_sdk_python.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: agentkit-sdk-python
-Version: 0.5.8
+Version: 0.5.10
 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.5.8 → agentkit_sdk_python-0.5.10}/agentkit_sdk_python.egg-info/SOURCES.txt

@@ -13,6 +13,7 @@ agentkit/apps/a2a_app/telemetry.py
 agentkit/apps/agent_server_app/__init__.py
 agentkit/apps/agent_server_app/agent_server_app.py
 agentkit/apps/agent_server_app/middleware.py
+agentkit/apps/agent_server_app/origin.py
 agentkit/apps/agent_server_app/telemetry.py
 agentkit/apps/mcp_app/__init__.py
 agentkit/apps/mcp_app/mcp_app.py

{agentkit_sdk_python-0.5.8 → agentkit_sdk_python-0.5.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentkit-sdk-python"
-version = "0.5.8"
+version = "0.5.10"
 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"