remove warm-up

RAG/bedrock_agent.py

@@ -54,8 +54,8 @@ def query_(inputs):
     )
 
     logger.info(pprint.pprint(agentResponse))
-    print("***agent*****response*********")
-    print(agentResponse)
+    #print("***agent*****response*********")
+    #print(agentResponse)
     event_stream = agentResponse['completion']
     total_context = []
     last_tool = ""
@@ -63,11 +63,11 @@ def query_(inputs):
     agent_answer = ""
     try:
         for event in event_stream:
-            print("***event*********")
-            print(event)
+            #print("***event*********")
+            #print(event)
             if 'trace' in event: 
-                print("trace*****total*********")
-                print(event['trace'])
+                #print("trace*****total*********")
+                #print(event['trace'])
                 if('orchestrationTrace' not in event['trace']['trace']):
                     continue
                 orchestration_trace = event['trace']['trace']['orchestrationTrace']
@@ -80,10 +80,10 @@ def query_(inputs):
                     total_context_item['invocationInput'] = orchestration_trace['invocationInput']['actionGroupInvocationInput']
                     last_tool_name = total_context_item['invocationInput']['function']
                 if('observation'  in orchestration_trace):
-                    print("trace****observation******")
+                    #print("trace****observation******")
                     total_context_item['observation'] = event['trace']['trace']['orchestrationTrace']['observation']
                     tool_output_last_obs = event['trace']['trace']['orchestrationTrace']['observation']
-                    print(tool_output_last_obs)
+                    #print(tool_output_last_obs)
                     if(tool_output_last_obs['type'] == 'ACTION_GROUP'):
                         last_tool = tool_output_last_obs['actionGroupInvocationOutput']['text']
                     if(tool_output_last_obs['type'] == 'FINISH'):   
@@ -92,8 +92,8 @@ def query_(inputs):
                     total_context_item['thinking'] = orchestration_trace['modelInvocationOutput']['rawResponse']
                 if(total_context_item!={}):
                     total_context.append(total_context_item)
-        print("total_context------")
-        print(total_context)    
+        #print("total_context------")
+        #print(total_context)    
     except botocore.exceptions.EventStreamError as error:
         raise error
         

pages/AI_Shopping_Assistant.py

@@ -152,7 +152,7 @@ def handle_input():
         'id': len(st.session_state.questions__)
     }
     st.session_state.questions__.append(question_with_id)
-    print(inputs)
+    #print(inputs)
     out_ = bedrock_agent.query_(inputs)
     st.session_state.answers__.append({
         'answer': out_['text'],