semantic-router

1
[tool.poetry]
2
name = "semantic-router"
3
version = "0.0.38"
4
description = "Super fast semantic router for AI decision making"
5
authors = [
6
    "James Briggs <james@aurelio.ai>",
7
    "Siraj Aizlewood <siraj@aurelio.ai>",
8
    "Simonas Jakubonis <simonas@aurelio.ai>",
9
    "Luca Mannini <luca@aurelio.ai>",
10
    "Bogdan Buduroiu <bogdan@aurelio.ai>",
11
    "Ismail Ashraq <ashraq@aurelio.ai>",
12
    "Daniel Griffin <daniel@aurelio.ai>"
13
]
14
readme = "README.md"
15
packages = [{include = "semantic_router"}]
16

17
[tool.poetry.dependencies]
18
python = ">=3.9,<3.13"
19
pydantic = "^2.5.3"
20
openai = "^1.10.0"
21
cohere = "^4.32"
22
mistralai= {version = "^0.0.12", optional = true}
23
numpy = "^1.25.2"
24
colorlog = "^6.8.0"
25
pyyaml = "^6.0.1"
26
pinecone-text = {version = "^0.7.1", optional = true}
27
fastembed = {version = "^0.2.4", optional = true, python = "<3.12"}
28
torch = {version = "^2.1.0", optional = true}
29
transformers = {version = "^4.36.2", optional = true}
30
llama-cpp-python = {version = "^0.2.28", optional = true}
31
colorama = "^0.4.6"
32
pinecone-client = {version="^3.0.0", optional = true}
33
regex = "^2023.12.25"
34
torchvision = { version = "^0.17.0", optional = true}
35
pillow = { version = "^10.2.0", optional = true}
36
tiktoken = "^0.6.0"
37
matplotlib = { version = "^3.8.3", optional = true}
38
qdrant-client = {version = "^1.8.0", optional = true}
39
google-cloud-aiplatform = {version = "^1.45.0", optional = true}
40
requests-mock = "^1.12.1"
41

42
[tool.poetry.extras]
43
hybrid = ["pinecone-text"]
44
fastembed = ["fastembed"]
45
local = ["torch", "transformers", "llama-cpp-python"]
46
pinecone = ["pinecone-client"]
47
vision = ["torch", "torchvision", "transformers", "pillow"]
48
processing = ["matplotlib"]
49
mistralai = ["mistralai"]
50
qdrant = ["qdrant-client"]
51
google = ["google-cloud-aiplatform"]
52

53
[tool.poetry.group.dev.dependencies]
54
ipykernel = "^6.25.0"
55
ruff = "^0.1.5"
56
pytest = "^7.4.3"
57
pytest-mock = "^3.12.0"
58
pytest-cov = "^4.1.0"
59
pytest-xdist = "^3.5.0"
60
mypy = "^1.7.1"
61
types-pyyaml = "^6.0.12.12"
62
types-requests = "^2.31.0"
63
black = "^24.0.0"
64

65
[build-system]
66
requires = ["poetry-core"]
67
build-backend = "poetry.core.masonry.api"
68

69
[tool.ruff.per-file-ignores]
70
"*.ipynb" = ["ALL"]
71

72
[tool.ruff]
73
line-length = 88
74

75
[tool.mypy]
76
ignore_missing_imports = true