openai[patch]: fix dropping response headers while streaming / Azure (#31580)

joshy-deshaw · web-flow · commit 8a0782c46c9f · 2025-06-23T17:59:58.000-04:00
diff --git a/libs/partners/openai/langchain_openai/chat_models/base.py b/libs/partners/openai/langchain_openai/chat_models/base.py
@@ -793,7 +793,8 @@ def _convert_chunk_to_generation_chunk(
         if len(choices) == 0:
             # logprobs is implicitly None
             generation_chunk = ChatGenerationChunk(
-                message=default_chunk_class(content="", usage_metadata=usage_metadata)
+                message=default_chunk_class(content="", usage_metadata=usage_metadata),
+                generation_info=base_generation_info,
             )
             return generation_chunk
 

Original file line number	Diff line number	Diff line change
`@@ -793,7 +793,8 @@ def _convert_chunk_to_generation_chunk(`
`793`	`793`	`if len(choices) == 0:`
`794`	`794`	`# logprobs is implicitly None`
`795`	`795`	`generation_chunk = ChatGenerationChunk(`
`796`		`- message=default_chunk_class(content="", usage_metadata=usage_metadata)`
	`796`	`+ message=default_chunk_class(content="", usage_metadata=usage_metadata),`
	`797`	`+ generation_info=base_generation_info,`
`797`	`798`	`)`
`798`	`799`	`return generation_chunk`
`799`	`800`