ubi integration

semantic_search/all_search_execute.py

@@ -484,7 +484,8 @@ def handler(input_,session_id):
             
             
     arr = []
-    dup = []
+    dup = [] 
+    docs = []
     for doc in docs:
         if(doc['_source']['image_url'] not in dup):
             res_ = {
@@ -508,6 +509,7 @@ def handler(input_,session_id):
 #                 res_['rekog'] = {'color':doc['_source']['rekog_color'],'category': doc['_source']['rekog_categories'],'objects':doc['_source']['rekog_objects']}
             
             res_['id'] = doc['_id']
+            docs.append(doc['_id'])
             res_['score'] = doc['_score']
             res_['title'] = doc['_source']['product_description']
             
@@ -517,7 +519,10 @@ def handler(input_,session_id):
 
     ########### ubi lambda call for query logging ###########
     query_payload = {
-    "query_id": "abc123",
+    "query_id": str(uuid.uuid4()),
+    "application": "Semantic Search",
+    "query_response_hit_ids": docs,
+    "timestamp": datetime.utcnow().isoformat() + "Z",
     "user_query": json.dumps(hybrid_payload),
     
     }