forked from aurelio-labs/semantic-router
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
88 lines (81 loc) · 2.76 KB
/
pyproject.toml
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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
[tool.poetry]
name = "semantic-router"
version = "0.0.63"
description = "Super fast semantic router for AI decision making"
authors = [
"James Briggs <[email protected]>",
"Siraj Aizlewood <[email protected]>",
"Simonas Jakubonis <[email protected]>",
"Luca Mannini <[email protected]>",
"Bogdan Buduroiu <[email protected]>",
"Ismail Ashraq <[email protected]>",
"Daniel Griffin <[email protected]>",
"Tolga Arslan <[email protected]>"
]
readme = "README.md"
packages = [{include = "semantic_router"}]
license = "MIT"
[tool.poetry.dependencies]
python = ">=3.9,<3.13"
pydantic = "^2.5.3"
openai = ">=1.10.0,<2.0.0"
cohere = ">=5.00,<6.00"
mistralai= {version = ">=0.0.12,<0.1.0", optional = true}
numpy = "^1.25.2"
colorlog = "^6.8.0"
pyyaml = "^6.0.1"
pinecone-text = {version = ">=0.7.1,<0.10.0", optional = true}
torch = {version = ">=2.1.0,<2.6.0", optional = true}
transformers = {version = ">=4.36.2", optional = true}
tokenizers = {version = ">=0.19", optional = true}
llama-cpp-python = {version = ">=0.2.28,<0.2.86", optional = true}
colorama = "^0.4.6"
pinecone-client = {version=">=3.0.0,<4.0.0", optional = true}
regex = ">=2023.12.25"
torchvision = { version = ">=0.17.0,<0.18.0", optional = true}
pillow = { version = ">=10.2.0,<11.0.0", optional = true}
tiktoken = ">=0.6.0,<1.0.0"
matplotlib = { version = "^3.8.3", optional = true}
qdrant-client = {version = "^1.8.0", optional = true}
google-cloud-aiplatform = {version = "^1.45.0", optional = true}
requests-mock = "^1.12.1"
boto3 = { version = "^1.34.98", optional = true }
botocore = {version = "^1.34.110", optional = true}
aiohttp = "^3.9.5"
fastembed = {version = "^0.3.0", optional = true}
psycopg2 = {version = "^2.9.9", optional = true}
sphinx = {version = "^7.0.0", optional = true}
sphinxawesome-theme = {version = "^5.2.0", optional = true}
[tool.poetry.extras]
hybrid = ["pinecone-text"]
local = ["torch", "transformers", "tokenizers", "huggingface-hub", "llama-cpp-python"]
pinecone = ["pinecone-client"]
vision = ["torch", "torchvision", "transformers", "pillow"]
processing = ["matplotlib"]
mistralai = ["mistralai"]
qdrant = ["qdrant-client"]
google = ["google-cloud-aiplatform"]
bedrock = ["boto3", "botocore"]
postgres = ["psycopg2"]
fastembed = ["fastembed"]
docs = ["sphinx", "sphinxawesome-theme"]
[tool.poetry.group.dev.dependencies]
ipykernel = "^6.25.0"
ruff = "^0.1.5"
pytest = "^7.4.3"
pytest-mock = "^3.12.0"
pytest-cov = "^4.1.0"
pytest-xdist = "^3.5.0"
mypy = "^1.7.1"
types-pyyaml = "^6.0.12.12"
types-requests = "^2.31.0"
black = {extras = ["jupyter"], version = ">=23.12.1,<24.5.0"}
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.ruff.per-file-ignores]
"*.ipynb" = ["ALL"]
[tool.ruff]
line-length = 88
[tool.mypy]
ignore_missing_imports = true