fix: fixed some stability and experience issues, added retry and Unicode support (#33)

Author: YumengBao

agentkit/apps/agent_server_app/agent_server_app.py

@@ -104,8 +104,7 @@ async def _invoke_compat(request: Request):
             )
             session_id = (
                 headers.get("session_id")
-                or headers.get("x-session-id")
-                or "agentkit_sample_session"
+                or ""
             )
 
             # Determine app_name from loader

agentkit/apps/agent_server_app/middleware.py

@@ -43,8 +43,8 @@ async def __call__(self, scope, receive, send):
         }
 
         # Currently unable to retrieve user_id and session_id from headers; keep logic for future use
-        user_id = headers.get("user_id") or headers.get("x-user-id") or ""
-        session_id = headers.get("session_id") or headers.get("x-session-id") or ""
+        user_id = headers.get("user_id")
+        session_id = headers.get("session_id")
         headers["user_id"] = user_id
         headers["session_id"] = session_id
         telemetry.trace_agent_server(

agentkit/apps/agent_server_app/telemetry.py

@@ -72,10 +72,10 @@ def trace_agent_server(
             value=safe_serialize_to_json_string(headers),
         )
 
-        session_id = headers.get("session_id") or headers.get("x-session-id") or ""
+        session_id = headers.get("session_id")
         if session_id:
             span.set_attribute(key="gen_ai.session.id", value=session_id)
-        user_id = headers.get("user_id") or headers.get("x-user-id") or ""
+        user_id = headers.get("user_id")
         if user_id:
             span.set_attribute(key="gen_ai.user.id", value=user_id)
 

agentkit/toolkit/builders/ve_pipeline.py

@@ -32,7 +32,11 @@
 from agentkit.toolkit.models import BuildResult, ImageInfo
 from agentkit.toolkit.reporter import Reporter
 from agentkit.toolkit.errors import ErrorCode
-from agentkit.utils.misc import generate_random_id, calculate_nonlinear_progress
+from agentkit.utils.misc import (
+    generate_random_id,
+    calculate_nonlinear_progress,
+    retry,
+)
 from agentkit.toolkit.volcengine.services import CRServiceConfig
 from .base import Builder
 
@@ -1303,11 +1307,12 @@ def download_and_show_logs(run_id: str):
 
                 while True:
                     try:
-                        # Get pipeline run status
-                        status = cp_client.get_pipeline_run_status(
-                            workspace_id=workspace_id,
-                            pipeline_id=pipeline_id,
-                            run_id=run_id,
+                        status = retry(
+                            lambda: cp_client.get_pipeline_run_status(
+                                workspace_id=workspace_id,
+                                pipeline_id=pipeline_id,
+                                run_id=run_id,
+                            )
                         )
 
                         # Update progress description

agentkit/toolkit/config/config.py

@@ -297,7 +297,13 @@ def _save_config(self):
         """Save configuration file"""
         os.makedirs(self.config_path.parent, exist_ok=True)
         with open(self.config_path, "w", encoding="utf-8") as f:
-            yaml.dump(self._data, f, default_flow_style=False, sort_keys=False)
+            yaml.dump(
+                self._data,
+                f,
+                default_flow_style=False,
+                sort_keys=False,
+                allow_unicode=True,
+            )
 
     def get_common_config(self) -> CommonConfig:
         """Get common configuration"""

agentkit/toolkit/runners/ve_agentkit.py

@@ -39,6 +39,7 @@
     generate_apikey_name,
     generate_client_token,
     calculate_nonlinear_progress,
+    retry,
 )
 from agentkit.sdk.runtime.client import AgentkitRuntimeClient
 from agentkit.toolkit.volcengine.iam import VeIAM
@@ -834,8 +835,10 @@ def _wait_for_runtime_status_multiple(
         # Use reporter.long_task() for progress tracking
         with self.reporter.long_task(task_description, total=total_time) as task:
             while True:
-                runtime = self.agentkit_runtime_client.get_runtime(
-                    runtime_types.GetRuntimeRequest(runtime_id=runtime_id)
+                runtime = retry(
+                    lambda: self.agentkit_runtime_client.get_runtime(
+                        runtime_types.GetRuntimeRequest(runtime_id=runtime_id)
+                    )
                 )
 
                 # Check if target status reached

agentkit/toolkit/utils/__init__.py

@@ -16,4 +16,5 @@
 
 from .agent_parser import AgentParser
 
+
 __all__ = ["AgentParser"]

agentkit/utils/misc.py

@@ -15,6 +15,10 @@
 import math
 import string
 import random
+import time
+from typing import Callable, TypeVar
+
+T = TypeVar("T")
 
 
 def generate_random_id(length=8):
@@ -102,3 +106,17 @@ def calculate_nonlinear_progress(
     """
     progress = max_time * (1 - math.exp(-elapsed / expected_time))
     return min(progress, max_time * max_ratio)
+
+
+def retry(
+    func: Callable[[], T],
+    retries: int = 3,
+    delay: float = 1.0,
+) -> T:
+    for attempt in range(retries):
+        try:
+            return func()
+        except Exception:  # noqa: BLE001
+            if attempt == retries - 1:
+                raise
+            time.sleep(delay)

agentkit/version.py

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

pyproject.toml

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