From 150583bd4057e8eb70993fce73fda6ccc5177fb9 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Sun, 21 Jan 2024 09:12:18 -0800 Subject: [PATCH] py3.9 compat --- tests/test_pregel.py | 6 +++--- tests/test_pregel_async.py | 5 ++--- 2 files changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/test_pregel.py b/tests/test_pregel.py index ac198490b..205c34d23 100644 --- a/tests/test_pregel.py +++ b/tests/test_pregel.py @@ -4,7 +4,7 @@ import time import warnings from concurrent.futures import ThreadPoolExecutor from contextlib import contextmanager -from typing import Annotated, Generator, Optional, Self, TypedDict, Union +from typing import Annotated, Generator, Optional, TypedDict, Union import pytest from langchain_core.runnables import RunnablePassthrough @@ -983,7 +983,7 @@ def test_prebuilt_chat() -> None: from langchain_core.messages import AIMessage, FunctionMessage, HumanMessage class FakeFuntionChatModel(FakeMessagesListChatModel): - def bind_functions(self, functions: list) -> Self: + def bind_functions(self, functions: list): return self @tool() @@ -1129,7 +1129,7 @@ def test_message_graph() -> None: from langchain_core.messages import AIMessage, FunctionMessage, HumanMessage class FakeFuntionChatModel(FakeMessagesListChatModel): - def bind_functions(self, functions: list) -> Self: + def bind_functions(self, functions: list): return self @tool() diff --git a/tests/test_pregel_async.py b/tests/test_pregel_async.py index ce50a2271..286db9a3e 100644 --- a/tests/test_pregel_async.py +++ b/tests/test_pregel_async.py @@ -9,7 +9,6 @@ from typing import ( AsyncIterator, Generator, Optional, - Self, TypedDict, Union, ) @@ -1030,7 +1029,7 @@ async def test_prebuilt_chat() -> None: from langchain_core.messages import AIMessage, FunctionMessage, HumanMessage class FakeFuntionChatModel(FakeMessagesListChatModel): - def bind_functions(self, functions: list) -> Self: + def bind_functions(self, functions: list): return self @tool() @@ -1179,7 +1178,7 @@ async def test_message_graph() -> None: from langchain_core.messages import AIMessage, FunctionMessage, HumanMessage class FakeFuntionChatModel(FakeMessagesListChatModel): - def bind_functions(self, functions: list) -> Self: + def bind_functions(self, functions: list): return self @tool()