mb-rag 1.1.7__py3-none-any.whl → 1.1.9__py3-none-any.whl

mb_rag/chatbot/basic.py

@@ -2,6 +2,7 @@ import os
 from dotenv import load_dotenv
 from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
 from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+from langchain.document_loaders import TextLoader
 from IPython.display import display, HTML
 from typing import Optional, List, Dict, Any, Union
 from mb_rag.utils.extra import check_package
@@ -155,30 +156,40 @@ class ModelFactory:
         Returns:
             ChatGroq: Chatbot model
         """
-        if not check_package("langchain-groq"):
+        if not check_package("langchain_groq"):
             raise ImportError("Langchain Groq package not found. Please install it using: pip install langchain-groq")
 
         from langchain_groq import ChatGroq
         kwargs["model"] = model_name
         return ChatGroq(**kwargs)
 
-    def invoke_query(self,query: str,get_content_only: bool = True,images: list = None,pydantic_model = None) -> str:
+    def _reset_model(self):
+        """Reset the model"""
+        self.model = self.model.reset()
+
+    def invoke_query(self,query: str,file_path: str = None,get_content_only: bool = True,images: list = None,pydantic_model = None) -> str:
         """
         Invoke the model
         Args:
             query (str): Query to send to the model
+            file_path (str): Path to text file to load. Default is None
             get_content_only (bool): Whether to return only content
             images (list): List of images to send to the model
             pydantic_model: Pydantic model for structured output
         Returns:
             str: Response from the model
         """
-        
+        if file_path:
+            loader = TextLoader(file_path)
+            document = loader.load()
+            query = document.content
+
         if pydantic_model is not None:
-            try:
-                self.model = self.model.with_structured_output(pydantic_model)
-            except Exception as e:
-                raise ValueError(f"Error with pydantic_model: {e}")
+            if hasattr(self.model, 'with_structured_output'):
+                try:
+                    self.model = self.model.with_structured_output(pydantic_model)
+                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)
         else:

mb_rag/version.py

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

{mb_rag-1.1.7.dist-info → mb_rag-1.1.9.dist-info}/METADATA

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

{mb_rag-1.1.7.dist-info → mb_rag-1.1.9.dist-info}/RECORD

@@ -1,7 +1,7 @@
 mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/version.py,sha256=Cpayvd-NH5r-VWXnUrdxc3Wl5e2683dhC75IB5-U87A,206
+mb_rag/version.py,sha256=OJ5C8dPfTjHQppzvBtr3tex8vKiblUMZ8i5wkIS8_U0,206
 mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/chatbot/basic.py,sha256=2TNPI1gAOGvhw1jgjGhy4M972tHVwNTs_cZR65PO8HM,14996
+mb_rag/chatbot/basic.py,sha256=19spb9hBOhO4UYUZ3q6bzBOHnCrq6sqPFS0EhGPIugw,15448
 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
@@ -9,7 +9,7 @@ mb_rag/rag/embeddings.py,sha256=KjBdekFDb5M3dRMco4r3dDMXMsG5dxdzKImuVIipsd0,2709
 mb_rag/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mb_rag/utils/bounding_box.py,sha256=G0hdDam8QmYtD9lfwMeDHGm-TTo6KZg-yK5ESFL9zaM,8366
 mb_rag/utils/extra.py,sha256=spbFrGgdruNyYQ5PzgvpSIa6Nm0rn9bb4qc8W9g582o,2492
-mb_rag-1.1.7.dist-info/METADATA,sha256=q-CMqSDz0STKKK-EftC6W-HsjqO3kBgDhdW5F2mwo6s,233
-mb_rag-1.1.7.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-mb_rag-1.1.7.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
-mb_rag-1.1.7.dist-info/RECORD,,
+mb_rag-1.1.9.dist-info/METADATA,sha256=XPqRcdq60lulFNTaphL63ke0xn8pAEUhwf_OBStADT4,233
+mb_rag-1.1.9.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mb_rag-1.1.9.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
+mb_rag-1.1.9.dist-info/RECORD,,