mb-rag 1.1.41__tar.gz → 1.1.43__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.

This version of mb-rag might be problematic. Click here for more details.

Files changed (23) hide show
  1. {mb_rag-1.1.41 → mb_rag-1.1.43}/PKG-INFO +1 -1
  2. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/chatbot/basic.py +1 -1
  3. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/rag/embeddings.py +2 -1
  4. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/version.py +1 -1
  5. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag.egg-info/PKG-INFO +1 -1
  6. {mb_rag-1.1.41 → mb_rag-1.1.43}/README.md +0 -0
  7. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/__init__.py +0 -0
  8. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/chatbot/__init__.py +0 -0
  9. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/chatbot/chains.py +0 -0
  10. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/chatbot/prompts.py +0 -0
  11. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/rag/__init__.py +0 -0
  12. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/utils/__init__.py +0 -0
  13. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/utils/bounding_box.py +0 -0
  14. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/utils/document_extract.py +0 -0
  15. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/utils/extra.py +0 -0
  16. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/utils/pdf_extract.py +0 -0
  17. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag.egg-info/SOURCES.txt +0 -0
  18. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag.egg-info/dependency_links.txt +0 -0
  19. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag.egg-info/requires.txt +0 -0
  20. {mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag.egg-info/top_level.txt +0 -0
  21. {mb_rag-1.1.41 → mb_rag-1.1.43}/pyproject.toml +0 -0
  22. {mb_rag-1.1.41 → mb_rag-1.1.43}/setup.cfg +0 -0
  23. {mb_rag-1.1.41 → mb_rag-1.1.43}/setup.py +0 -0
{mb_rag-1.1.41 → mb_rag-1.1.43}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mb_rag
3
- Version: 1.1.41
3
+ Version: 1.1.43
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/chatbot/basic.py RENAMED
@@ -344,7 +344,7 @@ class ModelFactory:
344
344
  """
345
345
  print("Model Name: ", self.model)
346
346
  print("Model Temperature: ", self.model.temperature)
347
- print("Model Max Tokens: ", self.model.max_tokens)
347
+ print("Model Max Tokens: ", self.model.max_output_tokens)
348
348
  print("Model Top P: ", self.model.top_p)
349
349
  print("Model Top K: ", self.model.top_k)
350
350
  print("Model Input Schema:",self.model.input_schema)
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/rag/embeddings.py RENAMED
@@ -501,7 +501,7 @@ class embedding_generator:
501
501
 
502
502
  def add_data(self, embeddings_folder_path: str, data: List[str],
503
503
  text_splitter_type: str = 'recursive_character',
504
- chunk_size: int = 1000, chunk_overlap: int = 5):
504
+ chunk_size: int = 1000, chunk_overlap: int = 5, collection_name: str = 'test'):
505
505
  """
506
506
  Add data to existing embeddings.
507
507
 
@@ -511,6 +511,7 @@ class embedding_generator:
511
511
  text_splitter_type (str): Type of text splitter
512
512
  chunk_size (int): Size of text chunks
513
513
  chunk_overlap (int): Overlap between chunks
514
+ collection_name (str): Name of the collection. Default: 'test'
514
515
  """
515
516
  if self.vector_store_type == 'chroma':
516
517
  db = self.load_embeddings(embeddings_folder_path, collection_name)
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/version.py RENAMED
@@ -1,5 +1,5 @@
1
1
  MAJOR_VERSION = 1
2
2
  MINOR_VERSION = 1
3
- PATCH_VERSION = 41
3
+ PATCH_VERSION = 43
4
4
  version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
5
5
  __all__ = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag.egg-info/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mb_rag
3
- Version: 1.1.41
3
+ Version: 1.1.43
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.41 → mb_rag-1.1.43}/README.md RENAMED
File without changes
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/__init__.py RENAMED
File without changes
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/rag/__init__.py RENAMED
File without changes
{mb_rag-1.1.41 → mb_rag-1.1.43}/mb_rag/utils/extra.py RENAMED
File without changes
{mb_rag-1.1.41 → mb_rag-1.1.43}/pyproject.toml RENAMED
File without changes
{mb_rag-1.1.41 → mb_rag-1.1.43}/setup.cfg RENAMED
File without changes
{mb_rag-1.1.41 → mb_rag-1.1.43}/setup.py RENAMED
File without changes