From 808f56b841d248730a04aba344e38181d73fcd01 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Fri, 30 May 2025 16:13:34 -0400 Subject: [PATCH] x --- .../integration_tests/chat_models/test_responses_api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/libs/partners/openai/tests/integration_tests/chat_models/test_responses_api.py b/libs/partners/openai/tests/integration_tests/chat_models/test_responses_api.py index 20b74c6c9fe..bf2a80c961f 100644 --- a/libs/partners/openai/tests/integration_tests/chat_models/test_responses_api.py +++ b/libs/partners/openai/tests/integration_tests/chat_models/test_responses_api.py @@ -12,6 +12,7 @@ from langchain_core.messages import ( BaseMessage, BaseMessageChunk, HumanMessage, + MessageLikeRepresentation, ) from pydantic import BaseModel from typing_extensions import TypedDict @@ -525,7 +526,9 @@ def test_image_generation_multi_turn() -> None: } 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) _check_response(ai_message) tool_output = ai_message.additional_kwargs["tool_outputs"][0]