kani/pyproject.toml at main · zhudotexe/kani · 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
110
111
112
113
114
115
116
117
118
119
[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[project]
name = "kani"
dynamic = ["version"]
authors = [
    { name = "Andrew Zhu", email = "andrew@zhu.codes" },
    { name = "Liam Dugan", email = "ldugan@seas.upenn.edu" },
    { name = "Alyssa Hwang", email = "ahwang16@seas.upenn.edu" },
]
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[multimodal,mcp,huggingface,llama,cpp,vllm,openai,anthropic,google]",
]

# extensions
multimodal = [
    "kani-multimodal-core<2.0.0",
    "transformers>=4.56.1,<6.0.0", # v4.56.1: ProcessorMixin.__call__ update
]

mcp = [
    "mcp>=1.15.0,<2.0.0", # 1.15.0: ClientSessionGroup
]

# providers
huggingface = [
    "transformers>=4.28.0,<6.0.0",
    "huggingface-hub",
]

llama = [
    "protobuf",
    "sentencepiece>=0.1.99,<1.0.0",
]

cpp = [
    "llama-cpp-python>=0.2.0,<1.0.0",
    "huggingface-hub",
    "sentencepiece>=0.1.99,<1.0.0",
]

vllm = [
    "kani-ext-vllm",
]

openai = [
    "openai>=1.26.0,<3.0.0",
    "tiktoken>=0.7.0,<1.0.0",
]

anthropic = [
    "anthropic>=0.27.0,<1.0.0",
]

google = [
    "google-genai>=1.0.0,<2.0.0",
]

[project.urls]
"Homepage" = "https://github.com/zhudotexe/kani"
"Bug Tracker" = "https://github.com/zhudotexe/kani/issues"

[project.scripts]
kani = "kani._cli:main"

[tool.hatch]
version.path = "kani/_version.py"

[tool.black]
line-length = 120
preview = true
unstable = true

[tool.pytest.ini_options]
asyncio_mode = "auto"
asyncio_default_fixture_loop_scope = "session"
asyncio_default_test_loop_scope = "session"  # to allow sharing async engine fixtures correctly
markers = [
    "local: these tests use a local language model.",
    "e2e: these tests run end-to-end, use KANI_E2E_HYDRATE to hydrate cache.",
    "request_model_capabilities",
]

[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

[tool.coverage.run]
omit = [
    "sandbox/*",
    "examples/*",
    "docs/*",
]