Skip to content

Commit 9031331

Browse files
sdks/python: add openai as extra dependency
1 parent 1315494 commit 9031331

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

sdks/python/setup.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -161,6 +161,7 @@ def cythonize(*args, **kwargs):
161161
]
162162

163163
milvus_dependency = ['pymilvus>=2.5.10,<3.0.0']
164+
openai_dependency = ['openai>=1.0.0,<2.0.0']
164165

165166

166167
def find_by_ext(root_dir, ext):
@@ -418,9 +419,8 @@ def get_portability_package_data():
418419
'docutils>=0.18.1',
419420
'markdown',
420421
'pandas<2.3.0',
421-
'openai',
422422
'virtualenv-clone>=0.5,<1.0',
423-
],
423+
] + openai_dependency,
424424
'test': [
425425
'cloud-sql-python-connector[pg8000]>=1.0.0,<2.0.0',
426426
'docstring-parser>=0.15,<1.0',
@@ -596,7 +596,8 @@ def get_portability_package_data():
596596
],
597597
'xgboost': ['xgboost>=1.6.0,<2.1.3', 'datatable==1.0.0'],
598598
'tensorflow-hub': ['tensorflow-hub>=0.14.0,<0.16.0'],
599-
'milvus': milvus_dependency
599+
'milvus': milvus_dependency,
600+
'openai': openai_dependency
600601
},
601602
zip_safe=False,
602603
# PyPI package information.

0 commit comments

Comments
 (0)