kagent-adk 0.6.21__py3-none-any.whl → 0.7.0__py3-none-any.whl

kagent/adk/_a2a.py

@@ -9,7 +9,7 @@ from a2a.server.apps import A2AFastAPIApplication
 from a2a.server.request_handlers import DefaultRequestHandler
 from a2a.server.tasks import InMemoryTaskStore
 from a2a.types import AgentCard
-from agentsts.adk import ADKRunner, ADKSessionService, ADKSTSIntegration, ADKTokenPropagationPlugin
+from agentsts.adk import ADKSTSIntegration, ADKTokenPropagationPlugin
 from fastapi import FastAPI, Request
 from fastapi.responses import PlainTextResponse
 from google.adk.agents import BaseAgent
@@ -80,13 +80,12 @@ class KAgentApp:
         plugins = []
         if sts_well_known_uri:
             sts_integration = ADKSTSIntegration(sts_well_known_uri)
-            session_service = ADKSessionService(sts_integration, session_service)
             plugins.append(ADKTokenPropagationPlugin(sts_integration))
 
         adk_app = App(name=self.app_name, root_agent=self.root_agent, plugins=plugins)
 
         def create_runner() -> Runner:
-            return ADKRunner(
+            return Runner(
                 app=adk_app,
                 session_service=session_service,
             )

kagent/adk/_agent_executor.py

@@ -21,6 +21,7 @@ from a2a.types import (
     TaskStatusUpdateEvent,
     TextPart,
 )
+from google.adk.events import Event, EventActions
 from google.adk.runners import Runner
 from google.adk.utils.context_utils import Aclosing
 from opentelemetry import trace
@@ -156,12 +157,23 @@ class A2aAgentExecutor(AgentExecutor):
         # Convert the a2a request to ADK run args
         run_args = convert_a2a_request_to_adk_run_args(context)
 
+        # ensure the session exists
+        session = await self._prepare_session(context, run_args, runner)
+
         # set request headers to session state
         headers = context.call_context.state.get("headers", {})
-        run_args["headers"] = headers
+        state_changes = {
+            "headers": headers,
+        }
+
+        actions_with_update = EventActions(state_delta=state_changes)
+        system_event = Event(
+            invocation_id="header_update",
+            author="system",
+            actions=actions_with_update,
+        )
 
-        # ensure the session exists
-        session = await self._prepare_session(context, run_args, runner)
+        await runner.session_service.append_event(session, system_event)
 
         current_span = trace.get_current_span()
         if run_args["user_id"]:
@@ -259,6 +271,7 @@ class A2aAgentExecutor(AgentExecutor):
             user_id=user_id,
             session_id=session_id,
         )
+
         if session is None:
             # Extract session name from the first TextPart (like the UI does)
             session_name = None

kagent/adk/_session_service.py

@@ -108,13 +108,18 @@ class KAgentSessionService(BaseSessionService):
                 events.append(Event.model_validate_json(event_data["data"]))
 
             # Convert to ADK Session format
-            return Session(
+            session = Session(
                 id=session_data["id"],
                 user_id=session_data["user_id"],
                 events=events,
                 app_name=app_name,
-                state={},  # TODO: restore State
+                state={},
             )
+
+            for event in events:
+                await super().append_event(session, event)
+
+            return session
         except httpx.HTTPStatusError as e:
             if e.response.status_code == 404:
                 return None

{kagent_adk-0.6.21.dist-info → kagent_adk-0.7.0.dist-info}/METADATA

@@ -1,16 +1,16 @@
 Metadata-Version: 2.4
 Name: kagent-adk
-Version: 0.6.21
+Version: 0.7.0
 Summary: kagent-adk is an sdk for integrating adk agents with kagent
 Requires-Python: >=3.11.0
 Requires-Dist: a2a-sdk>=0.3.1
-Requires-Dist: agentsts-adk>=0.0.5
-Requires-Dist: agentsts-core>=0.0.5
+Requires-Dist: agentsts-adk>=0.0.6
+Requires-Dist: agentsts-core>=0.0.6
 Requires-Dist: aiofiles>=24.1.0
 Requires-Dist: anthropic[vertex]>=0.49.0
 Requires-Dist: anyio>=4.9.0
 Requires-Dist: fastapi>=0.115.1
-Requires-Dist: google-adk>=1.11.0
+Requires-Dist: google-adk>=1.16.0
 Requires-Dist: google-auth>=2.40.2
 Requires-Dist: google-genai>=1.21.1
 Requires-Dist: httpx>=0.25.0

{kagent_adk-0.6.21.dist-info → kagent_adk-0.7.0.dist-info}/RECORD

@@ -1,7 +1,7 @@
 kagent/adk/__init__.py,sha256=Cam9hwhl6Z1tZ3WJIB1KATX24MwRMpiyBLYF64DTqQI,182
-kagent/adk/_a2a.py,sha256=B1ZiVFJHNJyJ_6HsCbAG5w5-wF9xrNKhr3-ok9CCick,6424
-kagent/adk/_agent_executor.py,sha256=3aFMR41FwluDZZJGkpsL4yhz7JgqTTSbKtLeENrr0TE,11090
-kagent/adk/_session_service.py,sha256=7o2T4-572hAvpD3KZQn1ExPvI67aEuLcyHqQXB3b2JI,5757
+kagent/adk/_a2a.py,sha256=oSpj5fAzPPA_KP5fTNLUpFSQ2J28tLXrTCBpQSNCjq0,6309
+kagent/adk/_agent_executor.py,sha256=U_ZxJhaGyn-FPxvoNn7L5ZQComZ-vfFG9TdCWRLoTm8,11469
+kagent/adk/_session_service.py,sha256=6kwFS4FF10HlUyyCV2SgwxqKIJ47s0ZZ8KW-iIRDyec,5858
 kagent/adk/_token.py,sha256=OL46m7U5vUTby1WWjVB7Jqzig4TWddzoAmLVLlfSdAg,2515
 kagent/adk/cli.py,sha256=_yluK6p7sbUld5juWCyIu3E8U1_Zys7eWxBabcHXXIo,3251
 kagent/adk/types.py,sha256=kcNpRwo6XcfMtgsOchgDuslYjMY-7pSlVnglihoUKss,5685
@@ -12,7 +12,7 @@ kagent/adk/converters/part_converter.py,sha256=8Ej9xGRYW8YoPnExGDnEUw1beurCfkNhA
 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=EpZTqxAEaKhgi-98sqyQhArckWwoGh-C34fC8MyClHk,17187
-kagent_adk-0.6.21.dist-info/METADATA,sha256=oPN2egxeoRZpz_hOoBSH1f8GgDSjRbe6pczg3UhfiSc,1014
-kagent_adk-0.6.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kagent_adk-0.6.21.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
-kagent_adk-0.6.21.dist-info/RECORD,,
+kagent_adk-0.7.0.dist-info/METADATA,sha256=hnbBr2v7iGKlmQq2KRpZcGt1ZWNr0yrMxighe_3a4gk,1013
+kagent_adk-0.7.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kagent_adk-0.7.0.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
+kagent_adk-0.7.0.dist-info/RECORD,,