From e3cdfa6c63a80cbec8aa7c5f1cf61cdaa68c0e79 Mon Sep 17 00:00:00 2001
From: Zachary Lee <w.zachary.lee@gmail.com>
Date: Fri, 22 Nov 2024 18:25:30 -0500
Subject: [PATCH] increased max_retries to 4 on 1p api (#175)

---
 computer-use-demo/computer_use_demo/loop.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/computer-use-demo/computer_use_demo/loop.py b/computer-use-demo/computer_use_demo/loop.py
index 9db525a3..de91c430 100644
--- a/computer-use-demo/computer_use_demo/loop.py
+++ b/computer-use-demo/computer_use_demo/loop.py
@@ -103,7 +103,7 @@ async def sampling_loop(
         betas = [COMPUTER_USE_BETA_FLAG]
         image_truncation_threshold = 10
         if provider == APIProvider.ANTHROPIC:
-            client = Anthropic(api_key=api_key)
+            client = Anthropic(api_key=api_key, max_retries=4)
             enable_prompt_caching = True
         elif provider == APIProvider.VERTEX:
             client = AnthropicVertex()