mb-rag 1.1.25__py3-none-any.whl → 1.1.28__py3-none-any.whl

mb_rag/chatbot/basic.py

@@ -146,10 +146,11 @@ class ModelFactory:
         if not check_package("langchain_ollama"):
             raise ImportError("Langchain Community package not found. Please install it using: pip install langchain_ollama")
         
-        from langchain_ollama import ChatOllama
+        from langchain_ollama import OllamaLLM
+
         print(f"Current Ollama serve model is {os.system('ollama ps')}")
         kwargs["model"] = model_name
-        return ChatOllama(**kwargs)
+        return OllamaLLM(**kwargs)
 
     @classmethod
     def create_groq(cls, model_name: str = "llama-3.3-70b-versatile", **kwargs) -> Any:

mb_rag/rag/embeddings.py

@@ -585,7 +585,7 @@ class embedding_generator:
             if not ModelProvider.check_package("langchain_openai"):
                 raise ImportError("OpenAI package not found. Please install: pip install langchain-openai")
             from langchain_openai import ChatOpenAI
-            llm = ChatOpenAI(model="gpt-4")
+            llm = ChatOpenAI(model="gpt-4o", temperature=0.8)
 
         history_aware_retriever = create_history_aware_retriever(llm, retriever,
                                                                contextualize_q_prompt)

mb_rag/version.py

@@ -1,5 +1,5 @@
 MAJOR_VERSION = 1
 MINOR_VERSION = 1
-PATCH_VERSION = 25
+PATCH_VERSION = 28
 version = '{}.{}.{}'.format(MAJOR_VERSION, MINOR_VERSION, PATCH_VERSION)
 __all__  = ['MAJOR_VERSION', 'MINOR_VERSION', 'PATCH_VERSION', 'version']

{mb_rag-1.1.25.dist-info → mb_rag-1.1.28.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mb_rag
-Version: 1.1.25
+Version: 1.1.28
 Summary: RAG function file
 Author: ['Malav Bateriwala']
 Requires-Python: >=3.8

{mb_rag-1.1.25.dist-info → mb_rag-1.1.28.dist-info}/RECORD

@@ -1,17 +1,17 @@
 mb_rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/version.py,sha256=jC-frepvF-y9ExGGQrOJNitI1xUCJRc5Ci2_oZF3Sz0,207
+mb_rag/version.py,sha256=Jjd2mCudGZRz7bRa7R4U_JjRiyIfx48EnBIOJCCQwhM,207
 mb_rag/chatbot/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/chatbot/basic.py,sha256=_D2U3QzRGs7BPT691ii3TM49Gj3d1mKXnlFdniz2Wy0,23104
+mb_rag/chatbot/basic.py,sha256=_-ZYE_xMX__H9gGGFI96oQ0R-nwhikGDFcFa91CIS4w,23103
 mb_rag/chatbot/chains.py,sha256=vDbLX5R29sWN1pcFqJ5fyxJEgMCM81JAikunAEvMC9A,7223
 mb_rag/chatbot/prompts.py,sha256=n1PyiLbU-5fkslRv6aVOzt0dDlwya_cEdQ7kRnRhMuY,1749
 mb_rag/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mb_rag/rag/embeddings.py,sha256=KjBdekFDb5M3dRMco4r3dDMXMsG5dxdzKImuVIipsd0,27091
+mb_rag/rag/embeddings.py,sha256=eutlhc8Pm3zfXu3NTSGEm7A_NpvyxE99Ocn9Vw_IOrM,27109
 mb_rag/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 mb_rag/utils/bounding_box.py,sha256=G0hdDam8QmYtD9lfwMeDHGm-TTo6KZg-yK5ESFL9zaM,8366
 mb_rag/utils/document_extract.py,sha256=vZiFB1RYm1BIEaNA0MveJ5Zp-KEi0ngKjW8xEdtPqXA,12558
 mb_rag/utils/extra.py,sha256=spbFrGgdruNyYQ5PzgvpSIa6Nm0rn9bb4qc8W9g582o,2492
 mb_rag/utils/pdf_extract.py,sha256=cVeMyhnAU4XZxjIZHKMYhrktTjUNOjhx2r_LZKReOZE,15598
-mb_rag-1.1.25.dist-info/METADATA,sha256=slQ2Q9jVH8aFg5fTk20iVMq6Fg4VMqaiz3MwmuCiTa4,234
-mb_rag-1.1.25.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-mb_rag-1.1.25.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
-mb_rag-1.1.25.dist-info/RECORD,,
+mb_rag-1.1.28.dist-info/METADATA,sha256=SrvA8RRcIxPKtwB19iQjO4-m8XuqKpmDkTT0u8p1Tu4,234
+mb_rag-1.1.28.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+mb_rag-1.1.28.dist-info/top_level.txt,sha256=FIK1eAa5uYnurgXZquBG-s3PIy-HDTC5yJBW4lTH_pM,7
+mb_rag-1.1.28.dist-info/RECORD,,