prasadnu commited on
Commit
3be1d39
·
1 Parent(s): b1dfcfd

ubi integration

Browse files
semantic_search/all_search_execute.py CHANGED
@@ -485,7 +485,7 @@ def handler(input_,session_id):
485
 
486
  arr = []
487
  dup = []
488
- docs = []
489
  for doc in docs:
490
  if(doc['_source']['image_url'] not in dup):
491
  res_ = {
@@ -509,7 +509,7 @@ def handler(input_,session_id):
509
  # res_['rekog'] = {'color':doc['_source']['rekog_color'],'category': doc['_source']['rekog_categories'],'objects':doc['_source']['rekog_objects']}
510
 
511
  res_['id'] = doc['_id']
512
- docs.append(doc['_id'])
513
  res_['score'] = doc['_score']
514
  res_['title'] = doc['_source']['product_description']
515
 
@@ -521,7 +521,7 @@ def handler(input_,session_id):
521
  query_payload = {
522
  "query_id": str(uuid.uuid4()),
523
  "application": "Semantic Search",
524
- "query_response_hit_ids": docs,
525
  "timestamp": datetime.utcnow().isoformat() + "Z",
526
  "user_query": json.dumps(hybrid_payload),
527
 
 
485
 
486
  arr = []
487
  dup = []
488
+ doc_ids = []
489
  for doc in docs:
490
  if(doc['_source']['image_url'] not in dup):
491
  res_ = {
 
509
  # res_['rekog'] = {'color':doc['_source']['rekog_color'],'category': doc['_source']['rekog_categories'],'objects':doc['_source']['rekog_objects']}
510
 
511
  res_['id'] = doc['_id']
512
+ doc_ids.append(doc['_id'])
513
  res_['score'] = doc['_score']
514
  res_['title'] = doc['_source']['product_description']
515
 
 
521
  query_payload = {
522
  "query_id": str(uuid.uuid4()),
523
  "application": "Semantic Search",
524
+ "query_response_hit_ids": doc_ids,
525
  "timestamp": datetime.utcnow().isoformat() + "Z",
526
  "user_query": json.dumps(hybrid_payload),
527