diff --git a/libs/langgraph/bench/sequential.py b/libs/langgraph/bench/sequential.py index 756e3e7d4..3ab92912f 100644 --- a/libs/langgraph/bench/sequential.py +++ b/libs/langgraph/bench/sequential.py @@ -32,6 +32,8 @@ if __name__ == "__main__": import asyncio import time + import uvloop + graph = create_sequential(2000).compile() input = {"messages": []} # Empty list of messages config = {"recursion_limit": 20000000000} @@ -39,6 +41,7 @@ if __name__ == "__main__": async def run(): len([c async for c in graph.astream(input, config=config)]) + uvloop.install() start = time.time() asyncio.run(run()) end = time.time()