Include openai in the pip requirements for morpheus_llm

nv-morpheus · Dec 17, 2024 · 5eaec88 · 5eaec88
1 parent bd24d98
commit 5eaec88
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 3 deletions.
diff --git a/conda/environments/all_cuda-125_arch-x86_64.yaml b/conda/environments/all_cuda-125_arch-x86_64.yaml
@@ -80,7 +80,7 @@ dependencies:
 - numexpr
 - numpydoc=1.5
 - onnx=1.15
-- openai=1.13
+- openai==1.13.*
 - papermill=2.4.0
 - pip
 - pkg-config=0.29

diff --git a/conda/environments/examples_cuda-125_arch-x86_64.yaml b/conda/environments/examples_cuda-125_arch-x86_64.yaml
@@ -38,7 +38,7 @@ dependencies:
 - numexpr
 - numpydoc=1.5
 - onnx=1.15
-- openai=1.13
+- openai==1.13.*
 - papermill=2.4.0
 - pip
 - pluggy=1.3

diff --git a/dependencies.yaml b/dependencies.yaml
@@ -436,6 +436,7 @@ dependencies:
            - &langchain-nvidia-ai-endpoints langchain-nvidia-ai-endpoints==0.0.11
            - &langchain-openai langchain-openai==0.1.3
            - milvus==2.3.5 # update to match pymilvus when available
+           - &openai openai==1.13.*
            - pymilvus==2.3.6
            - &nemollm nemollm==0.3.5
 
@@ -494,7 +495,7 @@ dependencies:
           - newspaper3k=0.2
           - numexpr
           - onnx=1.15
-          - openai=1.13
+          - *openai
           - pypdf=3.17.4
           - *pypdfium2
           - *python-docx

diff --git a/python/morpheus_llm/morpheus_llm/requirements_morpheus_llm.txt b/python/morpheus_llm/morpheus_llm/requirements_morpheus_llm.txt
@@ -8,5 +8,6 @@ langchain-openai==0.1.3
 langchain==0.1.16
 milvus==2.3.5
 nemollm==0.3.5
+openai==1.13.*
 pymilvus==2.3.6
 torch==2.4.0+cu124