Update generating responnse time

main.py

@@ -523,6 +523,7 @@ class AIAgent:
                 print(f"Parsed relative date: {parsed_date}")
             
             # 1. Route the query to determine which API endpoint to call
+            new_start_time = time.time()
             router_result = self.router_chain.invoke({
                 "endpoints_documentation": json.dumps(self.endpoints_documentation, indent=2),
                 "user_query": user_query,
@@ -532,7 +533,7 @@ class AIAgent:
             })
 
             # TODO: remove the print statement in production
-            print('End time of generating response: ', time.time() - start_time)
+            print('End time of generating response: ', time.time() - new_start_time)
             
             # 2. Parse the router response
             route_result = router_result["route_result"]