kagent-adk 0.6.18__py3-none-any.whl → 0.6.20__py3-none-any.whl

kagent/adk/_a2a.py

@@ -8,9 +8,11 @@ import httpx
 from a2a.server.apps import A2AFastAPIApplication
 from a2a.server.request_handlers import DefaultRequestHandler
 from a2a.types import AgentCard
+from agentsts.adk import ADKRunner, ADKSessionService, ADKSTSIntegration, ADKTokenPropagationPlugin
 from fastapi import FastAPI, Request
 from fastapi.responses import PlainTextResponse
 from google.adk.agents import BaseAgent
+from google.adk.apps import App
 from google.adk.runners import Runner
 from google.adk.sessions import InMemorySessionService
 from google.genai import types
@@ -21,7 +23,19 @@ from ._agent_executor import A2aAgentExecutor
 from ._session_service import KAgentSessionService
 from ._token import KAgentTokenService
 
+
 # --- Configure Logging ---
+def configure_logging() -> None:
+    """Configure logging based on LOG_LEVEL environment variable."""
+    log_level = os.getenv("LOG_LEVEL", "INFO").upper()
+    numeric_level = getattr(logging, log_level, logging.INFO)
+    logging.basicConfig(
+        level=numeric_level,
+    )
+    logging.info(f"Logging configured with level: {log_level}")
+
+
+configure_logging()
 logger = logging.getLogger(__name__)
 
 
@@ -39,6 +53,7 @@ def thread_dump(request: Request) -> PlainTextResponse:
 
 
 kagent_url_override = os.getenv("KAGENT_URL")
+sts_well_known_uri = os.getenv("STS_WELL_KNOWN_URI")
 
 
 class KAgentApp:
@@ -61,10 +76,17 @@ class KAgentApp:
         )
         session_service = KAgentSessionService(http_client)
 
+        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 Runner(
-                agent=self.root_agent,
-                app_name=self.app_name,
+            return ADKRunner(
+                app=adk_app,
                 session_service=session_service,
             )
 

kagent/adk/_agent_executor.py

@@ -156,6 +156,10 @@ class A2aAgentExecutor(AgentExecutor):
         # Convert the a2a request to ADK run args
         run_args = convert_a2a_request_to_adk_run_args(context)
 
+        # set request headers to session state
+        headers = context.call_context.state.get("headers", {})
+        run_args["headers"] = headers
+
         # ensure the session exists
         session = await self._prepare_session(context, run_args, runner)
 
@@ -256,12 +260,35 @@ class A2aAgentExecutor(AgentExecutor):
             session_id=session_id,
         )
         if session is None:
-            session = await runner.session_service.create_session(
-                app_name=runner.app_name,
-                user_id=user_id,
-                state={},
-                session_id=session_id,
-            )
+            # Extract session name from the first TextPart (like the UI does)
+            session_name = None
+            if context.message and context.message.parts:
+                for part in context.message.parts:
+                    # A2A parts have a .root property that contains the actual part (TextPart, FilePart, etc.)
+                    if isinstance(part, Part):
+                        root_part = part.root
+                        if isinstance(root_part, TextPart) and root_part.text:
+                            # Take first 20 chars + "..." if longer (matching UI behavior)
+                            text = root_part.text.strip()
+                            session_name = text[:20] + ("..." if len(text) > 20 else "")
+                            break
+
+            # Prepare kwargs for create_session
+            create_session_kwargs = {
+                "app_name": runner.app_name,
+                "user_id": user_id,
+                "state": {},
+                "session_id": session_id,
+            }
+
+            if session_name is not None:
+                create_session_kwargs["session_name"] = session_name
+
+            # We pass the kwargs using ** to avoid issues with optional
+            # parameters like `session_name` that exist within
+            # `KAgentSessionService` but not in ADK `BaseSessionService`
+            session = await runner.session_service.create_session(**create_session_kwargs)
+
             # Update run_args with the new session_id
             run_args["session_id"] = session.id
 

kagent/adk/_session_service.py

