agentkit-sdk-python 0.5.3__py3-none-any.whl → 0.5.4__py3-none-any.whl

agentkit/apps/agent_server_app/agent_server_app.py

@@ -169,20 +169,18 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
             async def event_generator():
                 try:
                     stream_mode = (
-                        StreamingMode.SSE
-                        if req.streaming
-                        else StreamingMode.NONE
+                        StreamingMode.SSE if req.streaming else StreamingMode.NONE
                     )
                     runner = await self.server.get_runner_async(req.app_name)
                     async with Aclosing(
-                            runner.run_async(
-                                user_id=req.user_id,
-                                session_id=req.session_id,
-                                new_message=req.new_message,
-                                state_delta=req.state_delta,
-                                run_config=RunConfig(streaming_mode=stream_mode),
-                                invocation_id=req.invocation_id,
-                            )
+                        runner.run_async(
+                            user_id=req.user_id,
+                            session_id=req.session_id,
+                            new_message=req.new_message,
+                            state_delta=req.state_delta,
+                            run_config=RunConfig(streaming_mode=stream_mode),
+                            invocation_id=req.invocation_id,
+                        )
                     ) as agen:
                         async for event in agen:
                             # ADK Web renders artifacts from `actions.artifactDelta`
@@ -191,9 +189,9 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                             # 2) a content-less "action-only" event carrying `artifactDelta`
                             events_to_stream = [event]
                             if (
-                                    event.actions.artifact_delta
-                                    and event.content
-                                    and event.content.parts
+                                event.actions.artifact_delta
+                                and event.content
+                                and event.content.parts
                             ):
                                 content_event = event.model_copy(deep=True)
                                 content_event.actions.artifact_delta = {}
@@ -230,9 +228,9 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
         routes = self.app.router.routes
         for i, r in enumerate(routes):
             if (
-                    getattr(r, "path", None) == "/run_sse"
-                    and "POST" in getattr(r, "methods", set())
-                    and getattr(r,"endpoint", None) == run_agent_sse
+                getattr(r, "path", None) == "/run_sse"
+                and "POST" in getattr(r, "methods", set())
+                and getattr(r, "endpoint", None) == run_agent_sse
             ):
                 routes.insert(0, routes.pop(i))
                 break
@@ -265,7 +263,9 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
             # Determine app_name from loader
             app_names = self.server.agent_loader.list_agents()
             if not app_names:
-                exception = HTTPException(status_code=404, detail="No agents configured")
+                exception = HTTPException(
+                    status_code=404, detail="No agents configured"
+                )
                 telemetry.trace_agent_server_finish(
                     path="/invoke", func_result="", exception=exception
                 )
@@ -293,7 +293,6 @@ class AgentkitAgentServerApp(BaseAgentkitApp):
                         text = ""
             content = types.UserContent(parts=[types.Part(text=text or "")])
 
-
             # Ensure session exists
             session = await self.server.session_service.get_session(
                 app_name=app_name, user_id=user_id, session_id=session_id

agentkit/apps/agent_server_app/telemetry.py

@@ -106,7 +106,9 @@ class Telemetry:
             if exception:
                 self.handle_exception(span, exception)
                 if getattr(exception, "status_code", None):
-                    attributes["error_type"] = f"{exception.__class__.__name__}_{exception.status_code}"
+                    attributes["error_type"] = (
+                        f"{exception.__class__.__name__}_{exception.status_code}"
+                    )
                 else:
                     attributes["error_type"] = exception.__class__.__name__
             # only record invoke request latency metrics

agentkit/platform/__init__.py

@@ -17,6 +17,7 @@ from __future__ import annotations
 from typing import Optional
 
 from agentkit.platform.configuration import VolcConfiguration, Endpoint, Credentials
+from agentkit.platform.console_urls import agentkit_enable_services_url
 from agentkit.platform.provider import CloudProvider
 from agentkit.platform.constants import DEFAULT_REGION_RULES
 
@@ -27,6 +28,7 @@ __all__ = [
     "CloudProvider",
     "resolve_endpoint",
     "resolve_credentials",
+    "agentkit_enable_services_url",
     "DEFAULT_REGION_RULES",
 ]
 

agentkit/platform/console_urls.py

@@ -0,0 +1,35 @@
+from __future__ import annotations
+
+import os
+from typing import Optional
+from urllib.parse import urlencode
+
+from agentkit.platform.configuration import VolcConfiguration
+from agentkit.platform.provider import CloudProvider
+
+
+def agentkit_enable_services_url(
+    *,
+    region: Optional[str] = None,
+    project_name: Optional[str] = None,
+    platform_config: Optional[VolcConfiguration] = None,
+) -> str:
+    if platform_config is not None and region is not None:
+        raise ValueError("Only one of 'region' or 'platform_config' can be provided.")
+    cfg = platform_config or VolcConfiguration(region=region)
+    ep = cfg.get_service_endpoint("agentkit")
+
+    base = (
+        "https://console.byteplus.com"
+        if cfg.provider == CloudProvider.BYTEPLUS
+        else "https://console.volcengine.com"
+    )
+
+    path = f"/agentkit/region:agentkit+{ep.region}/auth"
+
+    if cfg.provider != CloudProvider.BYTEPLUS:
+        return f"{base}{path}"
+
+    project = project_name or os.getenv("AGENTKIT_CONSOLE_PROJECT_NAME") or "default"
+    query = urlencode({"projectName": project})
+    return f"{base}{path}?{query}"

agentkit/toolkit/builders/ve_pipeline.py

@@ -64,6 +64,10 @@ class VeCPCRBuilderConfig(AutoSerializableMixin):
         default=None,
         metadata={"system": True, "description": "Resolved cloud provider"},
     )
