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

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

@@ -36,6 +36,7 @@ class RemoteAgentConfig(BaseModel):
 
 class BaseLLM(BaseModel):
     model: str
+    headers: dict[str, str] | None = None
 
 
 class OpenAI(BaseLLM):
@@ -97,20 +98,26 @@ class AgentConfig(BaseModel):
                     agent_card=f"{remote_agent.url}/{AGENT_CARD_WELL_KNOWN_PATH}",
                     description=remote_agent.description,
                 )
-                mcp_toolsets.append(AgentTool(agent=remote_agent, skip_summarization=True))
+                mcp_toolsets.append(
+                    AgentTool(agent=remote_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:

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

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

@@ -4,15 +4,15 @@ kagent/adk/_agent_executor.py,sha256=rxNVdrcS4L5Kwe8oF7mLUnD1KOYhe6vXK-si0O-KQ9o
 kagent/adk/_session_service.py,sha256=A47gsfDVp8jITzeW987AHTJLEhcU_mU3ik_SFptFGIc,5815
 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=7xW2k_16rcbLwTYchAnXZqTDbEyZIX8X3Zv3018dq7A,4534
 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.12.dist-info/METADATA,sha256=xcIldMuG_foeMSHo775znd6eRXn1bD3Uc-ynQPC3q_s,944
+kagent_adk-0.6.12.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+kagent_adk-0.6.12.dist-info/entry_points.txt,sha256=a1Q2Inc9L0dvXWEkwnCdf9cfXdpX5Dl2Q6DhNWNjhxw,50
+kagent_adk-0.6.12.dist-info/RECORD,,