text2sql-agent/requirements.txt at main · Sakeeb91/text2sql-agent · 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
# Arctic Text2SQL Agent - Full Requirements
#
# For optimized Docker builds, use the split requirements:
#   - requirements-base.txt (API, database, utilities)
#   - requirements-ml.txt (PyTorch, transformers)
#   - requirements-dev.txt (testing, linting)
#
# Full installation: pip install -r requirements.txt
# Production only: pip install -r requirements-base.txt -r requirements-ml.txt

# ============================================================================
# Base Dependencies (API, Database, Utilities)
# ============================================================================

# Core API Framework
fastapi==0.127.0
uvicorn[standard]==0.40.0
pydantic==2.12.5
pydantic-settings==2.12.0
sse-starlette>=2.0.0  # Server-Sent Events for streaming (Issue #8)

# Database
sqlalchemy==2.0.45
psycopg2-binary==2.9.11
alembic==1.17.2

# Async Database Drivers
asyncpg==0.31.0
aiomysql==0.3.2
aiosqlite==0.22.0

# Utilities
python-dotenv==1.2.1
tenacity==9.1.2
pyyaml==6.0.3

# Caching (Issue #8: Performance Optimization)
redis>=5.0.0
hiredis>=2.3.0  # C parser for better Redis performance

# Monitoring & Logging
prometheus-client==0.23.1
structlog==25.5.0

# Distributed Tracing (Issue #9: Monitoring & Observability)
opentelemetry-api>=1.20.0
opentelemetry-sdk>=1.20.0
opentelemetry-instrumentation-fastapi>=0.41b0
opentelemetry-instrumentation-sqlalchemy>=0.41b0
opentelemetry-instrumentation-httpx>=0.41b0
opentelemetry-exporter-otlp>=1.20.0
opentelemetry-exporter-prometheus>=0.41b0

# Security
cryptography>=39.0.0
python-jose[cryptography]==3.5.0
passlib[bcrypt]==1.7.4
slowapi==0.1.9

# ============================================================================
# ML Dependencies (Large packages - ~2GB total)
# ============================================================================

# Core ML
numpy<2  # Required for torch compatibility

# PyTorch - NOTE: ~1.8GB download
# For CPU-only: pip install torch --index-url https://download.pytorch.org/whl/cpu
torch>=2.2.0

# HuggingFace Ecosystem
transformers>=4.40.0
accelerate>=0.30.0
huggingface_hub>=0.31.2
bitsandbytes>=0.42.0

# Agent Framework (Issue #18: smolagents integration)
smolagents>=1.0.0

# ============================================================================
# Development Dependencies (Not needed in production)
# ============================================================================

# Testing
pytest==7.4.4
pytest-asyncio==0.23.4
pytest-cov==7.0.0
httpx==0.28.1
fakeredis>=2.21.0  # Redis mocking for tests (Issue #8)

# Code Quality
black==25.12.0
ruff==0.14.10
mypy==1.19.1
pre-commit==4.5.1