INTPYTHON-447 Finish integration tests

.github/workflows/_lint.yml

@@ -20,6 +20,7 @@ jobs:
     name: "make lint #${{ matrix.python-version }}"
     runs-on: ubuntu-latest
     strategy:
+      fail-fast: false
       matrix:
         # Only lint on the min and max supported Python versions.
         # It's extremely unlikely that there's a lint issue on any version in between

.github/workflows/_test.yml

@@ -18,6 +18,7 @@ jobs:
         working-directory: ${{ inputs.working-directory }}
     runs-on: ubuntu-latest
     strategy:
+      fail-fast: false
       matrix:
         python-version:
           - "3.9"
@@ -51,9 +52,12 @@ jobs:
           MONGODB_URI: ${{ secrets.MONGODB_ATLAS_URI }}
           OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
         run: |
-          if [[ "${{ github.event_name }}" == "push" ]]; then
-            make integration_tests
-          fi
+          # if [[ "${{ github.event_name }}" == "push" ]]; then
+            # Only run on the min python version.
+            if [[ "${{ matrix.python-version }}" == "3.9" ]]; then
+              make integration_tests
+            fi
+          #fi
         working-directory: ${{ inputs.working-directory }}
 
       - name: Ensure the tests did not create any additional files

.github/workflows/ci.yml

@@ -52,6 +52,7 @@ jobs:
     needs: [ build ]
     if: ${{ needs.build.outputs.dirs-to-lint != '[]' }}
     strategy:
+      fail-fast: false
       matrix:
         working-directory: ${{ fromJson(needs.build.outputs.dirs-to-lint) }}
     uses: ./.github/workflows/_lint.yml
@@ -64,6 +65,7 @@ jobs:
     needs: [ build ]
     if: ${{ needs.build.outputs.dirs-to-test != '[]' }}
     strategy:
+      fail-fast: false
       matrix:
         working-directory: ${{ fromJson(needs.build.outputs.dirs-to-test) }}
     uses: ./.github/workflows/_test.yml

.pre-commit-config.yaml

@@ -12,6 +12,8 @@ repos:
   - id: forbid-new-submodules
   - id: trailing-whitespace
     exclude_types: [json]
+    exclude: |
+      (?x)^(.*.ambr)$
 
 # We use the Python version instead of the original version which seems to require Docker
 # https://github.com/koalaman/shellcheck-precommit

libs/langgraph-checkpoint-mongodb/tests/integration_tests/__snapshots__/test_pregel.ambr

@@ -8,7 +8,7 @@
   	tool_two_slow --> finish;
   	prepare -.-> tool_two_fast;
   	tool_two_fast --> finish;
-
+  
   '''
 # ---
 # name: test_branch_then[mongodb].1
@@ -30,7 +30,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_conditional_entrypoint_graph
@@ -113,7 +113,7 @@
   	__start__ -.  go-left  .-> left;
   	__start__ -.  go-right  .-> right;
   	left -.-> __end__;
-
+  
   '''
 # ---
 # name: test_conditional_entrypoint_graph_state
@@ -196,7 +196,7 @@
   	__start__ -.  go-left  .-> left;
   	__start__ -.  go-right  .-> right;
   	left -.-> __end__;
-
+  
   '''
 # ---
 # name: test_conditional_entrypoint_to_multiple_state_graph
@@ -258,7 +258,7 @@
   	get_weather --> __end__;
   	__start__ -.-> get_weather;
   	__start__ -.-> __end__;
-
+  
   '''
 # ---
 # name: test_conditional_graph[mongodb]
@@ -339,7 +339,7 @@
   	tools --> agent;
   	agent -.  continue  .-> tools;
   	agent -.  exit  .-> __end__;
-
+  
   '''
 # ---
 # name: test_conditional_graph[mongodb].2
@@ -359,7 +359,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_conditional_graph[mongodb].3
@@ -440,7 +440,7 @@
   	tools --> agent;
   	agent -.  continue  .-> tools;
   	agent -.  exit  .-> __end__;
-
+  
   '''
 # ---
 # name: test_conditional_graph[mongodb].5
