From fd8f9e87dcfaaf6177ba793a579fe9b77a00c95e Mon Sep 17 00:00:00 2001
From: vbarda <vadym@langchain.dev>
Date: Mon, 22 Jul 2024 20:48:21 -0400
Subject: [PATCH] update types

---
 libs/langgraph/langgraph/pregel/executor.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/libs/langgraph/langgraph/pregel/executor.py b/libs/langgraph/langgraph/pregel/executor.py
index 69d3b5d31..5da78a26c 100644
--- a/libs/langgraph/langgraph/pregel/executor.py
+++ b/libs/langgraph/langgraph/pregel/executor.py
@@ -6,6 +6,7 @@ from contextvars import copy_context
 from types import TracebackType
 from typing import (
     AsyncContextManager,
+    Awaitable,
     Callable,
     ContextManager,
     Optional,
@@ -102,7 +103,7 @@ class AsyncBackgroundExecutor(AsyncContextManager):
 
     def submit(
         self,
-        fn: Callable[P, T],
+        fn: Callable[P, Awaitable[T]],
         *args: P.args,
         __name__: Optional[str] = None,
         __cancel_on_exit__: bool = False,