fix: resolve fixed hint url in byteplus issue (#60)

innsd · web-flow · commit 761e49106750 · 2026-02-28T18:15:23.000+08:00
diff --git a/agentkit/apps/agent_server_app/agent_server_app.py b/agentkit/apps/agent_server_app/agent_server_app.py
@@ -169,20 +169,18 @@ async def run_agent_sse(req: RunAgentRequest) -> StreamingResponse:
             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 @@ async def event_generator():
                             # 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 @@ async def event_generator():
         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 @@ async def _invoke_compat(request: Request):
             # 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 @@ async def _invoke_compat(request: Request):
                         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
diff --git a/agentkit/apps/agent_server_app/telemetry.py b/agentkit/apps/agent_server_app/telemetry.py
@@ -106,7 +106,9 @@ def trace_agent_server_finish(
             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
diff --git a/agentkit/platform/__init__.py b/agentkit/platform/__init__.py
@@ -17,6 +17,7 @@
 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 @@
     "CloudProvider",
     "resolve_endpoint",
     "resolve_credentials",
+    "agentkit_enable_services_url",
     "DEFAULT_REGION_RULES",
 ]
 
diff --git a/agentkit/platform/console_urls.py b/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}"
diff --git a/agentkit/toolkit/builders/ve_pipeline.py b/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 @@ def check_owned() -> bool:
 
         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(
diff --git a/agentkit/toolkit/executors/base_executor.py b/agentkit/toolkit/executors/base_executor.py
@@ -295,7 +295,13 @@ def _preflight_check(
 
             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=[])
diff --git a/agentkit/toolkit/executors/lifecycle_executor.py b/agentkit/toolkit/executors/lifecycle_executor.py
@@ -102,7 +102,13 @@ def _combined_preflight_check(
 
             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=[])
diff --git a/agentkit/toolkit/models.py b/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:
diff --git a/agentkit/toolkit/strategies/cloud_strategy.py b/agentkit/toolkit/strategies/cloud_strategy.py
@@ -301,6 +301,7 @@ def _to_builder_config(
         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"),
diff --git a/agentkit/version.py b/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"
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "agentkit-sdk-python"
-version = "0.5.3"
+version = "0.5.4"
 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"
diff --git a/tests/toolkit/builders/test_ve_pipeline_cr_domain.py b/tests/toolkit/builders/test_ve_pipeline_cr_domain.py
@@ -88,3 +88,27 @@ def test_resolve_cr_domain_prefers_explicit_builder_provider(monkeypatch):
 
     domain = builder._resolve_cr_domain(cfg, "ap-southeast-1")
     assert domain == "ins-ap-southeast-1.cr.bytepluses.com"
+
+
+def test_cloud_strategy_sets_agentkit_region_in_builder_config(monkeypatch):
+    monkeypatch.delenv("AGENTKIT_CLOUD_PROVIDER", raising=False)
+    monkeypatch.delenv("CLOUD_PROVIDER", raising=False)
+
+    from agentkit.toolkit.config import CommonConfig, CloudStrategyConfig
+    from agentkit.toolkit.config.region_resolver import RegionConfigResolver
+    from agentkit.toolkit.strategies.cloud_strategy import CloudStrategy
+
+    strategy_config = CloudStrategyConfig.from_dict(
+        {"region": "ap-southeast-1"}, skip_render=True
+    )
+    common_config = CommonConfig(
+        agent_name="agentkit-app",
+        entry_point="agent.py",
+        launch_type="cloud",
+    )
+
+    strategy = CloudStrategy(config_manager=None, reporter=None)
+    builder_cfg = strategy._to_builder_config(common_config, strategy_config)
+
+    resolver = RegionConfigResolver.from_strategy_config(strategy_config)
+    assert builder_cfg.agentkit_region == resolver.resolve("agentkit")
diff --git a/tests/toolkit/test_lifecycle_launch_preflight_platform_context.py b/tests/toolkit/test_lifecycle_launch_preflight_platform_context.py
@@ -89,3 +89,45 @@ def _fake_combined_preflight_check(*_args, **_kwargs):
 
     after = VolcConfiguration()
     assert after.provider.value == "volcengine"
+
+
+def test_agentkit_enable_services_url_byteplus_region(monkeypatch) -> None:
+    from agentkit.platform.context import default_cloud_provider
+    from agentkit.platform import agentkit_enable_services_url
+
+    monkeypatch.delenv("AGENTKIT_CONSOLE_PROJECT_NAME", raising=False)
+
+    with default_cloud_provider("byteplus"):
+        url = agentkit_enable_services_url(region="ap-southeast-1")
+
+    assert (
+        url
+        == "https://console.byteplus.com/agentkit/region:agentkit+ap-southeast-1/auth?projectName=default"
+    )
+
+
+def test_preflight_sets_auth_url_by_provider_and_region(monkeypatch) -> None:
+    import agentkit.sdk.account.client as account_client_mod
+    from agentkit.platform.context import default_cloud_provider
+    from agentkit.toolkit.executors.build_executor import BuildExecutor
+    from agentkit.toolkit.reporter import SilentReporter
+
+    class _FakeAccountClient:
+        def __init__(self, region=None):
+            self.region = region
+
+        def get_services_status(self, service_names):
+            return {name: "Disabled" for name in service_names}
+
+    monkeypatch.setattr(account_client_mod, "AgentkitAccountClient", _FakeAccountClient)
+    monkeypatch.delenv("AGENTKIT_CONSOLE_PROJECT_NAME", raising=False)
+
+    ex = BuildExecutor(reporter=SilentReporter())
+    with default_cloud_provider("byteplus"):
+        result = ex._preflight_check("build", "cloud", region="ap-southeast-1")
+
+    assert result.passed is False
+    assert (
+        result.auth_url
+        == "https://console.byteplus.com/agentkit/region:agentkit+ap-southeast-1/auth?projectName=default"
+    )
