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

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

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

{mb_rag-1.1.63 → mb_rag-1.1.65}/mb_rag/prompts_bank.py

@@ -34,7 +34,19 @@ Assistant: Goodbye! Have a great day!""",
 Human: Please create a to-do list for the following task: {task}
 Assistant:""",
 
-            "map_function": "*map(lambda x: image_url, baseframes_list)"
+            "map_function": "*map(lambda x: image_url, baseframes_list)",
+            
+            "SQL_AGENT_SYS_PROMPT": """You are an expert SQL agent. Your task is to generate and execute SQL queries based on user requests.
+RULES:
+- THINK step by step before answering.
+- Use the provided database schema to inform your queries.
+- When you need to retrieve data, generate a SQL query and execute it using the provided tools.
+- Read-only mode: Do not attempt to modify the database.
+- NO INSERT/UPDATE/DELETE/ALTER/DROP/CREATE/REPLACE/TRUNCATE statements allowed.
+- LIMIT your results to 10 rows. Unless specified otherwise.
+- If you encounter an error while executing a query, analyze the error message and adjust your query accordingly.
+- Prefer using explicit column names instead of SELECT * for better performance.    
+- Always ensure your SQL syntax is correct. """
         }
 
     def get_template(self, name: str) -> str:

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.63 → mb_rag-1.1.65}/mb_rag/version.py

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

{mb_rag-1.1.63 → 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.63
+Version: 1.1.65
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8

{mb_rag-1.1.63 → 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