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

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

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

mb_rag-1.1.64/mb_rag/prompts_bank.py

@@ -0,0 +1,77 @@
+from langchain_core.prompts.chat import ChatPromptTemplate
+
+__all__ = ["PromptManager"]
+
+class PromptManager:
+    """
+    Central class for storing and invoking prompt templates.
+
+    Example:
+        pm = PromptManager()
+        prompt_text = pm.render_prompt("greeting")
+        print(prompt_text)
+
+        pm = PromptManager()
+        prompt_text = pm.render_prompt("todo_task", {"task": "Plan a deep learning project for image recognition"})
+        print(prompt_text)
+    """
+
+    def __init__(self):
+        self.templates = {
+            "coding_python": """You are a Python developer.
+Human: {question}
+Assistant:""",
+
+            "greeting": """You are a friendly assistant.
+Human: Hello!
+Assistant: Hi there! How can I assist you today?""",
+
+            "goodbye": """You are a friendly assistant.
+Human: Goodbye!
+Assistant: Goodbye! Have a great day!""",
+
+            "todo_task": """You are a helpful assistant.
+Human: Please create a to-do list for the following task: {task}
+Assistant:""",
+
+            "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:
+        """
+        Get a prompt template by name.
+        Args:
+            name (str): The key name of the prompt.
+        Returns:
+            str: The prompt template string.
+        """
+        template = self.templates.get(name)
+        if not template:
+            raise ValueError(f"Prompt '{name}' not found. Available prompts: {list(self.templates.keys())}")
+        return template
+
+    def render_prompt(self, name: str, context: dict = None) -> str:
+        """
+        Fill and return a rendered prompt string.
+        Args:
+            name (str): The key name of the prompt.
+            context (dict): Variables to fill into the template.
+        Returns:
+            str: The final rendered prompt text.
+        """
+        template = self.get_template(name)
+        chat_prompt = ChatPromptTemplate.from_template(template)
+        rendered = chat_prompt.invoke(context or {})
+        return rendered.to_string()

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

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

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

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

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

@@ -3,6 +3,7 @@ pyproject.toml
 setup.py
 mb_rag/__init__.py
 mb_rag/basic.py
+mb_rag/prompts_bank.py
 mb_rag/version.py
 mb_rag.egg-info/PKG-INFO
 mb_rag.egg-info/SOURCES.txt
@@ -12,7 +13,6 @@ mb_rag.egg-info/top_level.txt
 mb_rag/chatbot/__init__.py
 mb_rag/chatbot/chains.py
 mb_rag/chatbot/conversation.py
-mb_rag/chatbot/prompts.py
 mb_rag/rag/__init__.py
 mb_rag/rag/embeddings.py
 mb_rag/utils/__init__.py

mb_rag-1.1.62/mb_rag/chatbot/prompts.py

@@ -1,59 +0,0 @@
-## file for storing basic prompts template
-from langchain.prompts import ChatPromptTemplate
-
-__all__ = ["prompts", "invoke_prompt"]
-
-class prompts:
-    """
-    Class to get different prompts example for chatbot and templates
-    """
-
-    def get_code_prompts(self):
-        """
-        Get code prompts
-        Returns:
-            str: Code prompt
-        """
-        list_code_prompts = {'coding_python ': """You are a Python developer.
-                                                Human: {}"""}
-
-    def get_text_prompts(self):
-        """
-        Get text prompts
-        Returns:
-            str: Text prompt
-        """
-        list_text_prompts = {
-            'multiple_placeholders': """You are a helpful assistant.
-                                        Human: Tell me a more about {adjective1} and its relation to {adjective2}.
-                                        Assistant:"""
-        }
-
-    def get_image_prompts(self):
-        """
-        Get image prompts
-        Returns:
-            str: Image prompt
-        """
-        list_image_prompts = {'map_function': "*map(lambda x: image_url, baseframes_list)"} # for passing multiple images from a video or a list of images
-
-    def get_assistant_prompts(self):
-        """
-        Get assistant prompts
-        Returns:
-            str: Assistant prompt
-        """
-        list_assistant_prompts = {}
-
-def invoke_prompt(template: str, input_dict : dict = None):
-    """
-    Invoke a prompt
-    Args:
-        template (str): Template for the prompt
-        input_dict (dict): Input dictionary for the prompt
-    Returns:
-        str: Prompt
-    """
-    prompt_multiple = ChatPromptTemplate.from_template(template)
-    prompt = prompt_multiple.invoke(input_dict)
-    return prompt