@@ -144,7 +144,7 @@ def optimize_python_code( # noqa: D417
144144 logger .info ("!lsp|Generating optimized candidates…" )
145145 console .rule ()
146146 try :
147- response = self .make_ai_service_request ("/optimize" , payload = payload , timeout = 600 )
147+ response = self .make_ai_service_request ("/optimize" , payload = payload , timeout = 60 )
148148 except requests .exceptions .RequestException as e :
149149 logger .exception (f"Error generating optimized candidates: { e } " )
150150 ph ("cli-optimize-error-caught" , {"error" : str (e )})
@@ -209,7 +209,7 @@ def optimize_python_code_line_profiler( # noqa: D417
209209 console .rule ()
210210 return []
211211 try :
212- response = self .make_ai_service_request ("/optimize-line-profiler" , payload = payload , timeout = 600 )
212+ response = self .make_ai_service_request ("/optimize-line-profiler" , payload = payload , timeout = 60 )
213213 except requests .exceptions .RequestException as e :
214214 logger .exception (f"Error generating optimized candidates: { e } " )
215215 ph ("cli-optimize-error-caught" , {"error" : str (e )})
@@ -261,7 +261,7 @@ def optimize_python_code_refinement(self, request: list[AIServiceRefinerRequest]
261261 logger .debug (f"Refining { len (request )} optimizations…" )
262262 console .rule ()
263263 try :
264- response = self .make_ai_service_request ("/refinement" , payload = payload , timeout = 600 )
264+ response = self .make_ai_service_request ("/refinement" , payload = payload , timeout = 120 )
265265 except requests .exceptions .RequestException as e :
266266 logger .exception (f"Error generating optimization refinements: { e } " )
267267 ph ("cli-optimize-error-caught" , {"error" : str (e )})
@@ -506,7 +506,7 @@ def generate_regression_tests( # noqa: D417
506506 "is_async" : function_to_optimize .is_async ,
507507 }
508508 try :
509- response = self .make_ai_service_request ("/testgen" , payload = payload , timeout = 600 )
509+ response = self .make_ai_service_request ("/testgen" , payload = payload , timeout = 90 )
510510 except requests .exceptions .RequestException as e :
511511 logger .exception (f"Error generating tests: { e } " )
512512 ph ("cli-testgen-error-caught" , {"error" : str (e )})
@@ -597,7 +597,7 @@ def get_optimization_review(
597597 }
598598 console .rule ()
599599 try :
600- response = self .make_ai_service_request ("/optimization_review" , payload = payload , timeout = 600 )
600+ response = self .make_ai_service_request ("/optimization_review" , payload = payload , timeout = 120 )
601601 except requests .exceptions .RequestException as e :
602602 logger .exception (f"Error generating optimization refinements: { e } " )
603603 ph ("cli-optimize-error-caught" , {"error" : str (e )})
0 commit comments