Connor Sutton
commited on
Commit
·
5cdfa32
1
Parent(s):
fc0a7ff
use cache_resource instead of cache_data when getting texts and retriever
Browse files
langchain-streamlit-demo/app.py
CHANGED
|
@@ -122,7 +122,7 @@ AZURE_KWARGS = (
|
|
| 122 |
)
|
| 123 |
|
| 124 |
|
| 125 |
-
@st.
|
| 126 |
def get_texts_and_retriever_cacheable_wrapper(
|
| 127 |
uploaded_file_bytes: bytes,
|
| 128 |
openai_api_key: str,
|
|
|
|
| 122 |
)
|
| 123 |
|
| 124 |
|
| 125 |
+
@st.cache_resource
|
| 126 |
def get_texts_and_retriever_cacheable_wrapper(
|
| 127 |
uploaded_file_bytes: bytes,
|
| 128 |
openai_api_key: str,
|
langchain-streamlit-demo/llm_resources.py
CHANGED
|
@@ -110,7 +110,6 @@ def get_llm(
|
|
| 110 |
|
| 111 |
return None
|
| 112 |
|
| 113 |
-
|
| 114 |
def get_texts_and_retriever(
|
| 115 |
uploaded_file_bytes: bytes,
|
| 116 |
openai_api_key: str,
|
|
@@ -146,7 +145,7 @@ def get_texts_and_retriever(
|
|
| 146 |
retrievers=[bm25_retriever, faiss_retriever],
|
| 147 |
weights=[0.5, 0.5],
|
| 148 |
)
|
| 149 |
-
|
| 150 |
return texts, ensemble_retriever
|
| 151 |
|
| 152 |
|
|
|
|
| 110 |
|
| 111 |
return None
|
| 112 |
|
|
|
|
| 113 |
def get_texts_and_retriever(
|
| 114 |
uploaded_file_bytes: bytes,
|
| 115 |
openai_api_key: str,
|
|
|
|
| 145 |
retrievers=[bm25_retriever, faiss_retriever],
|
| 146 |
weights=[0.5, 0.5],
|
| 147 |
)
|
| 148 |
+
|
| 149 |
return texts, ensemble_retriever
|
| 150 |
|
| 151 |
|