pocket-agent-cli/pyproject.toml at main · vimeto/pocket-agent-cli · 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
[project]
name = "pocket-agent-cli"
version = "0.1.0"
description = "A CLI tool for local LLM inference and benchmarking with comprehensive performance monitoring"
readme = "README.md"
requires-python = ">=3.11"
dependencies = [
    "llama-cpp-python",  # Version will be latest available
    "click",
    "rich",
    "httpx",
    "psutil",
    "docker",
    "jinja2",
    "pydantic",
    "tqdm",
    "aiofiles",
    "huggingface-hub",
    "line-profiler",
    "torch",
    "torchvision",
    "torchaudio",
    "matplotlib",
    "pandas",
    "seaborn",
]

[project.optional-dependencies]
mlx = [
    "mlx-lm>=0.31",
]
test = [
    "pytest>=7.0",
    "pytest-asyncio>=0.21",
    "pytest-cov>=4.0",
    "pytest-mock>=3.10",
]

[project.scripts]
pocket-agent = "pocket_agent_cli.cli:main"

[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["tests"]
python_files = ["test_*.py"]
python_functions = ["test_*"]
addopts = "-v --tb=short"
filterwarnings = [
    "ignore::DeprecationWarning",
    "ignore::UserWarning",
]

[tool.coverage.run]
source = ["pocket_agent_cli"]
omit = ["*/tests/*", "*/__pycache__/*"]

[tool.coverage.report]
exclude_lines = [
    "pragma: no cover",
    "if __name__ == .__main__.:",
    "raise NotImplementedError",
]

[tool.setuptools.packages.find]
include = ["pocket_agent_cli*"]

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"