mb-rag 1.1.42__tar.gz → 1.1.44__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.
Files changed (23) hide show
  1. {mb_rag-1.1.42 → mb_rag-1.1.44}/PKG-INFO +1 -1
  2. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/chatbot/basic.py +1 -1
  3. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/rag/embeddings.py +2 -1
  4. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/version.py +1 -1
  5. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag.egg-info/PKG-INFO +1 -1
  6. {mb_rag-1.1.42 → mb_rag-1.1.44}/README.md +0 -0
  7. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/__init__.py +0 -0
  8. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/chatbot/__init__.py +0 -0
  9. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/chatbot/chains.py +0 -0
  10. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/chatbot/prompts.py +0 -0
  11. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/rag/__init__.py +0 -0
  12. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/utils/__init__.py +0 -0
  13. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/utils/bounding_box.py +0 -0
  14. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/utils/document_extract.py +0 -0
  15. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/utils/extra.py +0 -0
  16. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/utils/pdf_extract.py +0 -0
  17. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag.egg-info/SOURCES.txt +0 -0
  18. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag.egg-info/dependency_links.txt +0 -0
  19. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag.egg-info/requires.txt +0 -0
  20. {mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag.egg-info/top_level.txt +0 -0
  21. {mb_rag-1.1.42 → mb_rag-1.1.44}/pyproject.toml +0 -0
  22. {mb_rag-1.1.42 → mb_rag-1.1.44}/setup.cfg +0 -0
  23. {mb_rag-1.1.42 → mb_rag-1.1.44}/setup.py +0 -0
{mb_rag-1.1.42 → mb_rag-1.1.44}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mb_rag
3
- Version: 1.1.42
3
+ Version: 1.1.44
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/chatbot/basic.py RENAMED
@@ -119,7 +119,7 @@ class ModelFactory:
119
119
  return ChatAnthropic(**kwargs)
120
120
 
121
121
  @classmethod
122
- def create_google(cls, model_name: str = "gemini-1.5-flash", **kwargs) -> Any:
122
+ def create_google(cls, model_name: str = "gemini-2.0-flash", **kwargs) -> Any:
123
123
  """
124
124
  Create Google chatbot model
125
125
  Args:
{mb_rag-1.1.42 → mb_rag-1.1.44}/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.42 → mb_rag-1.1.44}/mb_rag/version.py RENAMED
@@ -1,5 +1,5 @@
1
1
  MAJOR_VERSION = 1
2
2
  MINOR_VERSION = 1
3
- PATCH_VERSION = 42
3
+ PATCH_VERSION = 44
4
4
  version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
5
5
  __all__ = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']
{mb_rag-1.1.42 → mb_rag-1.1.44}/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.42
3
+ Version: 1.1.44
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.42 → mb_rag-1.1.44}/README.md RENAMED
File without changes
{mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/__init__.py RENAMED
File without changes
{mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/rag/__init__.py RENAMED
File without changes
{mb_rag-1.1.42 → mb_rag-1.1.44}/mb_rag/utils/extra.py RENAMED
File without changes
{mb_rag-1.1.42 → mb_rag-1.1.44}/pyproject.toml RENAMED
File without changes
{mb_rag-1.1.42 → mb_rag-1.1.44}/setup.cfg RENAMED
File without changes
{mb_rag-1.1.42 → mb_rag-1.1.44}/setup.py RENAMED
File without changes