Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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)
|