fix azure gpt-4o test

2026-07-29 04:20:42 +00:00 · 2024-09-19 16:20:43 -07:00
parent b022247168
commit b54bbf510e
1 changed files with 3 additions and 0 deletions
@@ -935,6 +935,9 @@ def test_completion_base64(model):
        print(f"\nResponse: {resp}")

        prompt_tokens = resp.usage.prompt_tokens
+    except litellm.ServiceUnavailableError as e:
+        print("got service unavailable error: ", e)
+        pass
    except Exception as e:
        if "500 Internal error encountered.'" in str(e):
            pass