Spaces:
Running
on
T4
Running
on
T4
model changed to Haiku 3.5 for query re=write
Browse files
semantic_search/query_rewrite.py
CHANGED
@@ -29,6 +29,7 @@ from langchain.embeddings import BedrockEmbeddings
|
|
29 |
#from langchain.vectorstores import OpenSearchVectorSearch
|
30 |
from opensearchpy import OpenSearch, RequestsHttpConnection
|
31 |
import utilities.invoke_models as invoke_models
|
|
|
32 |
|
33 |
|
34 |
|
@@ -234,7 +235,7 @@ def get_new_query_res(query):
|
|
234 |
if(query == ""):
|
235 |
query = st.session_state.input_rekog_label
|
236 |
if(st.session_state.input_is_rewrite_query == 'enabled'):
|
237 |
-
|
238 |
|
239 |
start = time.time()
|
240 |
res = invoke_models.invoke_llm_model( prompt_.format(query=query,schema = schema) ,False)
|
|
|
29 |
#from langchain.vectorstores import OpenSearchVectorSearch
|
30 |
from opensearchpy import OpenSearch, RequestsHttpConnection
|
31 |
import utilities.invoke_models as invoke_models
|
32 |
+
import time
|
33 |
|
34 |
|
35 |
|
|
|
235 |
if(query == ""):
|
236 |
query = st.session_state.input_rekog_label
|
237 |
if(st.session_state.input_is_rewrite_query == 'enabled'):
|
238 |
+
|
239 |
|
240 |
start = time.time()
|
241 |
res = invoke_models.invoke_llm_model( prompt_.format(query=query,schema = schema) ,False)
|