agent/agent-framework
1
0
Fork 0
mirror of https://github.com/microsoft/agent-framework.git synced 2026-06-16 21:04:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: capture streaming response format after prepare

Browse Source 
Agent-Logs-Url: https://github.com/microsoft/agent-framework/sessions/f62076ef-558d-49e8-8fe2-f38d527c9639

Co-authored-by: eavanvalkenburg <13749212+eavanvalkenburg@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-05-19 05:52:04 +00:00
committed by GitHub
Unverified
parent e1d4f1e11e
commit baeb73a58c
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/packages/openai/agent_framework_openai/_chat_client.py
+5 -5
View File
@@ -614,19 +614,18 @@ class RawOpenAIChatClient( # type: ignore[misc]
function_call_ids: dict[int, tuple[str, str]] = {}
seen_reasoning_delta_item_ids: set[str] = set()
validated_options: dict[str, Any] | None = None
response_format: Any | None = None
def _finalize_streamed_response(updates: Sequence[ChatResponseUpdate]) -> ChatResponse[Any]:
return self._finalize_response_updates(
updates,
response_format=validated_options.get("response_format") if validated_options else None,
)
return self._finalize_response_updates(updates, response_format=response_format)
async def _stream() -> AsyncIterable[ChatResponseUpdate]:
nonlocal validated_options
nonlocal response_format, validated_options
if continuation_token is not None:
# Resume a background streaming response by retrieving with stream=True
client = self.client
validated_options = await self._validate_options(options)
response_format = validated_options.get("response_format")
try:
raw_stream_response = await client.responses.with_raw_response.retrieve(
continuation_token["response_id"],
@@ -652,6 +651,7 @@ class RawOpenAIChatClient( # type: ignore[misc]
run_options,
validated_options,
) = await self._prepare_request(messages, options)
response_format = validated_options.get("response_format")
try:
if "text_format" in run_options:
# The SDK's ``responses.stream(text_format=...)`` helper preserves