agentkit-sdk-python 0.5.3__tar.gz → 0.5.5__tar.gz

{agentkit_sdk_python-0.5.3/agentkit_sdk_python.egg-info → agentkit_sdk_python-0.5.5}/PKG-INFO

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

@@ -372,7 +372,7 @@ def _interactive_config(config_file: Optional[str] = None):
     )
 
     common_config = create_common_config_interactively(
-        config.get_common_config().to_dict()
+        (config.get_raw_data() or {}).get("common", {})
     )
     config.update_common_config(common_config)
 

{agentkit_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/agentkit/toolkit/cli/interactive_config.py

@@ -178,6 +178,7 @@ class AutoPromptGenerator:
         dataclass_type: type,
         existing_config: Optional[Dict[str, Any]] = None,
         context: Optional[Dict[str, Any]] = None,
+        carry_over_config: Optional[Dict[str, Any]] = None,
     ) -> Dict[str, Any]:
         self.current_dataclass_type = dataclass_type
         if not is_dataclass(dataclass_type):
@@ -185,6 +186,12 @@ class AutoPromptGenerator:
 
         config = {}
         existing_config = existing_config or {}
+        if carry_over_config is not None and not isinstance(carry_over_config, dict):
+            raise TypeError(
+                "carry_over_config must be a dict when provided; "
+                f"got {type(carry_over_config).__name__}"
+            )
+        carry_over = existing_config if carry_over_config is None else carry_over_config
 
         # Get dataclass metadata
         # Try to get from class attributes; if not found, create instance to get field values
@@ -267,8 +274,8 @@ class AutoPromptGenerator:
             if field.metadata.get("hidden", False) or field.metadata.get(
                 "system", False
             ):
-                if field_name in existing_config:
-                    config[field_name] = existing_config[field_name]
+                if isinstance(carry_over, dict) and field_name in carry_over:
+                    config[field_name] = carry_over[field_name]
 
         # Filter out MISSING values
         filtered_config = {}
@@ -1227,8 +1234,14 @@ def generate_config_from_dataclass(
     dataclass_type: type,
     existing_config: Optional[Dict[str, Any]] = None,
     context: Optional[Dict[str, Any]] = None,
+    carry_over_config: Optional[Dict[str, Any]] = None,
 ) -> Dict[str, Any]:
-    return auto_prompt.generate_config(dataclass_type, existing_config, context=context)
+    return auto_prompt.generate_config(
+        dataclass_type,
+        existing_config,
+        context=context,
+        carry_over_config=carry_over_config,
+    )
 
 
 def create_common_config_interactively(
@@ -1250,6 +1263,11 @@ def create_common_config_interactively(
     """
     from agentkit.toolkit.config import CommonConfig
 
-    existing = CommonConfig.from_dict(existing_config or {})
-    config_dict = auto_prompt.generate_config(CommonConfig, existing.to_dict())
+    raw_existing_config = existing_config or {}
+    existing = CommonConfig.from_dict(raw_existing_config)
+    config_dict = auto_prompt.generate_config(
+        CommonConfig,
+        existing.to_dict(),
+        carry_over_config=raw_existing_config,
+    )
     return CommonConfig.from_dict(config_dict)

{agentkit_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/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_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/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_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/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_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/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_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/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.5"

{agentkit_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5/agentkit_sdk_python.egg-info}/PKG-INFO

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

@@ -27,6 +27,7 @@ agentkit/client/base_iam_client.py
 agentkit/client/base_service_client.py
 agentkit/platform/__init__.py
 agentkit/platform/configuration.py
+agentkit/platform/console_urls.py
 agentkit/platform/constants.py
 agentkit/platform/context.py
 agentkit/platform/provider.py

{agentkit_sdk_python-0.5.3 → agentkit_sdk_python-0.5.5}/pyproject.toml

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