mb-rag 1.1.64__py3-none-any.whl → 1.1.65__py3-none-any.whl

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 (6) hide show
  1. mb_rag/utils/llm_wrapper.py +22 -0
  2. mb_rag/version.py +1 -1
  3. {mb_rag-1.1.64.dist-info → mb_rag-1.1.65.dist-info}/METADATA +1 -1
  4. {mb_rag-1.1.64.dist-info → mb_rag-1.1.65.dist-info}/RECORD +6 -5
  5. {mb_rag-1.1.64.dist-info → mb_rag-1.1.65.dist-info}/WHEEL +0 -0
  6. {mb_rag-1.1.64.dist-info → mb_rag-1.1.65.dist-info}/top_level.txt +0 -0
mb_rag/utils/llm_wrapper.py ADDED
@@ -0,0 +1,22 @@
1
+ ## simple llm wrapper to replace invoke with invoke_query/own model query
2
+
3
+ __all__ = ["LLMWrapper"]
4
+
5
+ class LLMWrapper:
6
+ """A simple wrapper for the language model to standardize the invoke method.
7
+ """
8
+
9
+ def __init__(self, llm):
10
+ self.llm = llm
11
+
12
+ def invoke_query(self, prompt: str) -> str:
13
+ """
14
+ Invoke the language model with the given prompt.
15
+
16
+ Args:
17
+ prompt (str): The prompt to send to the language model.
18
+
19
+ Returns:
20
+ str: The generated response.
21
+ """
22
+ return self.llm.invoke(prompt)
mb_rag/version.py CHANGED
@@ -1,5 +1,5 @@
1
1
  MAJOR_VERSION = 1
2
2
  MINOR_VERSION = 1
3
- PATCH_VERSION = 64
3
+ PATCH_VERSION = 65
4
4
  version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
5
5
  __all__ = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']
{mb_rag-1.1.64.dist-info → mb_rag-1.1.65.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: mb_rag
3
- Version: 1.1.64
3
+ Version: 1.1.65
4
4
  Summary: RAG function file
5
5
  Author: ['Malav Bateriwala']
6
6
  Requires-Python: >=3.8
{mb_rag-1.1.64.dist-info → mb_rag-1.1.65.dist-info}/RECORD RENAMED
@@ -1,7 +1,7 @@
1
1
  mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
2
  mb_rag/basic.py,sha256=83ZOfj-nc1kV5OOuWTi2o760sqPB_v9M4fF9WPlbM9o,15149
3
3
  mb_rag/prompts_bank.py,sha256=NOY-a_0FyRvWpVuAiP6KFEPIgNLR-kVxQqNzGiqmq0c,2868
4
- mb_rag/version.py,sha256=0I-AGP8H-6XPb7DUSmz5LqaNl9RjWW9Caubjh7VQ9b8,207
4
+ mb_rag/version.py,sha256=Zm9JibUoh0fdvLsRE-ZebXhKkEdDYtD4W0_uNe129Mk,207
5
5
  mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
6
6
  mb_rag/chatbot/chains.py,sha256=vDbLX5R29sWN1pcFqJ5fyxJEgMCM81JAikunAEvMC9A,7223
7
7
  mb_rag/chatbot/conversation.py,sha256=6EsQ5U5KVgXqRU5mX2tellZHia9aA0_QLeytJFyewpg,6994
@@ -12,8 +12,9 @@ mb_rag/utils/all_data_extract.py,sha256=TL6O4vNc7mPW-OGK-LhXQQIkSr0o3_7BqNAD-YpT
12
12
  mb_rag/utils/bounding_box.py,sha256=G0hdDam8QmYtD9lfwMeDHGm-TTo6KZg-yK5ESFL9zaM,8366
13
13
  mb_rag/utils/document_extract.py,sha256=vZiFB1RYm1BIEaNA0MveJ5Zp-KEi0ngKjW8xEdtPqXA,12558
14
14
  mb_rag/utils/extra.py,sha256=spbFrGgdruNyYQ5PzgvpSIa6Nm0rn9bb4qc8W9g582o,2492
15
+ mb_rag/utils/llm_wrapper.py,sha256=F6GkiBA3ZxhD_dISnTmItXJKe0zRHvYSs-rC_RGNtzQ,587
15
16
  mb_rag/utils/pdf_extract.py,sha256=cVeMyhnAU4XZxjIZHKMYhrktTjUNOjhx2r_LZKReOZE,15598
16
- mb_rag-1.1.64.dist-info/METADATA,sha256=RyEURUE7Wc2OQuCDlWayEFCVpPPNa3YbFDABvqEQU8Y,234
17
- mb_rag-1.1.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
18
- mb_rag-1.1.64.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
19
- mb_rag-1.1.64.dist-info/RECORD,,
17
+ mb_rag-1.1.65.dist-info/METADATA,sha256=9gKHF9JLLDBlrugwhqkkUjYSerLQWL1uc-9xC3GOQus,234
18
+ mb_rag-1.1.65.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
19
+ mb_rag-1.1.65.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
20
+ mb_rag-1.1.65.dist-info/RECORD,,