Skip to content

Commit

Permalink
Add a conda_morpheus_llm key in dependencies.yaml
Browse files Browse the repository at this point in the history
These are pip dependencies needed to run the llm unit tests at the end
of conda build

Signed-off-by: Anuradha Karuppiah <[email protected]>
  • Loading branch information
AnuradhaKaruppiah committed Oct 10, 2024
1 parent 4b3d698 commit 6b38de4
Show file tree
Hide file tree
Showing 3 changed files with 24 additions and 7 deletions.
1 change: 1 addition & 0 deletions ci/conda/recipes/morpheus-libs/meta.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -234,6 +234,7 @@ outputs:
- pytest-cov
- pytest =7.4.4
source_files:
- dependencies.yaml
- pyproject.toml
- scripts/fetch_data.py
- tests/*
Expand Down
12 changes: 5 additions & 7 deletions ci/conda/recipes/morpheus-libs/morpheus_llm_test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -14,12 +14,10 @@
# limitations under the License.

# llm library and tests are dependent on a number of pypi packages - fixme
pip install --no-input milvus==2.3.5
pip install --no-input pymilvus==2.3.6
pip install --no-input langchain==0.1.16
pip install --no-input langchain-nvidia-ai-endpoints==0.0.11
pip install --no-input faiss-gpu==1.7.*
pip install --no-input google-search-results==2.4
pip install --no-input nemollm==0.3.5
rapids-dependency-file-generator \
--output requirements \
--file_key morpheus_llm --matrix "" | tee "llm_requirements.txt"

pip install -r llm_requirements.txt

pytest tests/morpheus_llm
18 changes: 18 additions & 0 deletions dependencies.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -174,6 +174,12 @@ files:
- checks
- python

# pip dependencies that are used during the conda test stage for morpheus-llm output
morpheus_llm:
output: none
includes:
- test_morpheus_llm_pip

channels:
- conda-forge
- huggingface
Expand Down Expand Up @@ -377,6 +383,18 @@ dependencies:
- pip:
- pytest-kafka==0.6.0

test_morpheus_llm_pip:
common:
- output_types: [requirements]
packages:
- milvus==2.3.5 # update to match pymilvus when available
- pymilvus==2.3.6
- langchain==0.1.16
- langchain-nvidia-ai-endpoints==0.0.11
- faiss-gpu==1.7.*
- google-search-results==2.4
- nemollm==0.3.5

example-dfp-prod:
common:
- output_types: [conda]
Expand Down

0 comments on commit 6b38de4

Please sign in to comment.