mb-rag 1.1.64__tar.gz → 1.1.65__tar.gz

{mb_rag-1.1.64 → mb_rag-1.1.65}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mb_rag
-Version: 1.1.64
+Version: 1.1.65
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8

mb_rag-1.1.65/mb_rag/utils/llm_wrapper.py

@@ -0,0 +1,22 @@
+## simple llm wrapper to replace invoke with invoke_query/own model query
+
+__all__ = ["LLMWrapper"]
+
+class LLMWrapper:
+    """A simple wrapper for the language model to standardize the invoke method.
+    """
+
+    def __init__(self, llm):
+        self.llm = llm
+
+    def invoke_query(self, prompt: str) -> str:
+        """
+        Invoke the language model with the given prompt.
+        
+        Args:
+            prompt (str): The prompt to send to the language model.
+        
+        Returns:
+            str: The generated response.
+        """
+        return self.llm.invoke(prompt)

{mb_rag-1.1.64 → mb_rag-1.1.65}/mb_rag/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 1
-PATCH_VERSION = 64
+PATCH_VERSION = 65
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mb_rag-1.1.64 → mb_rag-1.1.65}/mb_rag.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mb_rag
-Version: 1.1.64
+Version: 1.1.65
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8

{mb_rag-1.1.64 → mb_rag-1.1.65}/mb_rag.egg-info/SOURCES.txt

@@ -20,4 +20,5 @@ mb_rag/utils/all_data_extract.py
 mb_rag/utils/bounding_box.py
 mb_rag/utils/document_extract.py
 mb_rag/utils/extra.py
+mb_rag/utils/llm_wrapper.py
 mb_rag/utils/pdf_extract.py