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

mb_rag/chatbot/basic.py

@@ -66,7 +66,9 @@ class ModelFactory:
             'anthropic': self.create_anthropic,
             'google': self.create_google,
             'ollama': self.create_ollama,
-            'groq': self.create_groq
+            'groq': self.create_groq,
+            'deepseek': self.create_deepseek,
+            'qwen' : self.create_qwen,
         }
         
         model_data = creators.get(model_type)
@@ -139,12 +141,12 @@ class ModelFactory:
         Returns:
             Ollama: Chatbot model
         """
-        if not check_package("langchain_community"):
+        if not check_package("langchain-ollama"):
             raise ImportError("Langchain Community package not found. Please install it using: pip install langchain-community")
         
-        from langchain_community.llms import Ollama
+        from langchain_ollama import ChatOllama
         kwargs["model"] = model_name
-        return Ollama(**kwargs)
+        return ChatOllama(**kwargs)
 
     @classmethod
     def create_groq(cls, model_name: str = "llama-3.3-70b-versatile", **kwargs) -> Any:
@@ -163,6 +165,40 @@ class ModelFactory:
         kwargs["model"] = model_name
         return ChatGroq(**kwargs)
 
+    @classmethod
+    def create_deepseek(cls, model_name: str = "deepseek-chat", **kwargs) -> Any:
+        """
+        Create Deepseek chatbot model
+        Args:
+            model_name (str): Name of the model
+            **kwargs: Additional arguments
+        Returns:     
+            ChatDeepseek: Chatbot model
+        """
+        if not check_package("langchain_deepseek"):
+            raise ImportError("Langchain Deepseek package not found. Please install it using: pip install langchain-deepseek")
+
+        from langchain_deepseek import ChatDeepSeek
+        kwargs["model"] = model_name
+        return ChatDeepSeek(**kwargs)
+
+    @classmethod
+    def create_qwen(cls, model_name: str = "qwen", **kwargs) -> Any:
+        """
+        Create Qwen chatbot model
+        Args:
+            model_name (str): Name of the model
+            **kwargs: Additional arguments
+        Returns:
+            ChatQwen: Chatbot model
+        """
+        if not check_package("langchain_community"):
+            raise ImportError("Langchain Qwen package not found. Please install it using: pip install langchain_community")
+
+        from langchain_community.chat_models.tongyi import ChatTongyi
+        kwargs["model"] = model_name
+        return ChatTongyi(streaming=True,**kwargs)
+
     def _reset_model(self):
         """Reset the model"""
         self.model = self.model.reset()

mb_rag/version.py

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

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

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

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

@@ -1,7 +1,7 @@
 mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/version.py,sha256=OJ5C8dPfTjHQppzvBtr3tex8vKiblUMZ8i5wkIS8_U0,206
+mb_rag/version.py,sha256=HzmwFCQOy09joW49HSdBMmEK0R29-zIwhOZLIVAahFo,207
 mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/chatbot/basic.py,sha256=19spb9hBOhO4UYUZ3q6bzBOHnCrq6sqPFS0EhGPIugw,15448
+mb_rag/chatbot/basic.py,sha256=CQj9lXe3VNOADWQANWhR8aUmAz9efe3gwZfMYLVm1_0,16806
 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.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,,
+mb_rag-1.1.11.dist-info/METADATA,sha256=LM1DLdW_H0bhI9kvwiOjJyUxlKhL_EQ6WoraLL-skT4,234
+mb_rag-1.1.11.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+mb_rag-1.1.11.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
+mb_rag-1.1.11.dist-info/RECORD,,