prashant
commited on
Commit
·
2918b13
1
Parent(s):
2c7f87d
session state error fix
Browse files- utils/semantic_search.py +19 -12
utils/semantic_search.py
CHANGED
@@ -222,19 +222,26 @@ def semanticSearchPipeline(documents:List[Document], embedding_model:Text = Non
|
|
222 |
if 'retriever' in st.session_state:
|
223 |
# if st.session_state['retriever']:
|
224 |
retriever = st.session_state['retriever']
|
225 |
-
else:
|
226 |
-
if embedding_model:
|
227 |
-
retriever = loadRetriever(embedding_model = embedding_model,
|
228 |
-
embedding_model_format=embedding_model_format,
|
229 |
-
embedding_layer=embedding_layer,
|
230 |
-
retriever_top_k= retriever_top_k,
|
231 |
-
document_store = document_store)
|
232 |
-
|
233 |
-
st.session_state['retriever'] = retriever
|
234 |
else:
|
235 |
-
|
236 |
-
|
237 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
238 |
|
239 |
document_store.update_embeddings(retriever)
|
240 |
retriever.document_store = document_store
|
|
|
222 |
if 'retriever' in st.session_state:
|
223 |
# if st.session_state['retriever']:
|
224 |
retriever = st.session_state['retriever']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
225 |
else:
|
226 |
+
if embedding_model:
|
227 |
+
retriever = loadRetriever(embedding_model = embedding_model,
|
228 |
+
embedding_model_format=embedding_model_format,
|
229 |
+
embedding_layer=embedding_layer,
|
230 |
+
retriever_top_k= retriever_top_k,
|
231 |
+
document_store = document_store)
|
232 |
+
|
233 |
+
st.session_state['retriever'] = retriever
|
234 |
+
else:
|
235 |
+
logging.warning("no streamlit enviornment found, neither embedding model \
|
236 |
+
provided")
|
237 |
+
return
|
238 |
+
elif embedding_model:
|
239 |
+
retriever = loadRetriever(embedding_model = embedding_model,
|
240 |
+
embedding_model_format=embedding_model_format,
|
241 |
+
embedding_layer=embedding_layer,
|
242 |
+
retriever_top_k= retriever_top_k,
|
243 |
+
document_store = document_store)
|
244 |
+
|
245 |
|
246 |
document_store.update_embeddings(retriever)
|
247 |
retriever.document_store = document_store
|