github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-06-09 10:17:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(openai): accommodate dict response items in streaming (#36899)

Browse Source
This commit is contained in:
ccurme
2026-04-20 15:44:01 -04:00
committed by GitHub
parent 8fec4e7cee
commit 19b0805bc1
3 changed files with 48 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
libs/partners/openai/langchain_openai/chat_models/base.py
View File

@@ -4589,6 +4589,15 @@ def _construct_lc_result_from_responses_api(
return ChatResult(generations=[ChatGeneration(message=message)])
def _coerce_chunk_response(resp: Any) -> Any:
# dict `response` items on stream events have been observed in the wild
if isinstance(resp, dict):
from openai.types.responses import Response
return Response.model_validate(resp)
return resp
def _convert_responses_chunk_to_generation_chunk(
chunk: Any,
current_index: int, # index in content
@@ -4686,14 +4695,16 @@ def _convert_responses_chunk_to_generation_chunk(
}
)
elif chunk.type == "response.created":
id = chunk.response.id
response_metadata["id"] = chunk.response.id # Backwards compatibility
response = _coerce_chunk_response(chunk.response)
id = response.id
response_metadata["id"] = response.id # Backwards compatibility
elif chunk.type in ("response.completed", "response.incomplete"):
response = _coerce_chunk_response(chunk.response)
msg = cast(
AIMessage,
(
_construct_lc_result_from_responses_api(
chunk.response, schema=schema, output_version=output_version
response, schema=schema, output_version=output_version
)
.generations[0]
.message