kagent-adk 0.6.15__py3-none-any.whl → 0.6.16__py3-none-any.whl

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/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.15.dist-info → kagent_adk-0.6.16.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kagent-adk
-Version: 0.6.15
+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.15.dist-info → kagent_adk-0.6.16.dist-info}/RECORD

@@ -1,10 +1,10 @@
 kagent/adk/__init__.py,sha256=Cam9hwhl6Z1tZ3WJIB1KATX24MwRMpiyBLYF64DTqQI,182
 kagent/adk/_a2a.py,sha256=HvduKJlfuUIKzYuPs_U_9A-xuhcMdQutZTzdLNaGORc,4406
-kagent/adk/_agent_executor.py,sha256=rxNVdrcS4L5Kwe8oF7mLUnD1KOYhe6vXK-si0O-KQ9o,10730
+kagent/adk/_agent_executor.py,sha256=f6jYkEvp-JEIW34T09Md4B31Di-GblnisqPs7pGFXDY,10136
 kagent/adk/_session_service.py,sha256=Hm2P0NgH8LnfVHmiT-z2AHPHAxPrT9W_6vNkFfIzRK8,5644
 kagent/adk/_token.py,sha256=OL46m7U5vUTby1WWjVB7Jqzig4TWddzoAmLVLlfSdAg,2515
 kagent/adk/cli.py,sha256=Sdw9FXnUmeHfgJhdRDq1zpoNMf6GM7x35ZDz_OkQgJg,2963
-kagent/adk/types.py,sha256=E0HxxrSrpuZVWE7F9lbNY4N2276QK3qCwhuBdDbhMTY,4911
+kagent/adk/types.py,sha256=L9Q79zSSLUXA6ycW0c0iIJcQRzbfNGBlS2A0RXtsC1I,4910
 kagent/adk/converters/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kagent/adk/converters/error_mappings.py,sha256=1KUJPS8VrcaTv6yUKb5Whg-S2XX8YGJmtTIeZqnqvuw,2769
 kagent/adk/converters/event_converter.py,sha256=WKQRqREB11TbgGp6U_--mmukvJJgew6-VEkrGBqGVA4,10519
@@ -12,7 +12,7 @@ kagent/adk/converters/part_converter.py,sha256=Q9Kbteit8XdL_9Tb8bAtxRxOZQei8Wabs
 kagent/adk/converters/request_converter.py,sha256=iTmTmhlnyRfuYyFi4WmpTSXPz22xjjotbe750j-CvYA,1072
 kagent/adk/models/__init__.py,sha256=mqD0JhS9kT1rMpFNLq5-qnjstpp6lzT9xADaOfjrUKY,78
 kagent/adk/models/_openai.py,sha256=pOHgwEJeBLyQsnb0XgV82A7fEtf68Rv0jXWFNRaZypI,16783
-kagent_adk-0.6.15.dist-info/METADATA,sha256=WtMBIRzo5Cu7z-sYRt6Iv8k6SUju3pZ9ObT0DVR-OLg,944
-kagent_adk-0.6.15.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kagent_adk-0.6.15.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
-kagent_adk-0.6.15.dist-info/RECORD,,
+kagent_adk-0.6.16.dist-info/METADATA,sha256=7oRFScZfYLftiz6Np6KqWcGefo6C0qhaqy-jYgymoRM,944
+kagent_adk-0.6.16.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kagent_adk-0.6.16.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
+kagent_adk-0.6.16.dist-info/RECORD,,