mb-rag 1.1.1__tar.gz → 1.1.3__tar.gz

{mb_rag-1.1.1 → mb_rag-1.1.3}/PKG-INFO

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

{mb_rag-1.1.1 → mb_rag-1.1.3}/README.md

@@ -37,12 +37,12 @@ pip install mb_rag
 from mb_rag.chatbot.basic import ModelFactory, ConversationModel
 
 # 1. Simple Query with ModelFactory
-model = ModelFactory(model_type="openai", model_name="gpt-4")
+model = ModelFactory(model_type="openai", model_name="gpt-4o")
 response = model.invoke_query("What is artificial intelligence?")
 print(response)
 
 # 2. Image Analysis
-model = ModelFactory(model_type="openai", model_name="gpt-4-vision-preview")
+model = ModelFactory(model_type="openai", model_name="gpt-4o")
 response = model.invoke_query(
     "What's in these images?",
     images=["image1.jpg", "image2.jpg"]
@@ -51,7 +51,7 @@ print(response)
 
 # 3. Conversation with Context
 conversation = ConversationModel(
-    model_name="gpt-4",
+    model_name="gpt-4o",
     model_type="openai"
 )
 
@@ -147,7 +147,7 @@ em_gen.add_data(
 
 # Web scraping and embedding
 db = em_gen.firecrawl_web(
-    website="https://example.com",
+    website="https://github.com",
     mode="scrape",
     file_to_save='./web_embeddings'
 )

{mb_rag-1.1.1 → mb_rag-1.1.3}/mb_rag/chatbot/basic.py

@@ -54,7 +54,7 @@ class ModelFactory:
         """
         Factory method to create any type of model
         Args:
-            model_type (str): Type of model to create
+            model_type (str): Type of model to create. Default is OpenAI. Options are openai, anthropic, google, ollama
             model_name (str): Name of the model
             **kwargs: Additional arguments
         Returns:
@@ -120,8 +120,8 @@ class ModelFactory:
         Returns:
             ChatGoogleGenerativeAI: Chatbot model
         """
-        if not check_package("google.generativeai"):
-            raise ImportError("Google Generative AI package not found. Please install it using: pip install google-generativeai langchain-google-genai")
+        if not check_package("langchain_google_genai"):
+            raise ImportError("langchain_google_genai package not found. Please install it using: pip install google-generativeai langchain-google-genai")
         
         from langchain_google_genai import ChatGoogleGenerativeAI
         kwargs["model"] = model_name

{mb_rag-1.1.1 → mb_rag-1.1.3}/mb_rag/version.py

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

{mb_rag-1.1.1 → mb_rag-1.1.3}/mb_rag.egg-info/PKG-INFO

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