+    agentkit_region: str = field(
+        default="",
+        metadata={"system": True, "description": "AgentKit service region"},
+    )
 
     tos_bucket: str = field(
         default=AUTO_CREATE_VE,
@@ -1054,8 +1058,28 @@ class VeCPCRBuilder(Builder):
 
         except Exception as e:
             if "AccountDisable" in str(e):
+                from agentkit.platform import (
+                    VolcConfiguration,
+                    agentkit_enable_services_url,
+                )
+
+                provider = getattr(config, "cloud_provider", None) or getattr(
+                    getattr(config, "common_config", None), "cloud_provider", None
+                )
+                region_hint = (
+                    getattr(config, "agentkit_region", None)
+                    or getattr(config, "cp_region", None)
+                    or getattr(config, "cr_region", None)
+                    or getattr(config, "tos_region", None)
+                )
+                url = agentkit_enable_services_url(
+                    platform_config=VolcConfiguration(
+                        region=region_hint or None, provider=provider or None
+                    )
+                )
                 raise Exception(
-                    "Tos Service is not enabled, please enable it in the console. Enable services at: https://console.volcengine.com/agentkit/region:agentkit+cn-beijing/auth"
+                    "Tos Service is not enabled, please enable it in the console. "
+                    f"Enable services at: {url}"
                 )
             if "TooManyBuckets" in str(e):
                 raise Exception(

agentkit/toolkit/executors/base_executor.py

@@ -295,7 +295,13 @@ class BaseExecutor:
 
             if missing:
                 self.logger.warning(f"Services not enabled: {missing}")
-                return PreflightResult(passed=False, missing_services=missing)
+                from agentkit.platform import agentkit_enable_services_url
+
+                return PreflightResult(
+                    passed=False,
+                    missing_services=missing,
+                    auth_url=agentkit_enable_services_url(region=region),
+                )
 
             self.logger.debug(f"All required services are enabled: {required_services}")
             return PreflightResult(passed=True, missing_services=[])

agentkit/toolkit/executors/lifecycle_executor.py

@@ -102,7 +102,13 @@ class LifecycleExecutor(BaseExecutor):
 
             if missing:
                 self.logger.warning(f"Services not enabled: {missing}")
-                return PreflightResult(passed=False, missing_services=missing)
+                from agentkit.platform import agentkit_enable_services_url
+
+                return PreflightResult(
+                    passed=False,
+                    missing_services=missing,
+                    auth_url=agentkit_enable_services_url(region=region),
+                )
 
             self.logger.debug(f"All required services are enabled: {required_list}")
             return PreflightResult(passed=True, missing_services=[])

agentkit/toolkit/models.py

@@ -200,9 +200,7 @@ class PreflightResult:
 
     passed: bool
     missing_services: List[str] = field(default_factory=list)
-    auth_url: str = (
-        "https://console.volcengine.com/agentkit/region:agentkit+cn-beijing/auth"
-    )
+    auth_url: str = ""
 
     @property
     def message(self) -> str:

agentkit/toolkit/strategies/cloud_strategy.py

@@ -301,6 +301,7 @@ class CloudStrategy(Strategy):
         return VeCPCRBuilderConfig(
             common_config=common_config,
             cloud_provider=resolved_provider,
+            agentkit_region=resolver.resolve("agentkit"),
             cp_region=resolver.resolve("cp"),
             tos_bucket=strategy_config.tos_bucket,
             tos_region=resolver.resolve("tos"),

agentkit/version.py

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

{agentkit_sdk_python-0.5.3.dist-info → agentkit_sdk_python-0.5.4.dist-info}/METADATA

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

@@ -1,5 +1,5 @@
 agentkit/__init__.py,sha256=l27ZMDslc3VhmmnPZJyrqVvTDoZ0LqhCtM5hw0caHcU,1021
-agentkit/version.py,sha256=DOxUVvvSRCgcuXlTM3S8olYS4xxtC8XooED4kCwayMA,653
+agentkit/version.py,sha256=PXDQfD2g5-dYwHEggBaX8jxmHJ1nN1Jx9_igWSk3h-8,653
 agentkit/apps/__init__.py,sha256=-oXTjxV3gejpJ8pffTUcUAXw0LfxKCYZJk-_hIJhtLM,1921
 agentkit/apps/base_app.py,sha256=3hZZExL1wyTGWveJEZZoqXN086MzmkVS_WU1vyulIWg,754
 agentkit/apps/utils.py,sha256=IzimIDmT6FS6PV9MLimPh46mq5zT-EjVmnYPxBdyEq0,1934
@@ -7,9 +7,9 @@ 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=L4pckLwDunLhCizuAlewxZj8z9s04ZzgMRRYG8peq20,14554
+agentkit/apps/agent_server_app/agent_server_app.py,sha256=vGcf5qsOx9zmmZWoEGTUwf_n8nE4XjkIx1vKLQU79_c,14488
 agentkit/apps/agent_server_app/middleware.py,sha256=2gLCRN-JYjYsTQzHqRBU69S2OI8gD5AdbrFeoGzDcM0,2962
-agentkit/apps/agent_server_app/telemetry.py,sha256=AMjQsuhY62CJROa4BfJcT0LMeb0-duppt0j1JUguvMY,4350
+agentkit/apps/agent_server_app/telemetry.py,sha256=FLJsxJHj4Vg3sfIZnWYm442_Dv-zmxTIPqRgvgwVIFQ,4398
 agentkit/apps/mcp_app/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7qWbo,634
 agentkit/apps/mcp_app/mcp_app.py,sha256=2ngFbKcC3OsWNPwU0aoS-D8JVBC7-mFpXqj7sgcvUY8,5886
 agentkit/apps/mcp_app/telemetry.py,sha256=ObzEoJrQK9Yu_3KsCslJejIhKkACZl5c87KHMFblvgI,3929
@@ -21,8 +21,9 @@ agentkit/client/__init__.py,sha256=WHKgNGkpYsJp0xD20drYcxAMDclwDqrfURqxEVv0V1Y,1
 agentkit/client/base_agentkit_client.py,sha256=OELc-y1aT-G2scAsS7p8gO4L4y9lA1ZqfDTLMt2Ajvc,2688
 agentkit/client/base_iam_client.py,sha256=F3ggcZjt14CAcgDYpncF2XGr1zTGnhaOzqEJl4NGHwU,2432
 agentkit/client/base_service_client.py,sha256=PvpsaCeZRYvhVcHh9ZKIVXV15wA_7Ip6W45i0CV0Tb0,8168
-agentkit/platform/__init__.py,sha256=UksDc9neLK8aDXgpk_ghR4BmKWhbM-CzczxQ0QXW7vM,3307
+agentkit/platform/__init__.py,sha256=8wS2qwWQfhm3axoQl0h8drEvGcE4vo14dbXMoWIycAU,3415
 agentkit/platform/configuration.py,sha256=Gbig_-u5p1nMsIZKhhqedYB9ID86mZbUj0aYBzQ9INc,16761
+agentkit/platform/console_urls.py,sha256=JcnjwyuxN5c4RuXcPevsxy6aInurVFtchqbibJBqjSk,1125
 agentkit/platform/constants.py,sha256=811u7xDM30YVj76kjuSoxjN48Qhd2j7GAYW3iIV8BxQ,3693
 agentkit/platform/context.py,sha256=eb6ub4nUhoV1hXawjSVdY7gfc5XcgHJjFIhrzOuQ2G0,1662
 agentkit/platform/provider.py,sha256=Yy5bYveVYQOJiR8_s_DW7rEDtK4cXZQSxLaUifDozuY,2245
@@ -52,12 +53,12 @@ agentkit/sdk/tools/types.py,sha256=KS00zJOMRyAxRIfein8-EpobZeKkfR4xKIXip7cU4CY,2
 agentkit/toolkit/__init__.py,sha256=VZkLRcjybKRHgWvGmLHAm0Gsb1FCUpKmKuJYfZmbfE0,1249
 agentkit/toolkit/context.py,sha256=6CPDjmUwB830JB0OrKGFVNUjeiJ17GPDtbbQBtt6L1w,7554
 agentkit/toolkit/errors.py,sha256=0FrOlMYtygZGwaxW0XBC0-aq98qfmbM4ylIH8qq0K74,4431
-agentkit/toolkit/models.py,sha256=2gm8FSeAuKKLXUx_e4V53Zcu66B6YK4FAXj2Q7e5GUA,18439
+agentkit/toolkit/models.py,sha256=EfIO_dyNMcwQuF5vsEJ1kAo0qc14DevHK21KTyA5WTU,18352
 agentkit/toolkit/reporter.py,sha256=q3r2q9A3Ajmd4IVuaN-FDV3imxwf1eEDXupFerx6JyY,11351
 agentkit/toolkit/builders/__init__.py,sha256=iZp7tjp5HLnvqMohk82be7ct5qbrwvv0oExeP0oZHN4,1593
 agentkit/toolkit/builders/base.py,sha256=ayejVFm2DDBSL-S7egIkwHoUaNViap5AKKFULMwNlBQ,3881
 agentkit/toolkit/builders/local_docker.py,sha256=GBTLoQAUNjpWosw7NXCTL8ssYHRulSoZZAQRgQsPgdY,18025
-agentkit/toolkit/builders/ve_pipeline.py,sha256=EN-3huKEX38cSQMF_LruxFctNE4xM_TadIo9IvF3zkk,70463
+agentkit/toolkit/builders/ve_pipeline.py,sha256=vJv5CDySrE4uLNsJJhZBOFmGfadITUsyJEAWlNB4bRM,71405
 agentkit/toolkit/cli/__init__.py,sha256=pkSabKw7_ai4NOo56pXKL40EcaxIDh6HYxPXOY7qWbo,634
 agentkit/toolkit/cli/__main__.py,sha256=arwJ1gHkaaoQAXb0ezo_FeyxiMqVy0FXujB2WOX_ohU,775
 agentkit/toolkit/cli/cli.py,sha256=cdv1YWArDH4PcoCc9lbPZgcj3_C70ZI6uyOSntUvIA8,3767
@@ -101,12 +102,12 @@ agentkit/toolkit/docker/dockerfile/__init__.py,sha256=AMVZzFZE8YVBOK13KGzwW1UuF7
 agentkit/toolkit/docker/dockerfile/manager.py,sha256=iGE5KY8T8G4Q0c5Mo3pWhxf-Pq001XzPC1rZrdQsVUI,16148
 agentkit/toolkit/docker/dockerfile/metadata.py,sha256=O4TCK1RYHd_5W3Vry3fd4HBKRG3LWvUFwStyL9fJHME,9278
 agentkit/toolkit/executors/__init__.py,sha256=ocEKR3rqbIrNtHukTfx1yBYKziIJWNcT0bBOt-CoWiA,1737
-agentkit/toolkit/executors/base_executor.py,sha256=sIZyOYC5tBsUpRizpOH4DOxW2-y7YEUU4KsGH3XeYRo,21356
+agentkit/toolkit/executors/base_executor.py,sha256=b-Mm_X_idN7hrL4d6wyER_ZQkIsdZEXvYxbgY8i6YlQ,21565
 agentkit/toolkit/executors/build_executor.py,sha256=BfTj-yVlmDJquzpsdXwl_npQJhkda-OY8Qb1cNH3B7M,7401
 agentkit/toolkit/executors/deploy_executor.py,sha256=aMF7uNznU_Toqomllel9Il86iSTycqmL76K_BfeTB6U,6694
 agentkit/toolkit/executors/init_executor.py,sha256=6p4H48viyKcIE4Lxj4otRHH2NJXmS-6hlB2KuAoau0c,31430
 agentkit/toolkit/executors/invoke_executor.py,sha256=xFqeFRAbt--PZeOzB6Sdij4JRTdOHt7bp4KoaCqOXY4,6545
-agentkit/toolkit/executors/lifecycle_executor.py,sha256=iwzeUADylhTz50pOpyl0web_f55510cA7oPY9LyBKSw,16719
+agentkit/toolkit/executors/lifecycle_executor.py,sha256=J4bBJ8QomKCttuF_Ruqk27PeAcx78Dyrrnnc2kLgMWI,16928
 agentkit/toolkit/executors/status_executor.py,sha256=mFta-aBUzvrlkpvjmDiyFs-hq9FZMZdHfXd0F7wTZxs,5353
 agentkit/toolkit/resources/samples/a2a.py,sha256=eKM56lrjnNLD35lT3Py_E7sN9eCGU4HdLOHrVz_xLwg,2867
 agentkit/toolkit/resources/samples/agent_server.py,sha256=tVD79gT8qhP3dLncKJPWhCjXNDKmRYB0DSyZFoRKiS0,2282
@@ -158,7 +159,7 @@ agentkit/toolkit/sdk/bindings/__init__.py,sha256=sZVayHUi0CA2dwn1_fovKdb8z_y3k1-
 agentkit/toolkit/sdk/bindings/memory.py,sha256=85feCymQBkPA6djKOLiO4YhGxw1e89zENa89DWp_BTY,6068
 agentkit/toolkit/strategies/__init__.py,sha256=vgtqGePsuki2nYwd3ox3xILSrdijQmTaBMogygJ6DMc,1243
 agentkit/toolkit/strategies/base_strategy.py,sha256=l9att4i-YKJc4vUNlecKCu3ZQWQRgkdiJqovXiY8OWc,6753
-agentkit/toolkit/strategies/cloud_strategy.py,sha256=tjvo9hvHY_YhsD2126EYLHorjYEHLc0M31YDHDTwh0w,13690
+agentkit/toolkit/strategies/cloud_strategy.py,sha256=Ixsjs3JXn9ykvt49qJtuMM4Wb8hiaARLuM5q0Wq9VUM,13748
 agentkit/toolkit/strategies/hybrid_strategy.py,sha256=V6ltVAR6k2320Qj4OrVTN-qtGVF9zi8bPr62YtrEZ5A,20082
 agentkit/toolkit/strategies/local_strategy.py,sha256=xK1kJwjY3QLpLBhlfAeYtSwjegfuzGWgi91yNavbqBQ,9950
 agentkit/toolkit/utils/__init__.py,sha256=uSaoOXTyarlItNRdkmpXAoUEDZDwFBSB146W9hJ_2_4,739
@@ -180,9 +181,9 @@ agentkit/utils/misc.py,sha256=FXMlcupj3KzxfJvMYFp0uTQk3FRMx8FiEXkYiy-UQ8w,3437
 agentkit/utils/request.py,sha256=IVoat3EavR9rQ_fXi0eA2rZPCJ9lVZAXXn7uIIX6bY4,1614
 agentkit/utils/template_utils.py,sha256=Qjg9V6dEpjAd_yYezIIPwuDsDeeMmp6XTMn5ZECifNc,6336
 agentkit/utils/ve_sign.py,sha256=nrYiS2bYncvWKQVpVMzwYwPDfkA0OJS7LO9l_wfW0xc,9310
-agentkit_sdk_python-0.5.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-agentkit_sdk_python-0.5.3.dist-info/METADATA,sha256=Srqvt6k-G1KopdULfz-ZJ8wTOlrgTbPxCzym6w_Yozw,19394
-agentkit_sdk_python-0.5.3.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
-agentkit_sdk_python-0.5.3.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
-agentkit_sdk_python-0.5.3.dist-info/top_level.txt,sha256=ipy8JF-QQ-V0C1oRFLxsyaW8zwrasfJ-zjAh9vgOc7U,9
-agentkit_sdk_python-0.5.3.dist-info/RECORD,,
+agentkit_sdk_python-0.5.4.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+agentkit_sdk_python-0.5.4.dist-info/METADATA,sha256=4-AmxEStSPpVSwVgLTUwr3E0VoOkdDhAxf3VQwSp_Yc,19394
+agentkit_sdk_python-0.5.4.dist-info/WHEEL,sha256=YCfwYGOYMi5Jhw2fU4yNgwErybb2IX5PEwBKV4ZbdBo,91
+agentkit_sdk_python-0.5.4.dist-info/entry_points.txt,sha256=fhzZUsvsLXeB4mPaa0SBQiTBqFT404uDAKPaePAOUAE,58
+agentkit_sdk_python-0.5.4.dist-info/top_level.txt,sha256=ipy8JF-QQ-V0C1oRFLxsyaW8zwrasfJ-zjAh9vgOc7U,9
+agentkit_sdk_python-0.5.4.dist-info/RECORD,,