ragxo 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

ragxo/client.py

@@ -51,8 +51,19 @@ class Ragxo:
         self.system_prompt = prompt
         return self
     
-    def add_model(self, model: str) -> Self:
+    def add_model(self, model: str,                              limit: int = 10,
+                        temperature: float = 0.5,
+                        max_tokens: int = 1000,
+                        top_p: float = 1.0,
+                        frequency_penalty: float = 0.0,
+                        presence_penalty: float = 0.0) -> Self:
         self.model = model
+        self.limit = limit
+        self.temperature = temperature
+        self.max_tokens = max_tokens
+        self.top_p = top_p
+        self.frequency_penalty = frequency_penalty
+        self.presence_penalty = presence_penalty
         return self
     
     def index(self, data: list[Document]) -> Self:
@@ -240,17 +251,10 @@ class Ragxo:
             raise
     
     def generate_llm_response(self, 
-                              query: str, 
-                              limit: int = 10,
-                              data: list[dict] = None, 
-                              temperature: float = 0.5,
-                              max_tokens: int = 1000,
-                              top_p: float = 1.0,
-                              frequency_penalty: float = 0.0,
-                              presence_penalty: float = 0.0,
-                              ) -> ChatCompletion:
+                              query: str,
+                              data: list[dict] = None) -> ChatCompletion:
         if data is None:
-            data = self.query(query, limit=limit)[0]
+            data = self.query(query, limit=self.limit)[0]
         
         if not self.system_prompt:
             raise ValueError("System prompt not set. Please call add_system_prompt first.")
@@ -261,11 +265,11 @@ class Ragxo:
                 {"role": "system", "content": self.system_prompt},
                 {"role": "user", "content": "query: {} data: {}".format(query, data)}
             ],
-            temperature=temperature,
-            max_tokens=max_tokens,
-            top_p=top_p,
-            frequency_penalty=frequency_penalty,
-            presence_penalty=presence_penalty,
+            temperature=self.temperature,
+            max_tokens=self.max_tokens,
+            top_p=self.top_p,
+            frequency_penalty=self.frequency_penalty,
+            presence_penalty=self.presence_penalty,
         )
         
         return response

{ragxo-0.1.6.dist-info → ragxo-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ragxo
-Version: 0.1.6
+Version: 0.1.7
 Summary: A RAG (Retrieval-Augmented Generation) toolkit with Milvus integration
 Home-page: https://github.com/yourusername/ragx
 License: MIT
@@ -20,9 +20,11 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 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)
+Requires-Dist: mocker (>=1.1.1,<2.0.0)
 Requires-Dist: openai (>=1.61.1,<2.0.0)
 Requires-Dist: pydantic (>=2.10.6,<3.0.0)
 Requires-Dist: pymilvus (>=2.5.4,<3.0.0)
+Requires-Dist: pytest-mock (>=3.14.0,<4.0.0)
 Project-URL: Repository, https://github.com/yourusername/ragx
 Description-Content-Type: text/markdown
 

ragxo-0.1.7.dist-info/RECORD

@@ -0,0 +1,5 @@
+ragxo/__init__.py,sha256=0VVe-z4XkkGQLQIG0hF0Hyf87_RgX0E4T9TRwwTkbmE,68
+ragxo/client.py,sha256=kFKLvc1s9JfhO2nTkQd1hgkQ7X0Le47dIs8_3vI916Y,9263
+ragxo-0.1.7.dist-info/METADATA,sha256=g1U7d-b7OtvGwEU01yUxxSSSa8sEVvh6Kl15uCSol_I,6225
+ragxo-0.1.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+ragxo-0.1.7.dist-info/RECORD,,

ragxo-0.1.6.dist-info/RECORD

@@ -1,5 +0,0 @@
-ragxo/__init__.py,sha256=0VVe-z4XkkGQLQIG0hF0Hyf87_RgX0E4T9TRwwTkbmE,68
-ragxo/client.py,sha256=M4777mj6oPdRIm9TvqIwXoQuJUMc7Ywczlykutd6c70,9068
-ragxo-0.1.6.dist-info/METADATA,sha256=1W4vJeY0awkXbtM0o3dyhFRHEY8VLHwkXrm55KECbg4,6141
-ragxo-0.1.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-ragxo-0.1.6.dist-info/RECORD,,