zai-adk-python-preview 0.1.0__tar.gz → 0.1.2__tar.gz

{zai_adk_python_preview-0.1.0 → zai_adk_python_preview-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zai-adk-python-preview
-Version: 0.1.0
+Version: 0.1.2
 Summary: A Python SDK for building AI agents with LLM integration
 Project-URL: Homepage, https://github.com/zai-adk-python/zai-adk-python
 Project-URL: Documentation, https://github.com/zai-adk-python/zai-adk-python#readme

{zai_adk_python_preview-0.1.0 → zai_adk_python_preview-0.1.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "zai-adk-python-preview"
-version = "0.1.0"
+version = "0.1.2"
 description = "A Python SDK for building AI agents with LLM integration"
 readme = "README.md"
 requires-python = ">=3.11"

{zai_adk_python_preview-0.1.0 → zai_adk_python_preview-0.1.2}/uv.lock

@@ -439,17 +439,6 @@ wheels = [
     { url = "https://files.pythonhosted.org/packages/55/e2/2537ebcff11c1ee1ff17d8d0b6f4db75873e3b0fb32c2d4a2ee31ecb310a/docstring_parser-0.17.0-py3-none-any.whl", hash = "sha256:cf2569abd23dce8099b300f9b4fa8191e9582dda731fd533daf54c4551658708", size = 36896, upload-time = "2025-07-21T07:35:00.684Z" },
 ]
 
-[[package]]
-name = "dotenv"
-version = "0.9.9"
-source = { registry = "https://pypi.org/simple" }
-dependencies = [
-    { name = "python-dotenv" },
-]
-wheels = [
-    { url = "https://files.pythonhosted.org/packages/b2/b7/545d2c10c1fc15e48653c91efde329a790f2eecfbbf2bd16003b5db2bab0/dotenv-0.9.9-py2.py3-none-any.whl", hash = "sha256:29cf74a087b31dafdb5a446b6d7e11cbce8ed2741540e2339c69fbef92c94ce9", size = 1892, upload-time = "2025-02-19T22:15:01.647Z" },
-]
-
 [[package]]
 name = "fastapi"
 version = "0.128.0"
@@ -2277,21 +2266,28 @@ wheels = [
 ]
 
 [[package]]
-name = "zai-adk-python"
-version = "0.1.0"
-source = { virtual = "." }
+name = "zai-adk-python-preview"
+version = "0.1.2"
+source = { editable = "." }
 dependencies = [
     { name = "aiohttp" },
     { name = "anthropic" },
     { name = "anyio" },
-    { name = "dotenv" },
     { name = "httpx", extra = ["socks"] },
     { name = "mcp" },
     { name = "openai" },
     { name = "pydantic" },
+    { name = "python-dotenv" },
 ]
 
 [package.optional-dependencies]
+all = [
+    { name = "anthropic" },
+    { name = "fastapi" },
+    { name = "lmnr" },
+    { name = "openai" },
+    { name = "uvicorn", extra = ["standard"] },
+]
 anthropic = [
     { name = "anthropic" },
 ]
@@ -2322,7 +2318,6 @@ requires-dist = [
     { name = "anthropic", specifier = ">=0.40.0" },
     { name = "anthropic", marker = "extra == 'anthropic'", specifier = ">=0.40.0" },
     { name = "anyio", specifier = ">=4.12.1" },
-    { name = "dotenv", specifier = ">=0.9.9" },
     { name = "fastapi", marker = "extra == 'server'", specifier = ">=0.115.0" },
     { name = "httpx", extras = ["socks"], specifier = ">=0.28.1" },
     { name = "lmnr", marker = "extra == 'observability'", specifier = ">=0.4.0" },
@@ -2332,9 +2327,11 @@ requires-dist = [
     { name = "pydantic", specifier = ">=2.0.0" },
     { name = "pytest", marker = "extra == 'test'", specifier = ">=8.0.0" },
     { name = "pytest-asyncio", marker = "extra == 'test'", specifier = ">=0.23.0" },
+    { name = "python-dotenv", specifier = ">=0.9.9" },
     { name = "uvicorn", extras = ["standard"], marker = "extra == 'server'", specifier = ">=0.32.0" },
+    { name = "zai-adk-python-preview", extras = ["anthropic", "openai", "observability", "server"], marker = "extra == 'all'" },
 ]
-provides-extras = ["anthropic", "openai", "observability", "server", "agui", "test"]
+provides-extras = ["anthropic", "openai", "observability", "server", "test", "all"]
 
 [package.metadata.requires-dev]
 dev = [

{zai_adk_python_preview-0.1.0 → zai_adk_python_preview-0.1.2}/zai_adk/agent/service.py

@@ -7,6 +7,7 @@ import json
 import logging
 import random
 import re
+import time
 from collections.abc import AsyncIterator
 from contextlib import nullcontext
 from dataclasses import dataclass, field
@@ -249,7 +250,6 @@ class Agent:
             # No sandbox provided - create default LocalSandbox with cwd as work_dir
             default_config = SandboxConfig(kind="local", work_dir=str(Path.cwd()))
             self._sandbox = SandboxFactory().create(default_config)
-            logger.debug(f"[Agent] Created default LocalSandbox with work_dir={self._sandbox.work_dir}")
         elif isinstance(self.sandbox, SandboxConfig):
             # It's a SandboxConfig, need to create the sandbox
             self._sandbox = SandboxFactory().create(self.sandbox)
@@ -278,11 +278,9 @@ class Agent:
         if self.skills_manager is not None:
             # Use externally provided SkillsManager
             self._skills_manager = self.skills_manager
-            logger.info("[Agent] __post_init__: Using external SkillsManager")
         elif self.skills_dir is not None:
             # Create SkillsManager from skills_dir
             self._skills_manager = SkillsManager(skills_dir=self.skills_dir)
-            logger.info(f"[Agent] __post_init__: Created SkillsManager from skills_dir: {self.skills_dir}")
         else:
             # No skills configured
             self._skills_manager = None
@@ -329,26 +327,14 @@ class Agent:
             agent = Agent(llm=llm, tools=[tools], skills_dir="./skills")
             await agent.async_initialize()
         """
-        logger.info("[Agent] async_initialize: Starting")
-
         # Initialize skills if skills_manager is configured
         if self._skills_manager is not None:
             try:
-                logger.info("[Agent] async_initialize: Scanning skills directory...")
                 skills_metadata = await self._skills_manager.scan()
-
                 if skills_metadata:
-                    logger.info(f"[Agent] async_initialize: Found {len(skills_metadata)} skill(s)")
                     self._update_system_prompt_with_skills(skills_metadata)
-                    logger.info("[Agent] async_initialize: ✓ Injected skills metadata into system prompt")
-                else:
-                    logger.info("[Agent] async_initialize: No skills found, skipping injection")
             except Exception as e:
-                logger.error(f"[Agent] async_initialize: Failed to initialize skills: {e}")
-        else:
-            logger.info("[Agent] async_initialize: No SkillsManager configured, skipping")
-
-        logger.info("[Agent] async_initialize: Complete")
+                logger.error(f"[Agent] Failed to initialize skills: {e}")
 
     async def refresh_skills_metadata(self) -> None:
         """Refresh skills metadata for hot reload support.
@@ -365,10 +351,7 @@ class Agent:
         Example:
             await agent.refresh_skills_metadata()
         """
-        logger.info("[Agent] refresh_skills_metadata: Starting")
-
         if self._skills_manager is None:
-            logger.info("[Agent] refresh_skills_metadata: No SkillsManager configured, skipping")
             return
 
         try:
@@ -378,20 +361,13 @@ class Agent:
             old_prompt = self.system_prompt
             self.system_prompt = re.sub(skills_pattern, '', self.system_prompt).strip()
 
-            if old_prompt != self.system_prompt and "<skills_system" in old_prompt:
-                logger.info("[Agent] refresh_skills_metadata: Removed old skills section")
-
             # Re-scan and re-inject
             skills_metadata = await self._skills_manager.scan()
-
             if skills_metadata:
                 self._update_system_prompt_with_skills(skills_metadata)
-                logger.info(f"[Agent] refresh_skills_metadata: Refreshed with {len(skills_metadata)} skill(s)")
-            else:
-                logger.info("[Agent] refresh_skills_metadata: No skills found after scan")
 
         except Exception as e:
-            logger.error(f"[Agent] refresh_skills_metadata: Failed: {e}")
+            logger.error(f"[Agent] Failed to refresh skills: {e}")
 
     @property
     def tool_definitions(self) -> list[ToolDefinition]:
@@ -540,10 +516,6 @@ class Agent:
             messages_to_destroy = messages[:-keep_count] if keep_count > 0 else messages
 
             for msg in messages_to_destroy:
-                # Log which message is being destroyed
-                logger.debug(
-                    f"🗑️  Destroying ephemeral: {msg.tool_name} (keeping last {keep_count})"
-                )
 
                 # Save to disk if storage path is configured
                 if self.ephemeral_storage_path is not None:
@@ -582,6 +554,16 @@ class Agent:
                 is_error=True,
             )
 
+        # Parse arguments for logging
+        try:
+            args = json.loads(tool_call.function.arguments)
+            args_str = json.dumps(args)[:100]  # Truncate for log
+        except json.JSONDecodeError:
+            args_str = tool_call.function.arguments[:100]
+
+        logger.info(f"[Agent] Tool call: {tool_name}({args_str})")
+        start_time = time.time()
+
         # Create Laminar span for tool execution
         if Laminar is not None:
             span_context = Laminar.start_as_current_span(
@@ -608,6 +590,9 @@ class Agent:
                     _overrides=self._get_dependency_overrides(), **args
                 )
 
+                elapsed = time.time() - start_time
+                logger.info(f"[Agent] Tool {tool_name} completed in {elapsed:.2f}s")
+
                 # Check if the tool is marked as ephemeral (can be bool or int for keep count)
                 is_ephemeral = bool(tool.ephemeral)  # Convert int to bool (2 -> True)
 
@@ -677,7 +662,7 @@ class Agent:
         """Log retry and wait before next attempt."""
         delay = self._calculate_retry_delay(attempt)
         logger.warning(
-            f"⚠️ Got {reason}, retrying in {delay:.1f}s... "
+            f"[Agent] LLM call failed: {reason}, retrying in {delay:.1f}s "
             f"(attempt {attempt + 1}/{self.llm_max_retries})"
         )
         await asyncio.sleep(delay)
@@ -726,9 +711,9 @@ class Agent:
 
         Includes retry logic with exponential backoff for LLM errors.
         """
-        last_error: Exception | None = None
 
         for attempt in range(self.llm_max_retries):
+            start_time = time.time()
             try:
                 response = await self.llm.ainvoke(
                     messages=self._messages,
@@ -736,6 +721,9 @@ class Agent:
                     tool_choice=self.tool_choice if self.tools else None,
                 )
 
+                elapsed = time.time() - start_time
+                logger.info(f"[Agent] LLM call completed in {elapsed:.2f}s")
+
                 # Track token usage
                 if response.usage:
                     self._token_cost.add_usage(self.llm.model, response.usage)