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

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

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

{mb_rag-1.1.61 → mb_rag-1.1.63}/mb_rag/basic.py

@@ -357,6 +357,8 @@ class ModelFactory:
         Returns:
             str: Output from the model
         """
+        if not isinstance(images, list):
+            images = [images]
         base64_images = [self._image_to_base64(image) for image in images]
         image_prompt_create = [{"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{base64_images[i]}"}} for i in range(len(images))]
         prompt_new = [{"type": "text", "text": prompt}, *image_prompt_create]

mb_rag-1.1.63/mb_rag/prompts_bank.py

@@ -0,0 +1,65 @@
+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)"
+        }
+
+    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.61 → mb_rag-1.1.63}/mb_rag/version.py

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

{mb_rag-1.1.61 → mb_rag-1.1.63}/mb_rag.egg-info/PKG-INFO

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

{mb_rag-1.1.61 → mb_rag-1.1.63}/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.61/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