# Flip RAG SDK - Requirements
# Core dependencies for the Flip SDK

# ============================================================================
# Core Dependencies
# ============================================================================

# Configuration and utilities
pydantic>=2.0.0
tenacity>=8.2.0
tqdm>=4.65.0
python-dotenv>=1.0.0

# ============================================================================
# Vector Store
# ============================================================================

chromadb>=0.4.0

# ============================================================================
# Document Processing
# ============================================================================

# PDF processing
pypdf2>=3.0.0
pdfplumber>=0.10.0

# DOCX processing
python-docx>=1.0.0

# HTML processing
beautifulsoup4>=4.12.0
lxml>=4.9.0

# Markdown
markdown>=3.4.0

# ============================================================================
# Embeddings
# ============================================================================

# Sentence transformers (local embeddings)
sentence-transformers>=2.2.0

# Token counting
tiktoken>=0.5.0

# Numpy for embeddings
numpy>=1.24.0

# ============================================================================
# Retrieval (Advanced Features)
# ============================================================================

# BM25 for hybrid search
rank-bm25>=0.2.2

# ============================================================================
# LLM Providers
# ============================================================================

# OpenAI
openai>=1.0.0

# Anthropic
anthropic>=0.18.0

# Google Gemini
google-generativeai>=0.3.0

# HuggingFace
huggingface-hub>=0.20.0
transformers>=4.36.0

# PyTorch (for transformers and sentence-transformers)
torch>=2.0.0

# ============================================================================
# Utilities
# ============================================================================

requests>=2.31.0
aiohttp>=3.9.0

# ============================================================================
# Optional Dependencies
# ============================================================================

# Development tools (install with: pip install -r requirements-dev.txt)
# pytest>=7.4.0
# pytest-asyncio>=0.21.0
# pytest-cov>=4.1.0
# pytest-benchmark>=4.0.0
# black>=23.0.0
# ruff>=0.1.0
# mypy>=1.5.0

# Additional vector stores (install separately if needed)
# pinecone-client>=3.0.0
# qdrant-client>=1.7.0
# weaviate-client>=4.4.0
