kagent-adk 0.6.14__tar.gz → 0.6.16__tar.gz

kagent_adk-0.6.16/.python-version

@@ -0,0 +1 @@
+3.13.7

{kagent_adk-0.6.14 → kagent_adk-0.6.16}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kagent-adk
-Version: 0.6.14
+Version: 0.6.16
 Summary: kagent-adk is an sdk for integrating adk agents with kagent
 Requires-Python: >=3.12.11
 Requires-Dist: a2a-sdk>=0.3.1

{kagent_adk-0.6.14 → kagent_adk-0.6.16}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kagent-adk"
-version = "0.6.14"
+version = "0.6.16"
 description = "kagent-adk is an sdk for integrating adk agents with kagent"
 readme = "README.md"
 requires-python = ">=3.12.11"

{kagent_adk-0.6.14 → kagent_adk-0.6.16}/src/kagent/adk/_agent_executor.py

@@ -22,6 +22,7 @@ from a2a.types import (
     TextPart,
 )
 from google.adk.runners import Runner
+from google.adk.utils.context_utils import Aclosing
 from opentelemetry import trace
 from pydantic import BaseModel
 from typing_extensions import override
@@ -145,17 +146,6 @@ class A2aAgentExecutor(AgentExecutor):
                 )
             except Exception as enqueue_error:
                 logger.error("Failed to publish failure event: %s", enqueue_error, exc_info=True)
-        finally:
-            # Shield cleanup from external cancellation so toolsets (e.g., MCP) can
-            # gracefully close their sessions without being torn down mid-flight.
-            try:
-                await asyncio.wait_for(asyncio.shield(runner.close()), timeout=15.0)
-            except asyncio.CancelledError:
-                # Suppress cancellation during cleanup to avoid noisy tracebacks
-                # from libraries that assume non-cancelled close semantics.
-                logger.warning("Runner.close() was cancelled; suppressing during cleanup")
-            except Exception as close_error:
-                logger.error("Error during runner.close(): %s", close_error, exc_info=True)
 
     async def _handle_request(
         self,
@@ -203,12 +193,13 @@ class A2aAgentExecutor(AgentExecutor):
         )
 
         task_result_aggregator = TaskResultAggregator()
-        async for adk_event in runner.run_async(**run_args):
-            for a2a_event in convert_event_to_a2a_events(
-                adk_event, invocation_context, context.task_id, context.context_id
-            ):
-                task_result_aggregator.process_event(a2a_event)
-                await event_queue.enqueue_event(a2a_event)
+        async with Aclosing(runner.run_async(**run_args)) as agen:
+            async for adk_event in agen:
+                for a2a_event in convert_event_to_a2a_events(
+                    adk_event, invocation_context, context.task_id, context.context_id
+                ):
+                    task_result_aggregator.process_event(a2a_event)
+                    await event_queue.enqueue_event(a2a_event)
 
         # publish the task result event - this is final
         if (

{kagent_adk-0.6.14 → kagent_adk-0.6.16}/src/kagent/adk/models/_openai.py

@@ -395,7 +395,7 @@ class AzureOpenAI(BaseOpenAI):
     @cached_property
     def _client(self) -> AsyncAzureOpenAI:
         """Get the Azure OpenAI client."""
-        api_version = self.api_version or os.environ.get("AZURE_OPENAI_API_VERSION", "2024-02-15-preview")
+        api_version = self.api_version or os.environ.get("OPENAI_API_VERSION", "2024-02-15-preview")
         azure_endpoint = self.azure_endpoint or os.environ.get("AZURE_OPENAI_ENDPOINT")
         api_key = self.api_key or os.environ.get("AZURE_OPENAI_API_KEY")
 

{kagent_adk-0.6.14 → kagent_adk-0.6.16}/src/kagent/adk/types.py

@@ -1,8 +1,7 @@
-import httpx
 import logging
+from typing import Any, Literal, Union
 
-from typing import Literal, Any, Union
-
+import httpx
 from google.adk.agents import Agent
 from google.adk.agents.base_agent import BaseAgent
 from google.adk.agents.llm_agent import ToolUnion

kagent_adk-0.6.14/.python-version

@@ -1 +0,0 @@
-3.13.5