added print logs

RAG/rag_DocumentSearcher.py

@@ -53,12 +53,12 @@ def query_(awsauth,inputs, session_id,search_types):
         #         }
         # }
     }
-
+    print("started mm_search: "+st.session_state.input_index)
     path = st.session_state.input_index+"_mm/_search"
     url = host+path
     r = requests.get(url, auth=awsauth, json=query_mm, headers=headers)
     response_mm = json.loads(r.text)
-
+    print("Finished mm_search: "+st.session_state.input_index)
     hits = response_mm['hits']['hits']
     context = []
     context_tables = []
@@ -227,9 +227,11 @@ def query_(awsauth,inputs, session_id,search_types):
                                              "query_text": question
                                           }
                                         }}
+        print("started main search with/wo rr: "+st.session_state.input_index)
         r = requests.get(url, auth=awsauth, json=hybrid_payload, headers=headers)
         response_ = json.loads(r.text)
-        print(response_)
+        print("Finished main search with/wo rr: "+st.session_state.input_index)
+        #print(response_)
         hits = response_['hits']['hits']
         
     else:
@@ -341,7 +343,9 @@ def query_(awsauth,inputs, session_id,search_types):
     
 
     llm_prompt = prompt_template.format(context="\n".join(total_context[0:3]),question=question)
+    print("started LLM prompt: "+st.session_state.input_index)
     output = invoke_models.invoke_llm_model( "\n\nHuman: {input}\n\nAssistant:".format(input=llm_prompt) ,False)
+    print("Finished LLM prompt: "+st.session_state.input_index)
     if(len(images_2)==0):
         images_2 = images
     return {'text':output,'source':total_context,'image':images_2,'table':df}