vektori/pyproject.toml at 25ba2ab71bf2339673fe328bb7d12c4dd44fe83d · snigenigmatic/vektori · GitHub

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
105
106
107
108
109
[build-system]
requires = ["setuptools>=68", "setuptools-scm>=8"]
build-backend = "setuptools.build_meta"

[project]
name = "vektori"
dynamic = ["version"]
description = "Open-source, self-hostable memory engine for AI agents using a three-layer sentence graph architecture."
readme = "README.md"
license = { file = "LICENSE" }
authors = [{ name = "Laxman & Manit" }]
requires-python = ">=3.10"
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    "nltk>=3.8",
    "numpy>=1.24",
    "aiosqlite>=0.19",
    "openai>=1.12",
    "httpx>=0.27",
    "pydantic>=2.5",
    "typer>=0.9",
]

[project.optional-dependencies]
postgres = ["asyncpg>=0.29", "pgvector>=0.2"]
neo4j = ["neo4j>=5.14"]
qdrant = ["qdrant-client>=1.7"]
milvus = ["pymilvus>=2.5.3"]
anthropic = ["anthropic>=0.20", "voyageai>=0.2"]
sentence-transformers = ["sentence-transformers>=2.6"]
bge = ["FlagEmbedding>=1.2"]
litellm = ["litellm>=1.30"]
dev = [
    "pytest>=8.0",
    "pytest-asyncio>=0.23",
    "ruff>=0.9",
    "mypy>=1.8",
    "pre-commit>=3.6",
]
all = [
    "asyncpg>=0.29",
    "pgvector>=0.2",
    "neo4j>=5.14",
    "qdrant-client>=1.7",
    "pymilvus>=2.5.3",
    "anthropic>=0.20",
    "voyageai>=0.2",
    "sentence-transformers>=2.6",
    "FlagEmbedding>=1.2",
    "litellm>=1.30",
]

[project.scripts]
vektori = "vektori.cli:main"

[project.urls]
Homepage = "https://github.com/vektori-ai/vektori"
Repository = "https://github.com/vektori-ai/vektori"
Issues = "https://github.com/vektori-ai/vektori/issues"

[tool.setuptools.packages.find]
where = ["."]
include = ["vektori*"]

[tool.setuptools_scm]
write_to = "vektori/_version.py"
fallback_version = "0.0.0"
local_scheme = "no-local-version"

[tool.ruff]
line-length = 100
target-version = "py310"
exclude = [
    "benchmarks/",
    "examples/",
    "scripts/",
    "vektori/_version.py",
    "check_session.py",
    "fix_cache.py",
    "test_longmemeval_setup.py",
    "run_locomo.py",
    "run_longmemeval_s.py",
]

[tool.ruff.lint]
select = ["E", "F", "I", "N", "W", "UP"]
ignore = ["E501"]

[tool.ruff.format]
quote-style = "double"

[tool.mypy]
python_version = "3.10"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = false

[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["tests"]