github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-22 10:59:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

support annotations in streaming case

Browse Source
This commit is contained in:
Chester Curme 2025-03-11 18:12:26 -04:00
parent f48a3c661a
commit 909387202a
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -364,6 +364,13 @@ def _convert_responses_chunk_to_generation_chunk(
usage_metadata = None usage_metadata = None
if chunk.type == "response.output_text.delta": if chunk.type == "response.output_text.delta":
content += [{"type": "text", "text": chunk.delta, "index": chunk.content_index}] content += [{"type": "text", "text": chunk.delta, "index": chunk.content_index}]
elif chunk.type == "response.output_text.annotation.added":
content += [
{
"annotations": [chunk.annotation.model_dump()],
"index": chunk.content_index,
}
]
elif chunk.type == "response.completed": elif chunk.type == "response.completed":
token_usage = chunk.response.usage.model_dump() if chunk.response.usage else {} token_usage = chunk.response.usage.model_dump() if chunk.response.usage else {}
usage_metadata = _create_usage_metadata_responses(token_usage) usage_metadata = _create_usage_metadata_responses(token_usage)