-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathpyproject.toml
More file actions
229 lines (201 loc) · 5.8 KB
/
pyproject.toml
File metadata and controls
229 lines (201 loc) · 5.8 KB
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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
[project]
name = "arctic-text2sql-agent"
version = "0.1.0"
description = "Self-Correcting AI Agent for Natural Language to SQL powered by Snowflake's Arctic-Text2SQL-R1 model"
readme = "README.md"
license = { text = "MIT" }
requires-python = ">=3.10"
authors = [
{ name = "Sakeeb91", email = "rahman.sakeeb@gmail.com" }
]
keywords = ["text2sql", "nlp", "sql", "ai", "agent", "llm", "snowflake", "arctic"]
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",
"Topic :: Database",
]
[project.urls]
Homepage = "https://github.com/Sakeeb91/arctic-text2sql-agent"
Repository = "https://github.com/Sakeeb91/arctic-text2sql-agent"
Issues = "https://github.com/Sakeeb91/arctic-text2sql-agent/issues"
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"
[tool.setuptools.packages.find]
where = ["."]
include = ["app*", "db*", "models*"]
# Black configuration
[tool.black]
line-length = 88
target-version = ["py310", "py311", "py312"]
include = '\.pyi?$'
exclude = '''
/(
\.eggs
| \.git
| \.hg
| \.mypy_cache
| \.tox
| \.venv
| _build
| buck-out
| build
| dist
| migrations
)/
'''
# Ruff configuration
[tool.ruff]
target-version = "py310"
line-length = 88
fix = true
[tool.ruff.lint]
select = [
"E", # pycodestyle errors
"W", # pycodestyle warnings
"F", # Pyflakes
"I", # isort
"B", # flake8-bugbear
"C4", # flake8-comprehensions
"UP", # pyupgrade
"ARG", # flake8-unused-arguments
"SIM", # flake8-simplify
]
ignore = [
"E501", # line too long (handled by black)
"B008", # do not perform function calls in argument defaults
"C901", # too complex
"ARG001", # unused function argument (common in framework callbacks)
]
[tool.ruff.lint.per-file-ignores]
"tests/*" = ["ARG"]
[tool.ruff.lint.isort]
known-first-party = ["app", "db", "models"]
# MyPy configuration
[tool.mypy]
python_version = "3.10"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
check_untyped_defs = true
disallow_untyped_decorators = true
no_implicit_optional = true
warn_redundant_casts = true
warn_unused_ignores = true
warn_no_return = true
follow_imports = "silent"
ignore_missing_imports = true
show_error_codes = true
[[tool.mypy.overrides]]
module = "tests.*"
disallow_untyped_defs = false
[[tool.mypy.overrides]]
module = "models.*"
disallow_untyped_decorators = false
warn_return_any = false
warn_unused_ignores = false
[[tool.mypy.overrides]]
module = "app.routes"
disallow_untyped_decorators = false
[[tool.mypy.overrides]]
module = "app.routes_databases"
disallow_untyped_decorators = false
[[tool.mypy.overrides]]
module = "app.routes_examples"
disallow_untyped_decorators = false
disallow_untyped_defs = false
[[tool.mypy.overrides]]
module = "app.routes_feedback"
disallow_untyped_decorators = false
disallow_untyped_defs = false
[[tool.mypy.overrides]]
module = "app.routes_models"
disallow_untyped_decorators = false
disallow_untyped_defs = false
[[tool.mypy.overrides]]
module = "app.main"
disallow_untyped_decorators = false
[[tool.mypy.overrides]]
module = "app.agent.tools"
disallow_untyped_decorators = false
warn_unused_ignores = false
[[tool.mypy.overrides]]
module = "app.middleware"
warn_unused_ignores = false
[[tool.mypy.overrides]]
module = "app.security.rate_limiting"
warn_unused_ignores = false
[[tool.mypy.overrides]]
module = "db.executor"
warn_unused_ignores = false
disallow_untyped_decorators = false
[[tool.mypy.overrides]]
module = "app.error_handlers"
warn_unused_ignores = false
[[tool.mypy.overrides]]
module = "app.retry"
warn_unused_ignores = false
[[tool.mypy.overrides]]
module = "app.monitoring.endpoints"
disallow_untyped_decorators = false
warn_unused_ignores = false
# Pytest configuration
[tool.pytest.ini_options]
testpaths = ["tests"]
python_files = ["test_*.py", "*_test.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
addopts = "-v --tb=short"
asyncio_mode = "auto"
filterwarnings = [
"ignore::DeprecationWarning",
"ignore::PendingDeprecationWarning",
]
markers = [
"slow: marks tests as slow (deselect with '-m \"not slow\"')",
"integration: marks tests as integration tests",
"unit: marks tests as unit tests",
"e2e: End-to-end tests with real model/database",
"e2e_slow: Slow E2E tests (model loading)",
"e2e_multidb: Multi-database E2E tests",
"e2e_streaming: Streaming endpoint E2E tests",
"e2e_performance: Performance benchmark E2E tests",
]
# Coverage configuration
[tool.coverage.run]
source = ["app", "db", "models"]
branch = true
omit = [
"*/tests/*",
"*/__pycache__/*",
"*/migrations/*",
]
[tool.coverage.report]
exclude_lines = [
"pragma: no cover",
"def __repr__",
"raise AssertionError",
"raise NotImplementedError",
"if __name__ == .__main__.:",
"if TYPE_CHECKING:",
"class .*\\bProtocol\\):",
"@(abc\\.)?abstractmethod",
]
fail_under = 30 # Temporarily lowered for initial development phase
show_missing = true
# Bandit security linter configuration
[tool.bandit]
exclude_dirs = ["tests", "venv", ".venv"]
skips = [
"B101", # Skip assert warnings (acceptable in our codebase)
"B104", # Skip hardcoded_bind_all_interfaces (0.0.0.0 is intentional for Docker/container deployments)
"B608", # Skip hardcoded_sql_expressions (table names come from schema introspection, not user input)
"B615", # Skip huggingface_unsafe_download (model versioning handled via model name configuration)
]