Skip to content

Python: Fix AzureAIClient tool call bug for AG-UI use #3148

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
moonbox3 merged 2 commits into from
Jan 9, 2026
Merged

Python: Fix AzureAIClient tool call bug for AG-UI use #3148

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
804c817
Fiz AzureAIClient tool call bug
moonbox3 Jan 9, 2026
65a0c4c
Address copilot feedback
moonbox3 Jan 9, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion python/packages/ag-ui/agent_framework_ag_ui/_events.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -160,10 +160,10 @@ def _handle_function_call_content(self, content: FunctionCallContent) -> list[Ba
logger.warning(f"FunctionCallContent missing name and call_id. args_length={args_length}")

tool_call_id = self._coalesce_tool_call_id(content)
# Only emit ToolCallStartEvent once per tool call (when it's a new tool call)
if content.name and tool_call_id != self.current_tool_call_id:
self.streaming_tool_args = ""
self.state_delta_count = 0
if content.name:
self.current_tool_call_id = tool_call_id
self.current_tool_call_name = content.name

Expand Down
80 changes: 66 additions & 14 deletions python/packages/core/agent_framework/openai/_responses_client.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,14 @@
# Copyright (c) Microsoft. All rights reserved.

from collections.abc import AsyncIterable, Awaitable, Callable, Mapping, MutableMapping, MutableSequence, Sequence
from collections.abc import (
AsyncIterable,
Awaitable,
Callable,
Mapping,
MutableMapping,
MutableSequence,
Sequence,
)
from datetime import datetime, timezone
from itertools import chain
from typing import Any, TypeVar, cast
Expand All @@ -12,15 +20,19 @@
ParsedResponse,
)
from openai.types.responses.response import Response as OpenAIResponse
from openai.types.responses.response_stream_event import ResponseStreamEvent as OpenAIResponseStreamEvent
from openai.types.responses.response_stream_event import (
ResponseStreamEvent as OpenAIResponseStreamEvent,
)
from openai.types.responses.response_usage import ResponseUsage
from openai.types.responses.tool_param import (
CodeInterpreter,
CodeInterpreterContainerCodeInterpreterToolAuto,
Mcp,
ToolParam,
)
from openai.types.responses.web_search_tool_param import UserLocation as WebSearchUserLocation
from openai.types.responses.web_search_tool_param import (
UserLocation as WebSearchUserLocation,
)
from openai.types.responses.web_search_tool_param import WebSearchToolParam
from pydantic import BaseModel, ValidationError

Expand Down Expand Up @@ -139,13 +151,17 @@ async def _inner_get_streaming_response(
if "text_format" not in run_options:
async for chunk in await client.responses.create(stream=True, **run_options):
yield self._parse_chunk_from_openai(
chunk, chat_options=chat_options, function_call_ids=function_call_ids
chunk,
chat_options=chat_options,
function_call_ids=function_call_ids,
)
return
async with client.responses.stream(**run_options) as response:
async for chunk in response:
yield self._parse_chunk_from_openai(
chunk, chat_options=chat_options, function_call_ids=function_call_ids
chunk,
chat_options=chat_options,
function_call_ids=function_call_ids,
)
except BadRequestError as ex:
if ex.code == "content_filter":
Expand Down Expand Up @@ -555,7 +571,10 @@ def _prepare_content_for_openai(
if status := props.get("status"):
ret["status"] = status
if reasoning_text := props.get("reasoning_text"):
ret["content"] = {"type": "reasoning_text", "text": reasoning_text}
ret["content"] = {
"type": "reasoning_text",
"text": reasoning_text,
}
if encrypted_content := props.get("encrypted_content"):
ret["encrypted_content"] = encrypted_content
return ret
Expand Down Expand Up @@ -601,9 +620,17 @@ def _prepare_content_for_openai(
return file_obj
return {}
case FunctionCallContent():
if not content.call_id:
logger.warning(f"FunctionCallContent missing call_id for function '{content.name}'")
return {}
# Use fc_id from additional_properties if available, otherwise fallback to call_id
fc_id = call_id_to_id.get(content.call_id, content.call_id)
# OpenAI Responses API requires IDs to start with `fc_`
if not fc_id.startswith("fc_"):
fc_id = f"fc_{fc_id}"
return {
"call_id": content.call_id,
"id": call_id_to_id[content.call_id],
"id": fc_id,
"type": "function_call",
"name": content.name,
"arguments": content.arguments,
Expand Down Expand Up @@ -739,11 +766,17 @@ def _parse_response_from_openai(
)
)
case _:
logger.debug("Unparsed annotation type: %s", annotation.type)
logger.debug(
"Unparsed annotation type: %s",
annotation.type,
)
contents.append(text_content)
case "refusal":
contents.append(
TextContent(text=message_content.refusal, raw_representation=message_content)
TextContent(
text=message_content.refusal,
raw_representation=message_content,
)
)
case "reasoning": # ResponseOutputReasoning
if hasattr(item, "content") and item.content:
Expand All @@ -769,7 +802,12 @@ def _parse_response_from_openai(
if item_outputs := getattr(item, "outputs", None):
for code_output in item_outputs:
if getattr(code_output, "type", None) == "logs":
outputs.append(TextContent(text=code_output.logs, raw_representation=code_output))
outputs.append(
TextContent(
text=code_output.logs,
raw_representation=code_output,
)
)
elif getattr(code_output, "type", None) == "image":
outputs.append(
UriContent(
Expand Down Expand Up @@ -1008,7 +1046,10 @@ def _parse_chunk_from_openai(
# McpApprovalRequest,
# ResponseCustomToolCall,
case "function_call":
function_call_ids[event.output_index] = (event_item.call_id, event_item.name)
function_call_ids[event.output_index] = (
event_item.call_id,
event_item.name,
)
case "mcp_approval_request":
contents.append(
FunctionApprovalRequestContent(
Expand Down Expand Up @@ -1061,7 +1102,10 @@ def _parse_chunk_from_openai(
for code_output in event_item.outputs:
if getattr(code_output, "type", None) == "logs":
outputs.append(
TextContent(text=cast(Any, code_output).logs, raw_representation=code_output)
TextContent(
text=cast(Any, code_output).logs,
raw_representation=code_output,
)
)
elif getattr(code_output, "type", None) == "image":
outputs.append(
Expand All @@ -1075,7 +1119,12 @@ def _parse_chunk_from_openai(
contents.append(
CodeInterpreterToolCallContent(
call_id=call_id,
inputs=[TextContent(text=event_item.code, raw_representation=event_item)],
inputs=[
TextContent(
text=event_item.code,
raw_representation=event_item,
)
],
raw_representation=event_item,
)
)
Expand Down Expand Up @@ -1113,7 +1162,10 @@ def _parse_chunk_from_openai(
call_id=call_id,
name=name,
arguments=event.delta,
additional_properties={"output_index": event.output_index, "fc_id": event.item_id},
additional_properties={
"output_index": event.output_index,
"fc_id": event.item_id,
},
raw_representation=event,
)
)
Expand Down
Loading