Merge branch 'main' into main-hf

# Conflicts: # CHANGELOG.md # streamlit_app.py
lfoppiano · Aug 23, 2024 · da1c929 · da1c929
2 parents a4a1d01 + caaa5c8
commit da1c929
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,11 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
+## [0.4.2] - 2024-08-23
+
+### Fixed
++ Correct invalid dependency of promptlayer slipped in the build 
+
 ## [0.4.1] - 2024-08-23
 
 ### Added

diff --git a/streamlit_app.py b/streamlit_app.py
@@ -6,7 +6,7 @@
 import dotenv
 from grobid_quantities.quantities import QuantitiesAPI
 from langchain.memory import ConversationBufferWindowMemory
-from langchain_community.callbacks import PromptLayerCallbackHandler
+# from langchain_community.callbacks import PromptLayerCallbackHandler
 from langchain_community.chat_models import ChatOpenAI
 from langchain_community.llms.huggingface_endpoint import HuggingFaceEndpoint
 from langchain_huggingface import HuggingFaceEmbeddings
@@ -174,7 +174,7 @@ def init_qa(model, embeddings_name=None, api_key=None):
             temperature=0.01,
             max_new_tokens=4092,
             model_kwargs={"max_length": 8192},
-            callbacks=[PromptLayerCallbackHandler(pl_tags=[model, "document-qa"])]
+            # callbacks=[PromptLayerCallbackHandler(pl_tags=[model, "document-qa"])]
         )
         embeddings = HuggingFaceEmbeddings(
             model_name=OPEN_EMBEDDINGS[embeddings_name])