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

x

Browse Source
This commit is contained in:
Eugene Yurtsev 2025-05-30 16:13:34 -04:00
parent 3e7f7b63e3
commit 808f56b841
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
libs/partners/openai/tests/integration_tests/chat_models/test_responses_api.py
View File

@ -12,6 +12,7 @@ from langchain_core.messages import (
BaseMessage, BaseMessage,
BaseMessageChunk, BaseMessageChunk,
HumanMessage, HumanMessage,
MessageLikeRepresentation,
) )
from pydantic import BaseModel from pydantic import BaseModel
from typing_extensions import TypedDict from typing_extensions import TypedDict
@ -525,7 +526,9 @@ def test_image_generation_multi_turn() -> None:
} }
llm_with_tools = llm.bind_tools([tool]) llm_with_tools = llm.bind_tools([tool])
chat_history = [{"role": "user", "content": "Make a picture of a fuzzy cat"}] chat_history: list[MessageLikeRepresentation] = [
{"role": "user", "content": "Make a picture of a fuzzy cat"}
]
ai_message = llm_with_tools.invoke(chat_history) ai_message = llm_with_tools.invoke(chat_history)
_check_response(ai_message) _check_response(ai_message)
tool_output = ai_message.additional_kwargs["tool_outputs"][0] tool_output = ai_message.additional_kwargs["tool_outputs"][0]