test: fix watsonx api error

2024-05-13 19:01:19 -07:00 · 2024-05-13 19:01:19 -07:00 · 071a70c5fc
commit 071a70c5fc
parent 724d880a45
1 changed files with 2 additions and 0 deletions
--- a/litellm/tests/test_completion.py
+++ b/litellm/tests/test_completion.py
@ -3363,6 +3363,8 @@ def test_completion_stream_watsonx():
            print(chunk)
    except litellm.APIError as e:
        pass
+    except litellm.RateLimitError as e:
+        pass
    except Exception as e:
        pytest.fail(f"Error occurred: {e}")