ragxo 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

ragxo/client.py

@@ -11,6 +11,7 @@ import tempfile
 from botocore.exceptions import ClientError
 import openai
 from openai import ChatCompletion
+from ragxo.utils import with_loading
 
 logger = logging.getLogger(__name__)
 
@@ -140,6 +141,7 @@ class Ragxo:
         self.presence_penalty = presence_penalty
         return self
     
+    @with_loading("Indexing documents")
     def index(self, data: list[Document]) -> Self:
         """
         Index documents into the vector database.
@@ -210,6 +212,7 @@ class Ragxo:
             output_fields=output_fields
         )
 
+    @with_loading("Exporting Ragxo instance")
     def export(self, destination: str, s3_bucket: str = None) -> Self:
         """
         Export the Ragx instance to either local filesystem or S3.

ragxo/utils.py

@@ -0,0 +1,21 @@
+import functools
+from alive_progress import alive_bar
+
+
+def with_loading(title: str):
+    """
+    Decorator to add loading animation to methods.
+    
+    Args:
+        title (str): Title to display during loading
+    """
+
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(self, *args, **kwargs):
+            with alive_bar(title=title, bar=None, stats=False) as bar:
+                result = func(self, *args, **kwargs)
+                bar()
+            return result
+        return wrapper
+    return decorator

{ragxo-0.1.10.dist-info → ragxo-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragxo
-Version: 0.1.10
+Version: 0.1.11
 Summary: A RAG (Retrieval-Augmented Generation) toolkit with Milvus integration
 Home-page: https://github.com/yourusername/ragx
 License: MIT
@@ -17,6 +17,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Dist: alive-progress (>=3.1.1,<4.0.0)
 Requires-Dist: boto3 (>=1.36.14,<2.0.0)
 Requires-Dist: dill (>=0.3.9,<0.4.0)
 Requires-Dist: milvus (>=2.3.9,<3.0.0)
@@ -30,7 +31,7 @@ Description-Content-Type: text/markdown
 
 # RagXO
 
-Export, version and reuse your RAG pipeline everywhere 🚀
+Export, version and reuse your E2E RAG pipeline everywhere 🚀
 
 [![PyPI version](https://badge.fury.io/py/ragxo.svg)](https://badge.fury.io/py/ragxo)
 [![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
@@ -57,6 +58,10 @@ pip install ragxo
 
 ### Build a RAG pipeline
 
+```bash
+export OPENAI_API_KEY=<openai_key> 
+```
+
 ```python
 from ragxo import Ragxo, Document
 
@@ -232,3 +237,4 @@ This project is licensed under the MIT License - see the [LICENSE](LICENSE) file
 ## Contributing 🤝
 
 Contributions are welcome! Please feel free to submit a Pull Request.
+

ragxo-0.1.11.dist-info/RECORD

@@ -0,0 +1,6 @@
+ragxo/__init__.py,sha256=0VVe-z4XkkGQLQIG0hF0Hyf87_RgX0E4T9TRwwTkbmE,68
+ragxo/client.py,sha256=6VE9h9XoEaS01irBp70brrWev8tNagtYnHOoyvjBVmo,12914
+ragxo/utils.py,sha256=yy5_ejmxU75mRhfE_XGtKOvKtHo8AoV1QFQlwMDEiFw,537
+ragxo-0.1.11.dist-info/METADATA,sha256=t3vwZl6LHtRgnrGqI8jMRrWGUy2c9tQlkxG45UDjjf4,6472
+ragxo-0.1.11.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+ragxo-0.1.11.dist-info/RECORD,,

ragxo-0.1.10.dist-info/RECORD

@@ -1,5 +0,0 @@
-ragxo/__init__.py,sha256=0VVe-z4XkkGQLQIG0hF0Hyf87_RgX0E4T9TRwwTkbmE,68
-ragxo/client.py,sha256=rmV01TZ3F8McLavVOokjLSUv78fdCA80oDRVWSw-38M,12791
-ragxo-0.1.10.dist-info/METADATA,sha256=uk-eth9MdnPkG2WeZU7WGr7cVa6HuPtTZFjTkFVSJ_A,6371
-ragxo-0.1.10.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-ragxo-0.1.10.dist-info/RECORD,,