From cb95821eb0f0f40918e2ae1999b19ae82dda6e15 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Tue, 18 Mar 2025 22:20:18 -0400 Subject: [PATCH] x --- libs/langgraph/bench/__main__.py | 24 ++++++++++++++++++++++-- 1 file changed, 22 insertions(+), 2 deletions(-) diff --git a/libs/langgraph/bench/__main__.py b/libs/langgraph/bench/__main__.py index d6d57543a..45986e7f0 100644 --- a/libs/langgraph/bench/__main__.py +++ b/libs/langgraph/bench/__main__.py @@ -390,13 +390,33 @@ for name, agraph, graph, input in benchmarks: # Graph compilation times compilation_benchmarks = ( ( - "sequential_1000_compilation", + "sequential_1000", create_sequential(1_000), ), ( - "sequential_10000_compilation", + "sequential_10000", create_sequential(10_000), ), + ( + "pydantic_state_25x300", + pydantic_state(300), + ), + ( + "pydantic_state_15x600", + pydantic_state(600), + ), + ( + "pydantic_state_9x1200", + pydantic_state(1200), + ), + ( + "wide_state_15x600", + wide_state(600), + ), + ( + "wide_state_9x1200", + wide_state(1200), + ), ) for name, graph, input in compilation_benchmarks: