Merge pull request #51 from blink1073/INTPYTHON-427

INTPYTHON-427 Add langgraph to ai-ml-testing pipeline

Author: blink1073

.evergreen/config.yml

@@ -133,6 +133,20 @@ tasks:
       - func: "setup remote atlas"
       - func: "execute tests"
 
+  - name: test-langgraph-python-local
+    tags: [local]
+    commands:
+      - func: "fetch repo"
+      - func: "setup local atlas"
+      - func: "execute tests"
+
+  - name: test-langgraph-python-remote
+    tags: [remote]
+    commands:
+      - func: "fetch repo"
+      - func: "setup remote atlas"
+      - func: "execute tests"
+
   - name: test-chatgpt-retrieval-plugin-local
     tags: [local]
     commands:
@@ -234,6 +248,20 @@ buildvariants:
       - name: test-langchain-python-remote
         batchtime: 10080  # 1 week
 
+  - name: test-langgraph-python-rhel
+    display_name: Langgraph RHEL Python
+    expansions:
+      DIR: langgraph-python
+      REPO_NAME: langchain-mongodb
+      CLONE_URL: https://github.com/langchain-ai/langchain-mongodb.git
+      DATABASE: langgraph-test
+    run_on:
+      - rhel87-small
+    tasks:
+      - name: test-langgraph-python-local
+      - name: test-langgraph-python-remote
+        batchtime: 10080  # 1 week
+
   - name: test-chatgpt-retrieval-plugin-rhel
     display_name: ChatGPT Retrieval Plugin
     expansions:

.evergreen/setup-remote.sh

@@ -19,7 +19,7 @@ case $DIR in
     semantic-kernel-csharp)
         MONGODB_URI=$SEMANTIC_KERNEL_MONGODB_URI
     ;;
-    langchain-python)
+    langchain-python | langgraph-python)
         MONGODB_URI=$LANGCHAIN_MONGODB_URI
     ;;
     chatgpt-retrieval-plugin)

langchain-python/run.sh

@@ -24,7 +24,7 @@ poetry lock --no-update
 
 poetry install --with dev
 
-export MONGODB_ATLAS_URI=$MONGODB_URI
+export MONGODB_URI=$MONGODB_URI
 export OPENAI_API_KEY=$OPENAI_API_KEY
 
 make test

langgraph-python/run.sh

@@ -0,0 +1,32 @@
+#!/bin/bash
+
+# WORKING_DIR = src/langchain-python/langchain
+set -eu
+
+# Get the MONGODB_URI and OPENAI_API_KEY.
+# shellcheck disable=SC2154
+. $workdir/src/env.sh
+
+# shellcheck disable=SC2154
+. $workdir/src/.evergreen/utils.sh
+
+PYTHON_BINARY=$(find_python3)
+
+# shellcheck disable=SC2164
+cd libs/langgraph-checkpoint-mongodb
+
+ $PYTHON_BINARY -m venv venv_pipeline
+ source venv_pipeline/bin/activate
+
+pip install poetry
+
+poetry lock --no-update
+
+poetry install --with dev
+
+export MONGODB_URI=$MONGODB_URI
+export OPENAI_API_KEY=$OPENAI_API_KEY
+
+make test
+
+make integration_test