prasadnu commited on
Commit
88dc90c
·
1 Parent(s): 1ad7661

model changed to Haiku 3.5 for query re=write

Browse files
Files changed (1) hide show
  1. semantic_search/query_rewrite.py +2 -1
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
- import time
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)