From c70376d0b7b37c1cde3f9e5ede28fab934e2932d Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Mon, 26 Feb 2024 10:10:52 -0800 Subject: [PATCH] Use faster/cached create_model in get_input_schema/get_output_schema --- langgraph/pregel/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/langgraph/pregel/__init__.py b/langgraph/pregel/__init__.py index 5f3064382..4c91814ab 100644 --- a/langgraph/pregel/__init__.py +++ b/langgraph/pregel/__init__.py @@ -25,7 +25,7 @@ from langchain_core.callbacks.manager import ( CallbackManagerForChainRun, ) from langchain_core.globals import get_debug -from langchain_core.pydantic_v1 import BaseModel, Field, create_model, root_validator +from langchain_core.pydantic_v1 import BaseModel, Field, root_validator from langchain_core.runnables import ( Runnable, RunnableSerializable, @@ -38,6 +38,7 @@ from langchain_core.runnables.config import ( ) from langchain_core.runnables.utils import ( ConfigurableFieldSpec, + create_model, get_unique_config_specs, ) from langchain_core.tracers.log_stream import LogStreamCallbackHandler