Re-enable benchmarks

langchain-ai · Sep 12, 2024 · 26c52d7 · 26c52d7
1 parent 1d2917e
commit 26c52d7
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 32 deletions.
diff --git a/.github/workflows/bench.yml b/.github/workflows/bench.yml
@@ -16,8 +16,8 @@ jobs:
         working-directory: libs/langgraph
     steps:
       - uses: actions/checkout@v4
-        # with:
-        #   ref: main
+        with:
+          ref: main
       - name: Set up Python 3.11 + Poetry ${{ env.POETRY_VERSION }}
         uses: "./.github/actions/poetry_setup"
         with:

diff --git a/libs/langgraph/bench/__main__.py b/libs/langgraph/bench/__main__.py
@@ -24,36 +24,36 @@ async def run(graph: Pregel, input: dict, config: Optional[dict]):
         },
         None,
     ),
-    # (
-    #     "fanout_to_subgraph_10x_checkpoint",
-    #     fanout_to_subgraph().compile(checkpointer=MemorySaver()),
-    #     {
-    #         "subjects": [
-    #             random.choices("abcdefghijklmnopqrstuvwxyz", k=1000) for _ in range(10)
-    #         ]
-    #     },
-    #     {"configurable": {"thread_id": "1"}},
-    # ),
-    # (
-    #     "fanout_to_subgraph_100x",
-    #     fanout_to_subgraph().compile(checkpointer=None),
-    #     {
-    #         "subjects": [
-    #             random.choices("abcdefghijklmnopqrstuvwxyz", k=1000) for _ in range(100)
-    #         ]
-    #     },
-    #     None,
-    # ),
-    # (
-    #     "fanout_to_subgraph_100x_checkpoint",
-    #     fanout_to_subgraph().compile(checkpointer=MemorySaver()),
-    #     {
-    #         "subjects": [
-    #             random.choices("abcdefghijklmnopqrstuvwxyz", k=1000) for _ in range(100)
-    #         ]
-    #     },
-    #     {"configurable": {"thread_id": "1"}},
-    # ),
+    (
+        "fanout_to_subgraph_10x_checkpoint",
+        fanout_to_subgraph().compile(checkpointer=MemorySaver()),
+        {
+            "subjects": [
+                random.choices("abcdefghijklmnopqrstuvwxyz", k=1000) for _ in range(10)
+            ]
+        },
+        {"configurable": {"thread_id": "1"}},
+    ),
+    (
+        "fanout_to_subgraph_100x",
+        fanout_to_subgraph().compile(checkpointer=None),
+        {
+            "subjects": [
+                random.choices("abcdefghijklmnopqrstuvwxyz", k=1000) for _ in range(100)
+            ]
+        },
+        None,
+    ),
+    (
+        "fanout_to_subgraph_100x_checkpoint",
+        fanout_to_subgraph().compile(checkpointer=MemorySaver()),
+        {
+            "subjects": [
+                random.choices("abcdefghijklmnopqrstuvwxyz", k=1000) for _ in range(100)
+            ]
+        },
+        {"configurable": {"thread_id": "1"}},
+    ),
 )