Minor updates to tests from upstream.

langgraph-tests/tests/test_interruption.py

@@ -1,7 +1,6 @@
-from typing import TypedDict
-
 import pytest
 from pytest_mock import MockerFixture
+from typing_extensions import TypedDict
 
 from langgraph.graph import END, START, StateGraph
 from tests.conftest import (

langgraph-tests/tests/test_large_cases.py

@@ -4,13 +4,14 @@
 import time
 from contextlib import contextmanager
 from dataclasses import replace
-from typing import Annotated, Any, Iterator, Literal, Optional, TypedDict, Union, cast
+from typing import Annotated, Any, Iterator, Literal, Optional, Union, cast
 
 import httpx
 import pytest
 from langchain_core.runnables import RunnableConfig, RunnableMap, RunnablePick
 from pytest_mock import MockerFixture
 from syrupy import SnapshotAssertion
+from typing_extensions import TypedDict
 
 from langgraph.channels.context import Context
 from langgraph.channels.last_value import LastValue

langgraph-tests/tests/test_large_cases_async.py

@@ -9,7 +9,6 @@
     AsyncIterator,
     Literal,
     Optional,
-    TypedDict,
     Union,
     cast,
 )
@@ -21,6 +20,7 @@
 from pydantic import BaseModel
 from pytest_mock import MockerFixture
 from syrupy import SnapshotAssertion
+from typing_extensions import TypedDict
 
 from langgraph.channels.context import Context
 from langgraph.channels.last_value import LastValue

langgraph-tests/tests/test_prebuilt.py

@@ -72,6 +72,9 @@ def bind_tools(
         tools: Sequence[Union[Dict[str, Any], Type[BaseModel], Callable, BaseTool]],
         **kwargs: Any,
     ) -> Runnable[LanguageModelInput, BaseMessage]:
+        if len(tools) == 0:
+            raise ValueError("Must provide at least one tool")
+
         tool_dicts = []
         for tool in tools:
             if not isinstance(tool, BaseTool):

langgraph-tests/tests/test_pregel.py

@@ -11,7 +11,6 @@
     Iterator,
     Literal,
     Optional,
-    TypedDict,
     Union,
 )
 
@@ -22,6 +21,7 @@
 )
 from pytest_mock import MockerFixture
 from syrupy import SnapshotAssertion
+from typing_extensions import TypedDict
 
 from langgraph.channels.binop import BinaryOperatorAggregate
 from langgraph.channels.context import Context

langgraph-tests/tests/test_pregel_async.py

@@ -14,9 +14,7 @@
     Dict,
     Literal,
     Optional,
-    TypedDict,
     Union,
-    cast,
 )
 
 import httpx
@@ -27,6 +25,7 @@
 from langchain_core.utils.aiter import aclosing
 from pytest_mock import MockerFixture
 from syrupy import SnapshotAssertion
+from typing_extensions import TypedDict
 
 from langgraph.channels.binop import BinaryOperatorAggregate
 from langgraph.channels.context import Context

pyproject.toml

@@ -32,7 +32,7 @@ mypy = "^1.10.0"
 pymysql = "^1.1.1"
 aiomysql = "^0.2.0"
 types-PyMySQL = "^1.1.0"
-langgraph = "0.2.58"
+langgraph = "0.2.61"
 syrupy = "^4.0.2"
 pytest-repeat = "^0.9.3"
 pytest-xdist = {extras = ["psutil"], version = "^3.6.1"}