Merge pull request #8 from google-marketing-solutions/dev

Update README and increase rate limiting threshold
google-marketing-solutions · Aug 13, 2024 · 6d79677 · 6d79677
2 parents 8fe1afd + d64a97d
commit 6d79677
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/common/vertex_client.py b/api/common/vertex_client.py
@@ -112,9 +112,9 @@ def generate_descriptions_from_medias(
       retry=tenacity.retry_if_exception_type(
           google.api_core.exceptions.ResourceExhausted
       ),
-      wait=tenacity.wait_exponential(min=3, multiplier=3, max=60),
+      wait=tenacity.wait_exponential(min=5, multiplier=2, max=70),
       reraise=False,
-      stop=tenacity.stop_after_attempt(5),
+      stop=tenacity.stop_after_attempt(10),
   )
   def _generate_descriptions_from_media(
       self, media_path: str