-
-
Notifications
You must be signed in to change notification settings - Fork 30
/
pyproject.toml
81 lines (69 loc) · 1.85 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
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"
[project]
name = "kani"
dynamic = ["version"]
authors = [
]
description = "kani (カニ) is a lightweight and highly hackable framework for chat-based language models with tool usage/function calling."
readme = "README.md"
license = { file = "LICENSE" }
requires-python = ">=3.10"
classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
# https://pypi.org/classifiers/
"Development Status :: 4 - Beta",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
"pydantic>=2.0.0,<3.0.0",
]
[project.optional-dependencies]
all = [
"kani[huggingface,llama,cpp,openai,anthropic]",
]
huggingface = [
"transformers>=4.28.0,<5.0.0",
]
llama = [
"protobuf",
"sentencepiece>=0.1.99,<1.0.0",
]
cpp = [
"llama-cpp-python>=0.2.0,<1.0.0",
"huggingface-hub",
]
openai = [
"openai>=1.26.0,<2.0.0",
"tiktoken>=0.7.0,<1.0.0",
]
anthropic = [
"anthropic>=0.27.0,<1.0.0",
]
[project.urls]
"Homepage" = "https://github.com/zhudotexe/kani"
"Bug Tracker" = "https://github.com/zhudotexe/kani/issues"
[tool.hatch]
version.path = "kani/_version.py"
[tool.black]
line-length = 120
preview = true
[tool.pytest.ini_options]
asyncio_mode = "auto"
markers = ["llama: these tests use a real LLaMA v2 model."]
[tool.isort]
profile = "black"
skip_gitignore = true
line_length = 120
known_first_party = ["kani"]
# I am bound by pycharm's import autosorting rules
no_lines_before = "LOCALFOLDER"
reverse_relative = true
combine_as_imports = true