ragxo 0.1.8__tar.gz → 0.1.10__tar.gz

{ragxo-0.1.8 → ragxo-0.1.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragxo
-Version: 0.1.8
+Version: 0.1.10
 Summary: A RAG (Retrieval-Augmented Generation) toolkit with Milvus integration
 Home-page: https://github.com/yourusername/ragx
 License: MIT
@@ -60,17 +60,19 @@ pip install ragxo
 ```python
 from ragxo import Ragxo, Document
 
-def preprocess_text_lower(text: str) -> str:
-    return text.lower()
+
+
+ragxo_client = Ragxo(dimension=1536)
 
 def preprocess_text_remove_special_chars(text: str) -> str:
     return re.sub(r'[^a-zA-Z0-9\s]', '', text)
 
+def preprocess_text_lower(text: str) -> str:
+    return text.lower()
+
 def get_embeddings(text: str) -> list[float]:
     return openai.embeddings.create(input=text, model="text-embedding-ada-002").data[0].embedding
 
-ragxo_client = Ragxo(dimension=768)
-
 ragxo_client.add_preprocess(preprocess_text_lower)
 ragxo_client.add_preprocess(preprocess_text_remove_special_chars)
 ragxo_client.add_embedding_fn(get_embeddings)
@@ -78,6 +80,7 @@ ragxo_client.add_embedding_fn(get_embeddings)
 ragxo_client.add_system_prompt("You are a helpful assistant that can answer questions about the data provided.")
 ragxo_client.add_model(
     "gpt-4o-mini",
+    limit=10,
     temperature=0.5,
     max_tokens=1000,
     top_p=1.0,
@@ -93,6 +96,9 @@ ragxo_client.index([
 
 ragxo_client.export("my_rag_v1.0.0")
 
+# or export to s3
+ragxo_client.export("my_rag_v1.0.0", s3_bucket="my_bucket")
+
 ```
 
 
@@ -106,6 +112,7 @@ vector_search_results = loaded_ragxo_client.query("What is the capital of France
 llm_response = loaded_ragxo_client.generate_llm_response(
     "What is the capital of France?")
 
+print(llm_response.choices[0].message.content)
 ```
 
 

{ragxo-0.1.8 → ragxo-0.1.10}/README.md

@@ -30,17 +30,19 @@ pip install ragxo
 ```python
 from ragxo import Ragxo, Document
 
-def preprocess_text_lower(text: str) -> str:
-    return text.lower()
+
+
+ragxo_client = Ragxo(dimension=1536)
 
 def preprocess_text_remove_special_chars(text: str) -> str:
     return re.sub(r'[^a-zA-Z0-9\s]', '', text)
 
+def preprocess_text_lower(text: str) -> str:
+    return text.lower()
+
 def get_embeddings(text: str) -> list[float]:
     return openai.embeddings.create(input=text, model="text-embedding-ada-002").data[0].embedding
 
-ragxo_client = Ragxo(dimension=768)
-
 ragxo_client.add_preprocess(preprocess_text_lower)
 ragxo_client.add_preprocess(preprocess_text_remove_special_chars)
 ragxo_client.add_embedding_fn(get_embeddings)
@@ -48,6 +50,7 @@ ragxo_client.add_embedding_fn(get_embeddings)
 ragxo_client.add_system_prompt("You are a helpful assistant that can answer questions about the data provided.")
 ragxo_client.add_model(
     "gpt-4o-mini",
+    limit=10,
     temperature=0.5,
     max_tokens=1000,
     top_p=1.0,
@@ -63,6 +66,9 @@ ragxo_client.index([
 
 ragxo_client.export("my_rag_v1.0.0")
 
+# or export to s3
+ragxo_client.export("my_rag_v1.0.0", s3_bucket="my_bucket")
+
 ```
 
 
@@ -76,6 +82,7 @@ vector_search_results = loaded_ragxo_client.query("What is the capital of France
 llm_response = loaded_ragxo_client.generate_llm_response(
     "What is the capital of France?")
 
+print(llm_response.choices[0].message.content)
 ```
 
 

{ragxo-0.1.8 → ragxo-0.1.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ragxo"
-version = "0.1.8"
+version = "0.1.10"
 description = "A RAG (Retrieval-Augmented Generation) toolkit with Milvus integration"
 authors = ["Mohamed Sadek <mohamedfawzydes@gmail.com>"]
 readme = "README.md"

{ragxo-0.1.8 → ragxo-0.1.10}/ragxo/client.py

@@ -109,7 +109,8 @@ class Ragxo:
         self.system_prompt = prompt
         return self
     
-    def add_model(self, model: str,                              limit: int = 10,
+    def add_model(self, model: str,                              
+                        limit: int = 10,
                         temperature: float = 0.5,
                         max_tokens: int = 1000,
                         top_p: float = 1.0,