From df415417a19db06a3f7743fea0de8deda3e58fa8 Mon Sep 17 00:00:00 2001 From: isaac hershenson Date: Tue, 5 Nov 2024 07:36:36 -0800 Subject: [PATCH] fmt --- libs/core/langchain_core/messages/utils.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/libs/core/langchain_core/messages/utils.py b/libs/core/langchain_core/messages/utils.py index 35e14815bbd..575d9519536 100644 --- a/libs/core/langchain_core/messages/utils.py +++ b/libs/core/langchain_core/messages/utils.py @@ -36,7 +36,12 @@ from langchain_core.messages.function import FunctionMessage, FunctionMessageChu from langchain_core.messages.human import HumanMessage, HumanMessageChunk from langchain_core.messages.modifier import RemoveMessage from langchain_core.messages.system import SystemMessage, SystemMessageChunk -from langchain_core.messages.tool import ToolCall, ToolMessage, ToolMessageChunk +from langchain_core.messages.tool import ( + InvalidToolCall, + ToolCall, + ToolMessage, + ToolMessageChunk, +) if TYPE_CHECKING: from langchain_text_splitters import TextSplitter @@ -1404,7 +1409,7 @@ def _get_message_openai_role(message: BaseMessage) -> str: def _convert_to_openai_tool_calls( - tool_calls: list[ToolCall], invalid=False + tool_calls: list[Union[ToolCall, InvalidToolCall]], invalid: bool = False ) -> list[dict]: return [ {