User-Modeling-Agent / requirements.txt
Israelbliz's picture
Upload Dockerfile and requirements
1d6ed18 verified
# ── Core ─────────────────────────────────────────────
openai>=1.50.0
langchain>=0.3.0
langchain-openai>=0.2.0
langchain-google-genai>=2.0.0 # Gemini support (optional, only if LLM_PROVIDER=gemini)
langchain-core>=0.3.0
python-dotenv>=1.0.0
tenacity>=8.2.0 # retry decorator for flaky LLM calls
# ── Data ─────────────────────────────────────────────
pandas>=2.2.0
numpy>=1.26.0
pyarrow>=15.0.0 # parquet for Amazon Reviews 2023
datasets>=2.18.0 # huggingface datasets loader
# ── Embeddings & retrieval ───────────────────────────
chromadb>=0.5.0
sentence-transformers>=3.0.0
# ── API + UI ─────────────────────────────────────────
fastapi>=0.110.0
uvicorn[standard]>=0.29.0
pydantic>=2.6.0
streamlit>=1.32.0
httpx>=0.27.0
# ── Evaluation ───────────────────────────────────────
bert-score>=0.3.13
scikit-learn>=1.4.0
tqdm>=4.66.0
# ── Dev / paper ──────────────────────────────────────
jupyter>=1.0.0
matplotlib>=3.8.0
seaborn>=0.13.0