This commit is contained in:
Bagatur 2025-01-06 16:18:31 -05:00
parent 22863b8ac3
commit 47b386d28f

View File

@ -777,6 +777,8 @@ class BaseChatOpenAI(BaseChatModel):
):
message = response.choices[0].message # type: ignore[attr-defined]
if hasattr(message, "parsed"):
generations[0].message.parsed = message.parsed
# For backwards compatibility.
generations[0].message.additional_kwargs["parsed"] = message.parsed
if hasattr(message, "refusal"):
generations[0].message.additional_kwargs["refusal"] = message.refusal