mb-rag 1.1.41__py3-none-any.whl → 1.1.43__py3-none-any.whl

mb_rag/chatbot/basic.py

@@ -344,7 +344,7 @@ class ModelFactory:
         """
         print("Model Name: ", self.model)
         print("Model Temperature: ", self.model.temperature)
-        print("Model Max Tokens: ", self.model.max_tokens)
+        print("Model Max Tokens: ", self.model.max_output_tokens)
         print("Model Top P: ", self.model.top_p)
         print("Model Top K: ", self.model.top_k)
         print("Model Input Schema:",self.model.input_schema)

mb_rag/rag/embeddings.py

@@ -501,7 +501,7 @@ class embedding_generator:
 
     def add_data(self, embeddings_folder_path: str, data: List[str],
                 text_splitter_type: str = 'recursive_character',
-                chunk_size: int = 1000, chunk_overlap: int = 5):
+                chunk_size: int = 1000, chunk_overlap: int = 5, collection_name: str = 'test'):
         """
         Add data to existing embeddings.
 
@@ -511,6 +511,7 @@ class embedding_generator:
             text_splitter_type (str): Type of text splitter
             chunk_size (int): Size of text chunks
             chunk_overlap (int): Overlap between chunks
+            collection_name (str): Name of the collection. Default: 'test'
         """
         if self.vector_store_type == 'chroma':
             db = self.load_embeddings(embeddings_folder_path, collection_name)

mb_rag/version.py

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

{mb_rag-1.1.41.dist-info → mb_rag-1.1.43.dist-info}/METADATA

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

{mb_rag-1.1.41.dist-info → mb_rag-1.1.43.dist-info}/RECORD

@@ -1,17 +1,17 @@
 mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/version.py,sha256=BFSZUhT1jVjQwU570kNAHMJrCSLstBwSZiV2VECV7d8,207
+mb_rag/version.py,sha256=F0q66lXUV7IiL9pvn0fGJZImdHtQZkDF6FEwisuxspA,207
 mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/chatbot/basic.py,sha256=ZlErs5mfzxGwauJqrujPQmK_DYYgfAUpNOBqlYLVQVY,23818
+mb_rag/chatbot/basic.py,sha256=jsoPp0b-JhYRYW15WRv73h4OoZzwT2LTtptEOHnSQJo,23825
 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
-mb_rag/rag/embeddings.py,sha256=MhxG0n7Q0XwPGYIRjCpbSZTlZLyj6Shxo64n347BEWA,27497
+mb_rag/rag/embeddings.py,sha256=H6dDkZi4Ez9NZQrzaYzMlQ66ILLZiOIM-k1LNFfcjSM,27603
 mb_rag/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mb_rag/utils/bounding_box.py,sha256=G0hdDam8QmYtD9lfwMeDHGm-TTo6KZg-yK5ESFL9zaM,8366
 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.41.dist-info/METADATA,sha256=ZuHbGDp3YGKALWvaJZkbZR3KamJYQRcPySxTtJhpjdU,234
-mb_rag-1.1.41.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mb_rag-1.1.41.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
-mb_rag-1.1.41.dist-info/RECORD,,
+mb_rag-1.1.43.dist-info/METADATA,sha256=xzXHksEUFKJAEm48bYtz1U0uu7evE4lrnoYUX3zVje8,234
+mb_rag-1.1.43.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+mb_rag-1.1.43.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
+mb_rag-1.1.43.dist-info/RECORD,,