github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-19 09:30:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

add tool output in streaming case

Browse Source
This commit is contained in:
Chester Curme 2025-03-11 19:14:14 -04:00
parent 5da74b5053
commit 307dffdc92
2 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -375,6 +375,18 @@ def _convert_responses_chunk_to_generation_chunk(
token_usage = chunk.response.usage.model_dump() if chunk.response.usage else {}
usage_metadata = _create_usage_metadata_responses(token_usage)
generation_info = {"model_name": chunk.response.model}
elif chunk.type in (
"response.web_search_call.completed",
"response.file_search_call.completed",
):
tool_output = chunk.model_dump()
if "item_id" in tool_output:
tool_output["id"] = tool_output.pop("item_id")
tool_output["type"] = (
tool_output["type"].replace("response.", "").replace(".completed", "")
)
tool_output["status"] = "completed"
generation_info = {"tool_outputs": [tool_output]}
else:
return None

1
libs/partners/openai/tests/integration_tests/chat_models/test_base.py
View File

@ -2,7 +2,6 @@
import base64
import json
import os
from pathlib import Path
from textwrap import dedent
from typing import Any, AsyncIterator, List, Literal, Optional, cast