mb-rag 1.1.29__tar.gz → 1.1.31__tar.gz

{mb_rag-1.1.29 → mb_rag-1.1.31}/PKG-INFO

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

{mb_rag-1.1.29 → mb_rag-1.1.31}/mb_rag/chatbot/basic.py

@@ -73,6 +73,8 @@ class ModelFactory:
             'hugging_face': self.create_hugging_face
         }
         
+        self.model_type = model_type
+        self.model_name = model_name
         model_data = creators.get(model_type)
         if not model_data:
             raise ValueError(f"Unsupported model type: {model_type}")
@@ -308,17 +310,16 @@ class ModelFactory:
         """
         Function to invoke the model with images
         Args:
-            model (ChatOpenAI): Chatbot model
             images (list): List of images
             prompt (str): Prompt
             pydantic_model (PydanticModel): Pydantic model
-    Returns:
+        Returns:
         str: Output from the model
     """
         base64_images = [self._image_to_base64(image) for image in images]
-        if self.model_name=='ollama':
+        if self.model_type =='ollama':
             ollama_model = self.model.bind(images=[base64_images])
-            response = ollama_model.invoke([HumanMessage(content=prompt)])
+            response = ollama_model.invoke(prompt)
             return response.content
         else:
             image_prompt_create = [{"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{base64_images[i]}"}} for i in range(len(images))]

{mb_rag-1.1.29 → mb_rag-1.1.31}/mb_rag/version.py

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

{mb_rag-1.1.29 → mb_rag-1.1.31}/mb_rag.egg-info/PKG-INFO

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