Handle TimeoutError with retry (#113)

landing-ai · Jun 5, 2024 · 42455ee · 42455ee
1 parent bfdb16e
commit 42455ee
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/vision_agent/utils/execute.py b/vision_agent/utils/execute.py
@@ -426,6 +426,11 @@ def close(self, *args: Any, **kwargs: Any) -> None:
  def restart_kernel(self) -> None:
  self.interpreter.notebook.restart_kernel()
 
+ @tenacity.retry(
+ wait=tenacity.wait_exponential_jitter(),
+ stop=tenacity.stop_after_attempt(2),
+ retry=tenacity.retry_if_exception_type(TimeoutError),
+ )
  def exec_cell(self, code: str) -> Execution:
  execution = self.interpreter.notebook.exec_cell(code, timeout=self.timeout)
  return Execution.from_e2b_execution(execution)