mb-rag 1.1.58__tar.gz → 1.1.59__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 (25) hide show
  1. {mb_rag-1.1.58 → mb_rag-1.1.59}/PKG-INFO +1 -1
  2. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/basic.py +3 -2
  3. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/version.py +1 -1
  4. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag.egg-info/PKG-INFO +1 -1
  5. {mb_rag-1.1.58 → mb_rag-1.1.59}/README.md +0 -0
  6. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/__init__.py +0 -0
  7. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/chatbot/__init__.py +0 -0
  8. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/chatbot/chains.py +0 -0
  9. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/chatbot/conversation.py +0 -0
  10. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/chatbot/prompts.py +0 -0
  11. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/rag/__init__.py +0 -0
  12. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/rag/embeddings.py +0 -0
  13. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/__init__.py +0 -0
  14. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/all_data_extract.py +0 -0
  15. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/bounding_box.py +0 -0
  16. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/document_extract.py +0 -0
  17. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/extra.py +0 -0
  18. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/pdf_extract.py +0 -0
  19. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag.egg-info/SOURCES.txt +0 -0
  20. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag.egg-info/dependency_links.txt +0 -0
  21. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag.egg-info/requires.txt +0 -0
  22. {mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag.egg-info/top_level.txt +0 -0
  23. {mb_rag-1.1.58 → mb_rag-1.1.59}/pyproject.toml +0 -0
  24. {mb_rag-1.1.58 → mb_rag-1.1.59}/setup.cfg +0 -0
  25. {mb_rag-1.1.58 → mb_rag-1.1.59}/setup.py +0 -0
{mb_rag-1.1.58 → mb_rag-1.1.59}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mb_rag
3
- Version: 1.1.58
3
+ Version: 1.1.59
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/basic.py RENAMED
@@ -2,6 +2,7 @@
2
2
 
3
3
  import os
4
4
  from langchain_core.messages import HumanMessage
5
+ import torch
5
6
  from mb_rag.utils.extra import check_package
6
7
  import base64
7
8
  from .utils.extra import check_package
@@ -41,7 +42,7 @@ class ModelFactory:
41
42
 
42
43
  self.model_type = model_type
43
44
  self.model_name = model_name
44
- model_data = creators.get(model_type)
45
+ model_data = creators[model_type] if model_type in creators else None
45
46
  if not model_data:
46
47
  raise ValueError(f"Unsupported model type: {model_type}")
47
48
 
@@ -195,7 +196,7 @@ class ModelFactory:
195
196
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, AutoModelForImageTextToText,AutoProcessor
196
197
  import torch
197
198
 
198
- device = torch.device(device) if torch.cuda.is_available() else torch.device("cpu")
199
+ device = torch.device("cuda" if torch.cuda.is_available() and device == "cuda" else "cpu")
199
200
 
200
201
  temperature = kwargs.pop("temperature", 0.7)
201
202
  max_length = kwargs.pop("max_length", 1024)
{mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/version.py RENAMED
@@ -1,5 +1,5 @@
1
1
  MAJOR_VERSION = 1
2
2
  MINOR_VERSION = 1
3
- PATCH_VERSION = 58
3
+ PATCH_VERSION = 59
4
4
  version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
5
5
  __all__ = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']
{mb_rag-1.1.58 → mb_rag-1.1.59}/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.58
3
+ Version: 1.1.59
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.58 → mb_rag-1.1.59}/README.md RENAMED
File without changes
{mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/__init__.py RENAMED
File without changes
{mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/rag/__init__.py RENAMED
File without changes
{mb_rag-1.1.58 → mb_rag-1.1.59}/mb_rag/utils/extra.py RENAMED
File without changes
{mb_rag-1.1.58 → mb_rag-1.1.59}/pyproject.toml RENAMED
File without changes
{mb_rag-1.1.58 → mb_rag-1.1.59}/setup.cfg RENAMED
File without changes
{mb_rag-1.1.58 → mb_rag-1.1.59}/setup.py RENAMED
File without changes