mb-rag 1.1.2__tar.gz → 1.1.4__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.
- {mb_rag-1.1.2 → mb_rag-1.1.4}/PKG-INFO +1 -1
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/chatbot/basic.py +3 -3
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/utils/bounding_box.py +1 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/version.py +1 -1
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag.egg-info/PKG-INFO +1 -1
- {mb_rag-1.1.2 → mb_rag-1.1.4}/README.md +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/__init__.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/chatbot/__init__.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/chatbot/chains.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/chatbot/prompts.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/rag/__init__.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/rag/embeddings.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/utils/__init__.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag/utils/extra.py +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag.egg-info/SOURCES.txt +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag.egg-info/dependency_links.txt +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag.egg-info/requires.txt +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/mb_rag.egg-info/top_level.txt +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/pyproject.toml +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/setup.cfg +0 -0
- {mb_rag-1.1.2 → mb_rag-1.1.4}/setup.py +0 -0
|
@@ -54,7 +54,7 @@ class ModelFactory:
|
|
|
54
54
|
"""
|
|
55
55
|
Factory method to create any type of model
|
|
56
56
|
Args:
|
|
57
|
-
model_type (str): Type of model to create
|
|
57
|
+
model_type (str): Type of model to create. Default is OpenAI. Options are openai, anthropic, google, ollama
|
|
58
58
|
model_name (str): Name of the model
|
|
59
59
|
**kwargs: Additional arguments
|
|
60
60
|
Returns:
|
|
@@ -120,8 +120,8 @@ class ModelFactory:
|
|
|
120
120
|
Returns:
|
|
121
121
|
ChatGoogleGenerativeAI: Chatbot model
|
|
122
122
|
"""
|
|
123
|
-
if not check_package("
|
|
124
|
-
raise ImportError("
|
|
123
|
+
if not check_package("langchain_google_genai"):
|
|
124
|
+
raise ImportError("langchain_google_genai package not found. Please install it using: pip install google-generativeai langchain-google-genai")
|
|
125
125
|
|
|
126
126
|
from langchain_google_genai import ChatGoogleGenerativeAI
|
|
127
127
|
kwargs["model"] = model_name
|
|
@@ -159,6 +159,7 @@ class BoundingBoxProcessor:
|
|
|
159
159
|
if google_bb:
|
|
160
160
|
value = [int(value[0] * img.shape[0] * 0.001), int(value[1] * img.shape[1]* 0.001),
|
|
161
161
|
int(value[2] * img.shape[0] * 0.001), int(value[3] * img.shape[1] * 0.001)]
|
|
162
|
+
print("Orignal Bounding Box from GOOGLE BBOX: ", value)
|
|
162
163
|
|
|
163
164
|
self._cv2.rectangle(
|
|
164
165
|
img=img,
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|