Skip to content

Commit

Permalink
INTPYTHON-427 Add langgraph to ai-ml-testing pipeline
Browse files Browse the repository at this point in the history
  • Loading branch information
blink1073 committed Dec 9, 2024
1 parent e94c697 commit e387e7d
Show file tree
Hide file tree
Showing 3 changed files with 62 additions and 2 deletions.
30 changes: 29 additions & 1 deletion .evergreen/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -225,7 +239,7 @@ buildvariants:
expansions:
DIR: langchain-python
REPO_NAME: langchain-mongodb
CLONE_URL: https://github.com/langchain-ai/langchain-mongodb.git
CLONE_URL: --branch INTPYTHON-447-3 https://github.com/langchain-ai/langchain-mongodb.git
DATABASE: langchain_test_db
run_on:
- rhel87-small
Expand All @@ -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: --branch INTPYTHON-447-3 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:
Expand Down
2 changes: 1 addition & 1 deletion langchain-python/run.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
32 changes: 32 additions & 0 deletions langgraph-python/run.sh
Original file line number Diff line number Diff line change
@@ -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

0 comments on commit e387e7d

Please sign in to comment.