ubi integration

pages/Semantic_Search.py

@@ -102,7 +102,10 @@ if 'user_id' in st.session_state:
 
 
 if 'session_id' not in st.session_state:
-    st.session_state['session_id'] = ""
+    st.session_state['session_id'] = f"sess_{uuid.uuid4()}"
+
+if 'query_id' not in st.session_state:
+    st.session_state["query_id"] =""
     
 if 'input_reranker' not in st.session_state:
     st.session_state['input_reranker'] = "None"#"Cross Encoder"
@@ -922,7 +925,7 @@ def write_chat_message(md, q,index):
         res = json.loads(md['answer']['body'])
     else:
         res = md['answer']
-    st.session_state['session_id'] = "1234"
+    #st.session_state['session_id'] = "1234"
     chat = st.container()
     with chat:
         render_answer(res,index)

semantic_search/all_search_execute.py

@@ -518,8 +518,10 @@ def handler(input_,session_id):
             dup.append(doc['_source']['image_url'])
 
     ########### ubi lambda call for query logging ###########
+    st.session_state["query_id"] = str(uuid.uuid4())
     query_payload = {
-    "query_id": str(uuid.uuid4()),
+    "session_id": session_id,
+    "query_id": st.session_state["query_id"],
     "application": "Semantic Search",
     "query_response_hit_ids": doc_ids,
     "timestamp": datetime.utcnow().isoformat() + "Z",