-
Notifications
You must be signed in to change notification settings - Fork 1
/
pyproject.toml
122 lines (117 loc) · 3.95 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
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
120
121
122
[tool.poetry]
name = "prompto"
version = "0.2.0"
description = "Library for asynchronous querying of LLM API endpoints and logging progress"
authors = [
"rchan <[email protected]>",
"fedenanni <[email protected]>",
"evelinag <[email protected]>",
"edchapman88 <[email protected]>",
"edwinb12 <[email protected]>"
]
readme = "README.md"
[tool.poetry.dependencies]
python = ">=3.11,<4.0"
tqdm = "^4.66.4"
python-dotenv = "^1.0.1"
pandas = "^2.2.3"
black = { version = "^24.3.0", optional = true }
isort = { version = "^5.13.2", optional = true }
pre-commit = { version = "^3.7.0", optional = true }
pytest = { version = "^8.1.1", optional = true }
pytest-asyncio = { version = "^0.23.6", optional = true }
pytest-cov = { version = "^5.0.0", optional = true }
ipykernel = { version = "^6.29.4", optional = true }
mkdocs-material = { version = "^9.5.26", optional = true }
mkdocstrings-python = { version = "^1.10.3", optional = true }
mkdocs-gen-files = { version = "^0.5.0", optional = true }
mkdocs-literate-nav = { version = "^0.6.1", optional = true }
mkdocs-section-index = { version = "^0.3.9", optional = true }
mkdocs-same-dir = { version = "^0.1.3", optional = true }
mkdocs-jupyter = { version = "^0.24.7", optional = true }
cli-test-helpers = { version = "^4.0.0", optional = true }
vertexai = { version = "^1.66.0", optional = true }
google-cloud-aiplatform = { version = "^1.66.0", optional = true }
google-generativeai = { version = "^0.8.1", optional = true }
openai = { version = "^1.35.3", optional = true }
pillow = { version = "^10.4.0", optional = true }
ollama = { version = "^0.3.1", optional = true }
huggingface-hub = { version = "^0.23.4", optional = true }
quart = { version = "^0.19.6", optional = true }
transformers = { version = "^4.44.2", optional = true }
torch = { version = "^2.4.1", optional = true }
accelerate = { version = "^0.34.2", optional = true }
aiohttp = { version = "^3.9.5", optional = true }
anthropic = { version = "^0.31.1", optional = true }
[tool.poetry.extras]
all = [
"black",
"isort",
"pre-commit",
"pytest",
"pytest-asyncio",
"pytest-cov",
"ipython",
"ipykernel",
"mkdocs-material",
"mkdocstrings-python",
"mkdocs-gen-files",
"mkdocs-literate-nav",
"mkdocs-section-index",
"mkdocs-same-dir",
"mkdocs-jupyter",
"cli-test-helpers",
"vertexai",
"google-cloud-aiplatform",
"google-generativeai",
"openai",
"pillow",
"ollama",
"huggingface-hub",
"quart",
"transformers",
"torch",
"accelerate",
"aiohttp",
"anthropic"
]
dev = [
"black",
"isort",
"pre-commit",
"pytest",
"pytest-asyncio",
"pytest-cov",
"ipython",
"ipykernel",
"mkdocs-material",
"mkdocstrings-python",
"mkdocs-gen-files",
"mkdocs-literate-nav",
"mkdocs-section-index",
"mkdocs-same-dir",
"mkdocs-jupyter",
"cli-test-helpers",
]
gemini = ["vertexai", "google-cloud-aiplatform", "google-generativeai", "pillow"]
vertexai = ["vertexai", "google-cloud-aiplatform", "google-generativeai", "pillow"]
azure_openai = ["openai", "pillow"]
openai = ["openai", "pillow"]
ollama = ["ollama"]
huggingface_tgi = ["openai", "huggingface-hub"]
quart = ["quart", "transformers", "torch", "accelerate", "aiohttp"]
anthropic = ['anthropic']
[tool.pytest.ini_options]
log_format = "%(asctime)s %(levelname)s %(message)s"
log_date_format = "%Y-%m-%d %H:%M:%S"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.poetry.scripts]
prompto_run_pipeline = "prompto.scripts.run_pipeline:main"
prompto_run_experiment = "prompto.scripts.run_experiment:cli"
prompto_check_experiment = "prompto.scripts.check_experiment:main"
prompto_create_judge_file = "prompto.scripts.create_judge_file:main"
prompto_obtain_missing_jsonl = "prompto.scripts.obtain_missing_id_jsonl:main"
prompto_convert_images = "prompto.scripts.convert_images:main"
prompto_quart_server = "prompto.apis.quart.quart_api:main"