@@ -32,6 +32,7 @@ class KAgentSessionService(BaseSessionService):
         user_id: str,
         state: Optional[dict[str, Any]] = None,
         session_id: Optional[str] = None,
+        session_name: Optional[str] = None,
     ) -> Session:
         # Prepare request data
         request_data = {
@@ -40,6 +41,8 @@ class KAgentSessionService(BaseSessionService):
         }
         if session_id:
             request_data["id"] = session_id
+        if session_name:
+            request_data["name"] = session_name
 
         # Make API call to create session
         response = await self.client.post(

kagent/adk/types.py

@@ -101,7 +101,7 @@ class AgentConfig(BaseModel):
             for http_tool in self.http_tools:  # add http tools
                 tools.append(MCPToolset(connection_params=http_tool.params, tool_filter=http_tool.tools))
         if self.sse_tools:
-            for sse_tool in self.sse_tools:  # add stdio tools
+            for sse_tool in self.sse_tools:  # add sse tools
                 tools.append(MCPToolset(connection_params=sse_tool.params, tool_filter=sse_tool.tools))
         if self.remote_agents:
             for remote_agent in self.remote_agents:  # Add remote agents as tools
@@ -119,7 +119,7 @@ class AgentConfig(BaseModel):
                     httpx_client=client,
                 )
 
-                tools.append(AgentTool(agent=remote_a2a_agent, skip_summarization=True))
+                tools.append(AgentTool(agent=remote_a2a_agent))
 
         extra_headers = self.model.headers or {}
 

{kagent_adk-0.6.18.dist-info → kagent_adk-0.6.20.dist-info}/METADATA

@@ -1,9 +1,11 @@
 Metadata-Version: 2.4
 Name: kagent-adk
-Version: 0.6.18
+Version: 0.6.20
 Summary: kagent-adk is an sdk for integrating adk agents with kagent
-Requires-Python: >=3.12.11
+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: aiofiles>=24.1.0
 Requires-Dist: anthropic[vertex]>=0.49.0
 Requires-Dist: anyio>=4.9.0

{kagent_adk-0.6.18.dist-info → kagent_adk-0.6.20.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=f6jYkEvp-JEIW34T09Md4B31Di-GblnisqPs7pGFXDY,10136
-kagent/adk/_session_service.py,sha256=Hm2P0NgH8LnfVHmiT-z2AHPHAxPrT9W_6vNkFfIzRK8,5644
+kagent/adk/_a2a.py,sha256=4bb8imLvJR1j8iyocjUlIc3o9ww58XNlWx_oNCb8DJY,5272
+kagent/adk/_agent_executor.py,sha256=XU1ps53I6E3toI6cGM-K4cxQOqaz4cELh-biFlZM9lE,11504
+kagent/adk/_session_service.py,sha256=M8RS0jk2YiPx-nDFSeRka7a25CFEe-Pzb0b18SrojXE,5761
 kagent/adk/_token.py,sha256=OL46m7U5vUTby1WWjVB7Jqzig4TWddzoAmLVLlfSdAg,2515
 kagent/adk/cli.py,sha256=Sdw9FXnUmeHfgJhdRDq1zpoNMf6GM7x35ZDz_OkQgJg,2963
-kagent/adk/types.py,sha256=z7V85oTsizUn5qV5_Ke1vlIuc1ZskzBhMCLFMj3rmVw,5712
+kagent/adk/types.py,sha256=kcNpRwo6XcfMtgsOchgDuslYjMY-7pSlVnglihoUKss,5685
 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=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.18.dist-info/METADATA,sha256=EgJ1JbALuhUEojHlPvV_e-f-5kv2D0lxLIcbZYCnVl8,944
-kagent_adk-0.6.18.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kagent_adk-0.6.18.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
-kagent_adk-0.6.18.dist-info/RECORD,,
+kagent_adk-0.6.20.dist-info/METADATA,sha256=B0AtUJ_qD2Pr6NT_vuJDBpJChk7skUbjIq7TAZ-e0Ng,1014
+kagent_adk-0.6.20.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kagent_adk-0.6.20.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
+kagent_adk-0.6.20.dist-info/RECORD,,