-
Notifications
You must be signed in to change notification settings - Fork 8
PYTHON-4506 Add autogen #28
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
Having the feature branch is a great addition. :)
autogen/run.sh
Outdated
#!/bin/sh | ||
|
||
# Sets up a virtual environment | ||
# Runs the mongodb tests of the upstream repo | ||
# NOTE: In this instance, the tests *programmtically* create_vector_search_index | ||
|
||
set -x | ||
|
||
. $workdir/src/.evergreen/utils.sh | ||
PYTHON_BINARY=$(find_python3) | ||
$PYTHON_BINARY -c "import sys; print(f'Python version found: {sys.version_info}')" | ||
|
||
# Create and activate an isolated python venv environment | ||
$PYTHON_BINARY -m venv venv | ||
source venv/bin/activate | ||
# Install autogen with extras | ||
$PYTHON_BINARY -m pip install .[test,"retrievechat-mongodb"] | ||
|
||
|
||
# Run tests. Sensitive variables in Evergreen come from Evergreen project: ai-ml-pipeline-testing/ | ||
MONGODB_URI=$autogen_mongodb_uri \ | ||
MONGODB_DATABASE="autogen_test_db" \ | ||
py.test -v test/agentchat/contrib/vectordb/test_mongodb.py |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
As an aside:
A lot of this set-up seems easily reproducible. Is it at all worth it to consolidate this script into a singular function?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Nah. It's simple and clean and everyone hates writing bash. :)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just saw this error
[2024/06/24 14:03:37.639] + py.test -v test/agentchat/contrib/vectordb/test_mongodb.py
[2024/06/24 14:03:37.639] ../run.sh: line 21: py.test: command not found
Adding work done together with @ranfysvalle02 (@fabian) on the Autogen framework.
As usual, we are pointing to the feature branch until it is merged. The upstream maintainers have been very responsive.