forked from stanfordnlp/dspy
-
Notifications
You must be signed in to change notification settings - Fork 0
/
setup.py
61 lines (55 loc) · 2.12 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
from setuptools import find_packages, setup
import os
# Read the content of the README file
with open("README.md", encoding="utf-8") as f:
long_description = f.read()
# Read the content of the requirements.txt file
with open("requirements.txt", encoding="utf-8") as f:
requirements = f.read().splitlines()
metadata = {}
here = os.path.abspath(os.path.dirname(__file__))
with open(os.path.join(here, "dspy", "__metadata__.py"), "r", encoding="utf-8") as f:
exec(f.read(), metadata)
f.close()
setup(
name=metadata["__name__"],
version=metadata["__version__"],
description=metadata["__description__"],
url=metadata["__url__"],
author=metadata["__author__"],
author_email=metadata["__author_email__"],
long_description=long_description,
long_description_content_type="text/markdown",
license="MIT License",
packages=find_packages(include=["dspy.*", "dspy"]),
python_requires=">=3.9",
install_requires=requirements,
extras_require={
"chromadb": ["chromadb~=0.4.14"],
"lancedb": ["lancedb~=0.11.0"],
"qdrant": ["qdrant-client", "fastembed"],
"marqo": ["marqo~=3.1.0"],
"mongodb": ["pymongo~=3.12.0"],
"pinecone": ["pinecone-client~=2.2.4"],
"weaviate": ["weaviate-client~=4.6.5"],
"faiss-cpu": ["sentence_transformers", "faiss-cpu"],
"milvus": ["pymilvus~=2.3.7"],
"google-vertex-ai": ["google-cloud-aiplatform==1.43.0"],
"myscale":["clickhouse-connect"],
"snowflake": ["snowflake-snowpark-python"],
"fastembed": ["fastembed"],
"groq": ["groq~=0.8.0"],
"langfuse": ["langfuse~=2.36.1"],
"pgvector": ["psycopg2~=2.9.9","pgvector~=0.2.5"],
"falkordb": ["falkordb", "redis", "async-timeout"]
},
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"License :: OSI Approved :: MIT License",
"Operating System :: POSIX :: Linux",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.8",
"Programming Language :: Python :: 3.9",
],
)