rag-python 0.1.0__py3-none-any.whl

rag_python-0.1.0.dist-info/METADATA

@@ -0,0 +1,158 @@
+Metadata-Version: 2.2
+Name: rag-python
+Version: 0.1.0
+Summary: Production-grade RAG for Python: multi-LLM, query rewriting, reranking, guardrails, and evaluation.
+Author-email: Raghav Singla <04raghavsingla28@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/RaghavOG/rag-python
+Project-URL: Repository, https://github.com/RaghavOG/rag-python
+Project-URL: Documentation, https://github.com/RaghavOG/rag-python#readme
+Project-URL: Issues, https://github.com/RaghavOG/rag-python/issues
+Keywords: rag,llm,embeddings,chromadb,openai,rag-python,retrieval-augmented-generation
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: openai>=1.12.0
+Requires-Dist: tiktoken>=0.5.0
+Requires-Dist: chromadb>=0.4.22
+Requires-Dist: pypdf>=3.17.0
+Requires-Dist: python-docx>=1.1.0
+Requires-Dist: langdetect>=1.0.9
+Requires-Dist: regex>=2023.0.0
+Requires-Dist: python-dotenv>=1.0.0
+Requires-Dist: requests>=2.31.0
+Provides-Extra: rerank
+Requires-Dist: sentence-transformers>=2.2.0; extra == "rerank"
+Requires-Dist: torch>=2.0.0; extra == "rerank"
+Provides-Extra: anthropic
+Requires-Dist: anthropic>=0.20.0; extra == "anthropic"
+Provides-Extra: gemini
+Requires-Dist: google-genai>=0.3.0; extra == "gemini"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: twine; extra == "dev"
+Provides-Extra: all
+Requires-Dist: rag-python[anthropic,gemini,rerank]; extra == "all"
+
+# rag-python
+
+[![Python 3.10+](https://img.shields.io/badge/python-3.10+-blue.svg)](https://www.python.org/downloads/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+[![GitHub](https://img.shields.io/badge/GitHub-RaghavOG%2Frag--python-blue)](https://github.com/RaghavOG/rag-python)
+
+**rag-python** is a production-oriented Python library for **Retrieval-Augmented Generation (RAG)**.
+
+Ingest your documents, ask questions, get grounded answers — with query rewriting, multi-query retrieval, reranking, guardrails, and multi-LLM support.
+
+**Author:** [Raghav Singla](https://github.com/RaghavOG)  
+**Repository:** [github.com/RaghavOG/rag-python](https://github.com/RaghavOG/rag-python)
+
+---
+
+## Features
+
+- Document pipeline: loaders → cleaning → chunking → embeddings → ChromaDB
+- Query pipeline: rewriting → multi-query retrieval → reranking
+- Generation with guardrails (prompt injection + hallucination checks)
+- Evaluation scores + self-correction retry loop
+- **LLM providers:** OpenAI, Azure OpenAI, Anthropic, Gemini, Ollama
+
+---
+
+## Install
+
+```bash
+pip install rag-python
+# or from source
+pip install -e .
+# with reranking + extra providers
+pip install -e ".[rerank,anthropic,gemini,all]"
+```
+
+---
+
+## Quickstart
+
+```python
+from rag_python import RAG
+
+rag = RAG(
+    llm_provider="openai",
+    llm_model="gpt-4o-mini",
+    embedding_provider="openai",
+    embedding_model="text-embedding-3-small",
+)
+
+rag.ingest(["./data"], reindex=True)
+answer = rag.query("How many days of annual leave?")
+print(answer.text)
+```
+
+### CLI
+
+```bash
+export OPENAI_API_KEY=sk-...
+rag-python ingest ./data --reindex
+rag-python query "How many days of annual leave?" -v
+```
+
+---
+
+## Environment variables
+
+| Variable | Required | Description |
+|----------|----------|-------------|
+| `OPENAI_API_KEY` | For OpenAI | Default LLM + embeddings |
+| `ANTHROPIC_API_KEY` | For Claude | LLM only |
+| `GEMINI_API_KEY` | For Gemini | LLM only |
+| `AZURE_OPENAI_ENDPOINT` | For Azure | Azure OpenAI |
+| `AZURE_OPENAI_API_KEY` | For Azure | Azure OpenAI |
+| `OPENAI_API_VERSION` | Azure | Default `2023-09-01-preview` |
+| `OLLAMA_BASE_URL` | Ollama | Default `http://localhost:11434` |
+| `RAG_PYTHON_DATA_DIR` | Optional | Default `./data` |
+| `RAG_PYTHON_CHROMA_DIR` | Optional | Default `./chroma_db` |
+
+See [`.env.example`](.env.example) for all tuning options.
+
+---
+
+## Project structure
+
+```text
+.
+├── src/rag_python/      # Installable package (PyPI: rag-python)
+│   ├── client.py        # High-level RAG API
+│   ├── rag_pipeline.py  # Full pipeline
+│   └── providers/       # OpenAI, Azure, Anthropic, Gemini, Ollama
+├── tests/
+├── examples/
+├── docs/
+├── data/                # Sample documents
+├── pyproject.toml
+└── main.py              # Local dev CLI wrapper
+```
+
+---
+
+## Docs
+
+- [Usage](docs/USAGE.md)
+- [Providers](docs/PROVIDERS.md)
+- [Changelog](CHANGELOG.md)
+
+---
+
+## License
+
+MIT © [Raghav Singla](https://github.com/RaghavOG)

rag_python-0.1.0.dist-info/RECORD

@@ -0,0 +1,31 @@
+rag_python/__init__.py,sha256=0mdBYu9JmPzQBOX947S5tdk5KuNU3JNO-y5GsDZsFlY,834
+rag_python/chunking.py,sha256=P1dbZ8ZY7487MxrWe2cypCiKhzIJ8zBPCTVz20vt8fo,6204
+rag_python/cleaning.py,sha256=fSux4T0pg7Xe_8NUP2pgzuForyRk1i2VPYIXSzRajzs,3193
+rag_python/cli.py,sha256=UlCnI6Ah7pmyZwCEP1c-gt5XDOhFzLOvnsk2vKxvp-A,3126
+rag_python/client.py,sha256=MhWAm92Ic2FQ1DTej4EhAlT9UoPN-GjxA0xrHIvwNA8,7656
+rag_python/config.py,sha256=Zw8TjQFKRvOUHpIb7kjEb7DtPFoYPzdQyOPzSXTqDcc,1389
+rag_python/document_loaders.py,sha256=izguVJjPq8v4hDWC8wGP2-LwiYUJbVe-DOsIX6n9J9E,2429
+rag_python/evaluation.py,sha256=gTiXMaAtTUIsV6Ffhywz829BhfR8YhfJFkYZYrD9WYI,3561
+rag_python/generation.py,sha256=t6aSct2vZELIf20JDwRVt8UTwPnTXx0bU3TKoliiwVg,1108
+rag_python/guardrails.py,sha256=hJLXvpPNI9o8emyipSy5PpePofGzktlDLyMAXfAxUXs,2520
+rag_python/options.py,sha256=QvangjsYbct204_p_avraAuw_Ry2mcjhxby2Sx96dE0,1858
+rag_python/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+rag_python/query_rewriting.py,sha256=og_XWai2-08C7W67mFndA3k-aTxMdqGnu70qHi1Ohgc,2293
+rag_python/rag_pipeline.py,sha256=R8roEc-OSFPqrjZmvJXp7IhYaOdqd6kZ63otN1PUKTI,8747
+rag_python/reranker.py,sha256=8RxCPfgp80c-KSKojllGzbpZ7iSku-i7VLgPHa1a3rk,2181
+rag_python/retrieval.py,sha256=A9ZMIkrifyDF1rhtgNt_uR09o1_hCkywIuODjvLaY6Q,2261
+rag_python/vector_store.py,sha256=16I9g7Q8bMjwpxhTNv0nCq2WXQwftjZdSaxFNteVsH4,2909
+rag_python/providers/__init__.py,sha256=SjhMvYoA30EY5VUYVXhEGwcmQnIU2tUomcNE0_0NFho,215
+rag_python/providers/anthropic_provider.py,sha256=dSiCdM4F90jI9w7z_wS10XuVsX-pR733-cAgJHtVV2Q,1493
+rag_python/providers/azure_openai_provider.py,sha256=8SbI7rDzQgvC4ZXP89Q8kjfqeWuBfX1KKgExGLFkmx0,1940
+rag_python/providers/base.py,sha256=M9DYowQvNvRuATaM6944CWovK0awJ0buBmbnQfroJos,593
+rag_python/providers/factory.py,sha256=uD0Hqrzyj9R88vNVN-95Fn2gj28hUIVNXuKe0G1R5wo,2522
+rag_python/providers/gemini_provider.py,sha256=OZzs1YJQSZituoxS5Gk8yv3jYNIFY1SVovWUu7lz5Z4,1842
+rag_python/providers/ollama_provider.py,sha256=DDhDriB6-Ob0r2-M-P3SvIFG37ruDAErtU7LWDK8xh0,1958
+rag_python/providers/openai_provider.py,sha256=oR7rCCaxCtirAVetJrR4oC3UrWySuqLc9kbosydoQAQ,1585
+rag_python-0.1.0.dist-info/LICENSE,sha256=PZ61Z6ve0hBHgztaC1rPgnxQTRXRkeHKASlnKkX2pvc,1079
+rag_python-0.1.0.dist-info/METADATA,sha256=R0dvEMcDKMgUDxImAEwmBeHMYcmR46NjPl31lhoBJw4,5199
+rag_python-0.1.0.dist-info/WHEEL,sha256=beeZ86-EfXScwlR_HKu4SllMC9wUEj_8Z_4FJ3egI2w,91
+rag_python-0.1.0.dist-info/entry_points.txt,sha256=558Rd4GWV_6mIyqdRSVNE4ZZi0-KdblTZhcMbIn3ryY,51
+rag_python-0.1.0.dist-info/top_level.txt,sha256=SrgudPwkJWfJ3gUn2n-dhrt9vN2XbQcaZ3wLQZed4Z4,11
+rag_python-0.1.0.dist-info/RECORD,,

rag_python-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (76.1.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

rag_python-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+rag-python = rag_python.cli:main

rag_python-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+rag_python