fix for issue No time zone found with key UTC

Pythagora-io · Jul 8, 2024 · 2984b3f · 2984b3f
1 parent cc64d9f
commit 2984b3f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/core/llm/anthropic_client.py b/core/llm/anthropic_client.py
@@ -116,7 +116,11 @@ def rate_limit_sleep(self, err: RateLimitError) -> Optional[datetime.timedelta]:
         except ValueError:
             return datetime.timedelta(seconds=5)
 
-        now = datetime.datetime.now(tz=zoneinfo.ZoneInfo("UTC"))
+        try:
+            now = datetime.datetime.now(tz=zoneinfo.ZoneInfo("UTC"))
+        except zoneinfo.ZoneInfoNotFoundError:
+            now = datetime.datetime.now(tz=datetime.timezone.utc)
+
         return reset_time - now