From 2d6ddd0a1d77f96342a0705d0cc68d01f4be6d64 Mon Sep 17 00:00:00 2001 From: vbarda Date: Thu, 21 Nov 2024 14:30:23 -0500 Subject: [PATCH] langgraph: fix issue w/ type annotations in tools_condition --- libs/langgraph/langgraph/prebuilt/tool_node.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/libs/langgraph/langgraph/prebuilt/tool_node.py b/libs/langgraph/langgraph/prebuilt/tool_node.py index cdcbdd819..3c43bfba8 100644 --- a/libs/langgraph/langgraph/prebuilt/tool_node.py +++ b/libs/langgraph/langgraph/prebuilt/tool_node.py @@ -1,11 +1,8 @@ -from __future__ import annotations - import asyncio import inspect import json from copy import copy from typing import ( - TYPE_CHECKING, Any, Callable, Dict, @@ -35,15 +32,13 @@ from langchain_core.runnables.utils import Input from langchain_core.tools import BaseTool, InjectedToolArg from langchain_core.tools import tool as create_tool from langchain_core.tools.base import get_all_basemodel_annotations +from pydantic import BaseModel from typing_extensions import Annotated, get_args, get_origin from langgraph.errors import GraphInterrupt from langgraph.store.base import BaseStore from langgraph.utils.runnable import RunnableCallable -if TYPE_CHECKING: - from pydantic import BaseModel - INVALID_TOOL_NAME_ERROR_TEMPLATE = ( "Error: {requested_tool} is not a valid tool, try one of [{available_tools}]." )