-
Notifications
You must be signed in to change notification settings - Fork 239
/
Copy pathpyproject.toml
104 lines (95 loc) · 2.71 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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
[project]
name = "semantic-router"
version = "0.1.0.dev10"
description = "Super fast semantic router for AI decision making"
authors = [{ name = "Aurelio AI", email = "[email protected]" }]
requires-python = ">=3.9,<3.14"
readme = "README.md"
license = "MIT"
dependencies = [
"pydantic>=2.10.2,<3",
"openai>=1.10.0,<2.0.0",
"numpy>=1.25.2,<2",
"colorlog>=6.8.0,<7",
"pyyaml>=6.0.1,<7",
"aurelio-sdk>=0.0.18,<0.0.19",
"colorama>=0.4.6,<0.5",
"regex>=2023.12.25",
"tiktoken>=0.6.0,<1.0.0",
"requests-mock>=1.12.1,<2",
"aiohttp>=3.10.11,<4",
"psycopg2-binary>=2.9.9,<3",
"tornado>=6.4.2,<7",
]
[project.optional-dependencies]
hybrid = ["pinecone-text>=0.7.1"]
local = [
"torch>=2.1.0,<2.6.0 ; python_version < '3.13'",
"transformers>=4.36.2 ; python_version < '3.13'",
"tokenizers>=0.19 ; python_version < '3.13'",
"llama-cpp-python>=0.2.28,<0.2.86 ; python_version < '3.13'",
]
pinecone = ["pinecone>=5.0.0,<6.0.0"]
vision = [
"torch>=2.1.0,<2.6.0 ; python_version < '3.13'",
"torchvision>=0.17.0,<0.18.0 ; python_version < '3.13'",
"transformers>=4.36.2 ; python_version < '3.13'",
"pillow>=10.2.0,<11.0.0 ; python_version < '3.13'",
]
mistralai = ["mistralai>=0.0.12,<0.1.0"]
qdrant = ["qdrant-client>=1.11.1,<2"]
google = ["google-cloud-aiplatform>=1.45.0,<2"]
bedrock = [
"boto3>=1.34.98,<2",
"botocore>=1.34.110,<2",
]
postgres = []
fastembed = ["fastembed>=0.3.0,<0.4 ; python_version < '3.13'"]
docs = [
"sphinx>=7.0.0,<8",
"sphinxawesome-theme>=5.2.0,<6",
]
cohere = ["cohere>=5.9.4,<6.00"]
all = [
"semantic-router[local]",
"semantic-router[hybrid]",
"semantic-router[pinecone]",
"semantic-router[vision]",
"semantic-router[mistralai]",
"semantic-router[qdrant]",
"semantic-router[google]",
"semantic-router[bedrock]",
"semantic-router[postgres]",
"semantic-router[fastembed]",
"semantic-router[cohere]",
]
[tool.uv.sources]
torch = { git = "https://github.com/pytorch/pytorch.git" }
torchvision = { git = "https://github.com/pytorch/vision.git" }
[dependency-groups]
dev = [
"ipykernel>=6.25.0,<7",
"ruff>=0.1.5,<0.2",
"pytest~=8.2",
"pytest-mock>=3.12.0,<4",
"pytest-cov>=4.1.0,<5",
"pytest-xdist>=3.5.0,<4",
"pytest-asyncio>=0.24.0,<0.25",
"mypy>=1.7.1,<2",
"types-pyyaml>=6.0.12.12,<7",
"types-requests>=2.31.0,<3",
"black[jupyter]>=23.12.1,<24.5.0",
]
[tool.hatch.build.targets.sdist]
include = ["semantic_router"]
[tool.hatch.build.targets.wheel]
include = ["semantic_router"]
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[tool.ruff.per-file-ignores]
"*.ipynb" = ["ALL"]
[tool.ruff]
line-length = 88
[tool.mypy]
ignore_missing_imports = true