From 18e71469e1d33115d1a21fe3954ce5e6b4562be2 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Thu, 31 Oct 2024 12:57:00 -0700 Subject: [PATCH] Lint --- libs/langgraph/langgraph/graph/state.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/libs/langgraph/langgraph/graph/state.py b/libs/langgraph/langgraph/graph/state.py index 1eba5f7de..fc429cb60 100644 --- a/libs/langgraph/langgraph/graph/state.py +++ b/libs/langgraph/langgraph/graph/state.py @@ -4,6 +4,7 @@ import typing import warnings from functools import partial from inspect import isclass, isfunction, ismethod, signature +from types import FunctionType from typing import ( Any, Callable, @@ -348,7 +349,11 @@ class StateGraph(Graph): ): if input is None: first_parameter_name = next( - iter(inspect.signature(action).parameters.keys()) + iter( + inspect.signature( + cast(FunctionType, action) + ).parameters.keys() + ) ) if input_hint := hints.get(first_parameter_name): if isinstance(input_hint, type) and get_type_hints(input_hint):