@@ -533,7 +533,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_conditional_state_graph[mongodb]
@@ -615,7 +615,7 @@
   	tools --> agent;
   	agent -.  continue  .-> tools;
   	agent -.  exit  .-> __end__;
-
+  
   '''
 # ---
 # name: test_conditional_state_graph_with_list_edge_inputs
@@ -687,7 +687,7 @@
   	B --> __end__;
   	__start__ --> A;
   	__start__ --> B;
-
+  
   '''
 # ---
 # name: test_in_one_fan_out_state_graph_waiting_edge[mongodb]
@@ -700,7 +700,7 @@
   	retriever_two --> qa;
   	rewrite_query --> analyzer_one;
   	rewrite_query --> retriever_two;
-
+  
   '''
 # ---
 # name: test_in_one_fan_out_state_graph_waiting_edge_custom_state_class_pydantic1[mongodb]
@@ -713,7 +713,7 @@
   	retriever_two --> qa;
   	rewrite_query --> analyzer_one;
   	rewrite_query -.-> retriever_two;
-
+  
   '''
 # ---
 # name: test_in_one_fan_out_state_graph_waiting_edge_custom_state_class_pydantic1[mongodb].1
@@ -783,7 +783,7 @@
   	retriever_two --> qa;
   	rewrite_query --> analyzer_one;
   	rewrite_query -.-> retriever_two;
-
+  
   '''
 # ---
 # name: test_in_one_fan_out_state_graph_waiting_edge_custom_state_class_pydantic2[mongodb].1
@@ -853,7 +853,7 @@
   	retriever_two --> qa;
   	rewrite_query --> analyzer_one;
   	rewrite_query -.-> retriever_two;
-
+  
   '''
 # ---
 # name: test_message_graph[mongodb]
@@ -934,7 +934,7 @@
   	tools --> agent;
   	agent -.  continue  .-> tools;
   	agent -.  end  .-> __end__;
-
+  
   '''
 # ---
 # name: test_multiple_sinks_subgraphs
@@ -957,7 +957,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_nested_graph
@@ -966,7 +966,7 @@
   	__start__ --> inner;
   	inner --> side;
   	side --> __end__;
-
+  
   '''
 # ---
 # name: test_nested_graph.1
@@ -983,7 +983,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_nested_graph_xray
@@ -1138,7 +1138,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_prebuilt_tool_chat
@@ -1218,7 +1218,7 @@
   	tools --> agent;
   	agent -.-> tools;
   	agent -.-> __end__;
-
+  
   '''
 # ---
 # name: test_repeat_condition
@@ -1234,7 +1234,7 @@
   	Call_Tool -.-> Researcher;
   	Call_Tool -.-> Chart_Generator;
   	Researcher -.  redo  .-> Researcher;
-
+  
   '''
 # ---
 # name: test_send_react_interrupt_control[mongodb]
@@ -1249,7 +1249,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_simple_multi_edge
@@ -1261,7 +1261,7 @@
   	up --> down;
   	up --> other;
   	up --> side;
-
+  
   '''
 # ---
 # name: test_start_branch_then[mongodb]
@@ -1279,7 +1279,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_state_graph_w_config_inherited_state_keys
@@ -1313,7 +1313,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_xray_bool
@@ -1349,7 +1349,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_xray_issue
@@ -1376,7 +1376,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_xray_lance

libs/langgraph-checkpoint-mongodb/tests/integration_tests/__snapshots__/test_pregel_async.ambr

@@ -9,7 +9,7 @@
   	retriever_two --> qa;
   	rewrite_query --> analyzer_one;
   	rewrite_query -.-> retriever_two;
-
+  
   '''
 # ---
 # name: test_in_one_fan_out_state_graph_waiting_edge_custom_state_class_pydantic2[mongodb_aio].1
@@ -132,7 +132,7 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
+  
   '''
 # ---
 # name: test_weather_subgraph[mongodb_aio]
@@ -157,6 +157,5 @@
   	classDef default fill:#f2f0ff,line-height:1.2
   	classDef first fill-opacity:0
   	classDef last fill:#bfb6fc
-
   '''
 # ---