kagent-adk 0.6.11__py3-none-any.whl → 0.6.14__py3-none-any.whl

kagent/adk/_session_service.py

@@ -24,10 +24,6 @@ class KAgentSessionService(BaseSessionService):
         super().__init__()
         self.client = client
 
-    async def _get_user_id(self) -> str:
-        """Get the default user ID. Override this method to implement custom user ID logic."""
-        return "admin@kagent.dev"
-
     @override
     async def create_session(
         self,

kagent/adk/models/_openai.py

@@ -9,6 +9,7 @@ from typing import TYPE_CHECKING, Any, AsyncGenerator, Iterable, Literal, Option
 from google.adk.models import BaseLlm
 from google.adk.models.llm_response import LlmResponse
 from google.genai import types
+from google.genai.types import FunctionCall, FunctionResponse
 from openai import AsyncAzureOpenAI, AsyncOpenAI
 from openai.types.chat import (
     ChatCompletion,
@@ -27,7 +28,7 @@ from openai.types.chat.chat_completion_message_tool_call_param import (
 from openai.types.chat.chat_completion_message_tool_call_param import (
     Function as ToolCallFunction,
 )
-from openai.types.shared_params import FunctionDefinition
+from openai.types.shared_params import FunctionDefinition, FunctionParameters
 from pydantic import Field
 
 if TYPE_CHECKING:
@@ -56,7 +57,7 @@ def _convert_content_to_openai_messages(
         messages.append(system_message)
 
     # First pass: collect all function responses to match with tool calls
-    all_function_responses = {}
+    all_function_responses: dict[str, FunctionResponse] = {}
     for content in contents:
         for part in content.parts or []:
             if part.function_response:
@@ -67,18 +68,18 @@ def _convert_content_to_openai_messages(
         role = _convert_role_to_openai(content.role)
 
         # Separate different types of parts
-        text_parts = []
-        function_calls = []
-        function_responses = []
+        text_parts: list[str] = []
+        function_calls: list[FunctionCall] = []
+        function_responses: list[FunctionResponse] = []
         image_parts = []
 
         for part in content.parts or []:
             if part.text:
                 text_parts.append(part.text)
             elif part.function_call:
-                function_calls.append(part)
+                function_calls.append(part.function_call)
             elif part.function_response:
-                function_responses.append(part)
+                function_responses.append(part.function_response)
             elif part.inline_data and part.inline_data.mime_type and part.inline_data.mime_type.startswith("image"):
                 if part.inline_data.data:
                     image_data = base64.b64encode(part.inline_data.data).decode()
@@ -98,43 +99,43 @@ def _convert_content_to_openai_messages(
 
             for func_call in function_calls:
                 tool_call_function: ToolCallFunction = {
-                    "name": func_call.function_call.name or "",
-                    "arguments": str(func_call.function_call.args) if func_call.function_call.args else "{}",
-                }
-                tool_call_id = func_call.function_call.id or "call_1"
-                tool_call: ChatCompletionMessageToolCallParam = {
-                    "id": tool_call_id,
-                    "type": "function",
-                    "function": tool_call_function,
+                    "name": func_call.name or "",
+                    "arguments": json.dumps(func_call.args) if func_call.args else "{}",
                 }
+                tool_call_id = func_call.id or "call_1"
+                tool_call = ChatCompletionMessageToolCallParam(
+                    id=tool_call_id,
+                    type="function",
+                    function=tool_call_function,
+                )
                 tool_calls.append(tool_call)
 
                 # Check if we have a response for this tool call
                 if tool_call_id in all_function_responses:
                     func_response = all_function_responses[tool_call_id]
-                    tool_message: ChatCompletionToolMessageParam = {
-                        "role": "tool",
-                        "tool_call_id": tool_call_id,
-                        "content": str(func_response.response.get("result", "")) if func_response.response else "",
-                    }
+                    tool_message = ChatCompletionToolMessageParam(
+                        role="tool",
+                        tool_call_id=tool_call_id,
+                        content=str(func_response.response.get("result", "")) if func_response.response else "",
+                    )
                     tool_response_messages.append(tool_message)
                 else:
                     # If no response is available, create a placeholder response
                     # This prevents the OpenAI API error
-                    tool_message: ChatCompletionToolMessageParam = {
-                        "role": "tool",
-                        "tool_call_id": tool_call_id,
-                        "content": "No response available for this function call.",
-                    }
+                    tool_message = ChatCompletionToolMessageParam(
+                        role="tool",
+                        tool_call_id=tool_call_id,
+                        content="No response available for this function call.",
+                    )
                     tool_response_messages.append(tool_message)
 
             # Create assistant message with tool calls
             text_content = "\n".join(text_parts) if text_parts else None
-            assistant_message: ChatCompletionAssistantMessageParam = {
-                "role": "assistant",
-                "content": text_content,
-                "tool_calls": tool_calls,
-            }
+            assistant_message = ChatCompletionAssistantMessageParam(
+                role="assistant",
+                content=text_content,
+                tool_calls=tool_calls,
+            )
             messages.append(assistant_message)
 
             # Add all tool response messages immediately after the assistant message
@@ -145,22 +146,22 @@ def _convert_content_to_openai_messages(
             if role == "user":
                 if image_parts and text_parts:
                     # Multi-modal content
-                    text_part: ChatCompletionContentPartTextParam = {"type": "text", "text": "\n".join(text_parts)}
+                    text_part = ChatCompletionContentPartTextParam(type="text", text="\n".join(text_parts))
                     content_parts = [text_part] + image_parts
-                    user_message: ChatCompletionUserMessageParam = {"role": "user", "content": content_parts}
+                    user_message = ChatCompletionUserMessageParam(role="user", content=content_parts)
                 elif image_parts:
                     # Image only
-                    user_message: ChatCompletionUserMessageParam = {"role": "user", "content": image_parts}
+                    user_message = ChatCompletionUserMessageParam(role="user", content=image_parts)
                 else:
                     # Text only
-                    user_message: ChatCompletionUserMessageParam = {"role": "user", "content": "\n".join(text_parts)}
+                    user_message = ChatCompletionUserMessageParam(role="user", content="\n".join(text_parts))
                 messages.append(user_message)
             elif role == "assistant":
                 # Assistant messages with text (no tool calls)
-                assistant_message: ChatCompletionAssistantMessageParam = {
-                    "role": "assistant",
-                    "content": "\n".join(text_parts),
-                }
+                assistant_message = ChatCompletionAssistantMessageParam(
+                    role="assistant",
+                    content="\n".join(text_parts),
+                )
                 messages.append(assistant_message)
 
     return messages
@@ -197,10 +198,10 @@ def _convert_tools_to_openai(tools: list[types.Tool]) -> list[ChatCompletionTool
         if tool.function_declarations:
             for func_decl in tool.function_declarations:
                 # Build function definition
-                function_def: FunctionDefinition = {
-                    "name": func_decl.name or "",
-                    "description": func_decl.description or "",
-                }
+                function_def = FunctionDefinition(
+                    name=func_decl.name or "",
+                    description=func_decl.description or "",
+                )
 
                 # Always include parameters field, even if empty
                 properties = {}
@@ -219,7 +220,7 @@ def _convert_tools_to_openai(tools: list[types.Tool]) -> list[ChatCompletionTool
                 function_def["parameters"] = {"type": "object", "properties": properties, "required": required}
 
                 # Create the tool param
-                openai_tool: ChatCompletionToolParam = {"type": "function", "function": function_def}
+                openai_tool = ChatCompletionToolParam(type="function", function=function_def)
                 openai_tools.append(openai_tool)
 
     return openai_tools
@@ -389,6 +390,7 @@ class AzureOpenAI(BaseOpenAI):
     api_version: Optional[str] = None
     azure_endpoint: Optional[str] = None
     azure_deployment: Optional[str] = None
+    headers: Optional[dict[str, str]] = None
 
     @cached_property
     def _client(self) -> AsyncAzureOpenAI:
@@ -407,4 +409,8 @@ class AzureOpenAI(BaseOpenAI):
                 "API key must be provided either via api_key parameter or AZURE_OPENAI_API_KEY environment variable"
             )
 
-        return AsyncAzureOpenAI(api_version=api_version, azure_endpoint=azure_endpoint, api_key=api_key)
+        default_headers = self.headers or {}
+
+        return AsyncAzureOpenAI(
+            api_version=api_version, azure_endpoint=azure_endpoint, api_key=api_key, default_headers=default_headers
+        )

kagent/adk/types.py

@@ -1,10 +1,12 @@
+import httpx
 import logging
-from typing import Literal, Self, Union
+
+from typing import Literal, Any, Union
 
 from google.adk.agents import Agent
 from google.adk.agents.base_agent import BaseAgent
 from google.adk.agents.llm_agent import ToolUnion
-from google.adk.agents.remote_a2a_agent import AGENT_CARD_WELL_KNOWN_PATH, RemoteA2aAgent
+from google.adk.agents.remote_a2a_agent import AGENT_CARD_WELL_KNOWN_PATH, DEFAULT_TIMEOUT, RemoteA2aAgent
 from google.adk.models.anthropic_llm import Claude as ClaudeLLM
 from google.adk.models.google_llm import Gemini as GeminiLLM
 from google.adk.models.lite_llm import LiteLlm
@@ -31,11 +33,14 @@ class SseMcpServerConfig(BaseModel):
 class RemoteAgentConfig(BaseModel):
     name: str
     url: str
+    headers: dict[str, Any] | None = None
+    timeout: float = DEFAULT_TIMEOUT
     description: str = ""
 
 
 class BaseLLM(BaseModel):
     model: str
+    headers: dict[str, str] | None = None
 
 
 class OpenAI(BaseLLM):
@@ -76,41 +81,56 @@ class AgentConfig(BaseModel):
     )
     description: str
     instruction: str
-    http_tools: list[HttpMcpServerConfig] | None = None  # tools, always MCP
-    sse_tools: list[SseMcpServerConfig] | None = None  # tools, always MCP
+    http_tools: list[HttpMcpServerConfig] | None = None  # Streamable HTTP MCP tools
+    sse_tools: list[SseMcpServerConfig] | None = None  # SSE MCP tools
     remote_agents: list[RemoteAgentConfig] | None = None  # remote agents
 
     def to_agent(self, name: str) -> Agent:
         if name is None or not str(name).strip():
             raise ValueError("Agent name must be a non-empty string.")
-        mcp_toolsets: list[ToolUnion] = []
+        tools: list[ToolUnion] = []
         if self.http_tools:
             for http_tool in self.http_tools:  # add http tools
-                mcp_toolsets.append(MCPToolset(connection_params=http_tool.params, tool_filter=http_tool.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
-                mcp_toolsets.append(MCPToolset(connection_params=sse_tool.params, tool_filter=sse_tool.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
-                remote_agent = RemoteA2aAgent(
+                client = None
+
+                if remote_agent.headers:
+                    client = httpx.AsyncClient(
+                        headers=remote_agent.headers, timeout=httpx.Timeout(timeout=remote_agent.timeout)
+                    )
+
+                remote_a2a_agent = RemoteA2aAgent(
                     name=remote_agent.name,
                     agent_card=f"{remote_agent.url}/{AGENT_CARD_WELL_KNOWN_PATH}",
                     description=remote_agent.description,
+                    httpx_client=client,
                 )
-                mcp_toolsets.append(AgentTool(agent=remote_agent, skip_summarization=True))
+
+                tools.append(
+                    AgentTool(agent=remote_a2a_agent, skip_summarization=True)
+                )  # Get headers from model config
+
+        extra_headers = self.model.headers or {}
 
         if self.model.type == "openai":
             model = OpenAINative(model=self.model.model, base_url=self.model.base_url, type="openai")
         elif self.model.type == "anthropic":
-            model = LiteLlm(model=f"anthropic/{self.model.model}", base_url=self.model.base_url)
+            model = LiteLlm(
+                model=f"anthropic/{self.model.model}", base_url=self.model.base_url, extra_headers=extra_headers
+            )
         elif self.model.type == "gemini_vertex_ai":
             model = GeminiLLM(model=self.model.model)
         elif self.model.type == "gemini_anthropic":
             model = ClaudeLLM(model=self.model.model)
         elif self.model.type == "ollama":
-            model = LiteLlm(model=f"ollama_chat/{self.model.model}")
+            model = LiteLlm(model=f"ollama_chat/{self.model.model}", extra_headers=extra_headers)
         elif self.model.type == "azure_openai":
-            model = OpenAIAzure(model=self.model.model, type="azure_openai")
+            model = OpenAIAzure(model=self.model.model, type="azure_openai", headers=extra_headers)
         elif self.model.type == "gemini":
             model = self.model.model
         else:
@@ -120,5 +140,5 @@ class AgentConfig(BaseModel):
             model=model,
             description=self.description,
             instruction=self.instruction,
-            tools=mcp_toolsets,
+            tools=tools,
         )

{kagent_adk-0.6.11.dist-info → kagent_adk-0.6.14.dist-info}/METADATA

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

@@ -1,18 +1,18 @@
 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/_session_service.py,sha256=A47gsfDVp8jITzeW987AHTJLEhcU_mU3ik_SFptFGIc,5815
+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=v7VbO0tAPvpjdEgWJroEq64jSCz6D3Q0yPdRM0a7MjM,4260
+kagent/adk/types.py,sha256=E0HxxrSrpuZVWE7F9lbNY4N2276QK3qCwhuBdDbhMTY,4911
 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
 kagent/adk/converters/part_converter.py,sha256=Q9Kbteit8XdL_9Tb8bAtxRxOZQei8Wabszwsl4YMe2c,7507
 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=0hQklnXacTHsb5h1gkPqmuoEyJl_ibh5R-FBY-wQ7Ts,16594
-kagent_adk-0.6.11.dist-info/METADATA,sha256=tgksopEr1rQBB4B6nGmtBeQeNTDG7dwxLpWXEitjTs8,944
-kagent_adk-0.6.11.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-kagent_adk-0.6.11.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
-kagent_adk-0.6.11.dist-info/RECORD,,
+kagent/adk/models/_openai.py,sha256=ze8JgMOqZFJidf4XEtCL2o6xV6aLY1ZR1eilR7tm5KI,16789
+kagent_adk-0.6.14.dist-info/METADATA,sha256=NLkf04Kk0C8NJqpS1F56VKHCe5mGKCM-mB5MOzMOu8s,944
+kagent_adk-0.6.14.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kagent_adk-0.6.14.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
+kagent_adk-0.6.14.dist-info/RECORD,,