mb-rag 1.1.39__py3-none-any.whl → 1.1.40__py3-none-any.whl

mb_rag/chatbot/basic.py

@@ -292,7 +292,7 @@ class ModelFactory:
                 except Exception as e:
                     raise ValueError(f"Error with pydantic_model: {e}")
         if images:
-            res = self._model_invoke_images(images=images,prompt=query,pydantic_model=pydantic_model)
+            res = self._model_invoke_images(images=images,prompt=query,pydantic_model=pydantic_model,get_content_only=get_content_only)
         else:
             res = self.model.invoke(query)
             if get_content_only:
@@ -306,7 +306,7 @@ class ModelFactory:
         with open(image, "rb") as f:
             return base64.b64encode(f.read()).decode('utf-8')
 
-    def _model_invoke_images(self,images: list, prompt: str,pydantic_model = None):
+    def _model_invoke_images(self,images: list, prompt: str,pydantic_model = None,get_content_only: bool = True) -> str:
         """
         Function to invoke the model with images
         Args:
@@ -328,11 +328,27 @@ class ModelFactory:
                 print("Continuing without structured output")
         message= HumanMessage(content=prompt_new,)
         response = self.model.invoke([message])
-        try:
-            return response.content
-        except Exception:
+        
+        if get_content_only:
+            try:
+                return response.content
+            except Exception:
+                print("Failed to get content from response. Returning response object")
+                return response
+        else:
             return response
 
+    def _get_llm_metadata(self):
+        """
+        Returns Basic metadata about the LLM
+        """
+        print("Model Name: ", self.model)
+        print("Model Temperature: ", self.temperature)
+        print("Model Max Tokens: ", self.max_output_tokens)
+        print("Model Top P: ", self.top_p)
+        print("Model Top K: ", self.top_k)
+        print("Model Input Schema:",self.input_schema)
+
 class ConversationModel:
     """
     A class to handle conversation with AI models

mb_rag/version.py

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

{mb_rag-1.1.39.dist-info → mb_rag-1.1.40.dist-info}/METADATA

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

{mb_rag-1.1.39.dist-info → mb_rag-1.1.40.dist-info}/RECORD

@@ -1,7 +1,7 @@
 mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/version.py,sha256=HcQVlfDNzK6DIr0RRqDVdD0Tb_es_b1S3Ta7qd1VAWE,207
+mb_rag/version.py,sha256=sXmcxrju4rVVuJ2zDmDuvU1l0HYAjyAazfGTs-wkYVU,207
 mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/chatbot/basic.py,sha256=wZ9rZvLtNEZPhaAFkOv0yaPCYiki9goNr77Pyqwfn5g,23139
+mb_rag/chatbot/basic.py,sha256=1Tv0ioI_iDbKoigTEoUEnfmyPlyfPR5kxK_b1hguKQQ,23795
 mb_rag/chatbot/chains.py,sha256=vDbLX5R29sWN1pcFqJ5fyxJEgMCM81JAikunAEvMC9A,7223
 mb_rag/chatbot/prompts.py,sha256=n1PyiLbU-5fkslRv6aVOzt0dDlwya_cEdQ7kRnRhMuY,1749
 mb_rag/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -11,7 +11,7 @@ mb_rag/utils/bounding_box.py,sha256=G0hdDam8QmYtD9lfwMeDHGm-TTo6KZg-yK5ESFL9zaM,
 mb_rag/utils/document_extract.py,sha256=vZiFB1RYm1BIEaNA0MveJ5Zp-KEi0ngKjW8xEdtPqXA,12558
 mb_rag/utils/extra.py,sha256=spbFrGgdruNyYQ5PzgvpSIa6Nm0rn9bb4qc8W9g582o,2492
 mb_rag/utils/pdf_extract.py,sha256=cVeMyhnAU4XZxjIZHKMYhrktTjUNOjhx2r_LZKReOZE,15598
-mb_rag-1.1.39.dist-info/METADATA,sha256=MlnqdDCIR0P88B8EiNDnjQQCrI4BMT-5pm1Dlu9B3Og,234
-mb_rag-1.1.39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mb_rag-1.1.39.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
-mb_rag-1.1.39.dist-info/RECORD,,
+mb_rag-1.1.40.dist-info/METADATA,sha256=fsTbt8ViXAEuh-H8JTyCJxoNIwKYlgmFL_CXnN5qwHs,234
+mb_rag-1.1.40.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mb_rag-1.1.40.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
+mb_rag-1.1.40.dist-info/RECORD,,