mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-06 02:48:13 +00:00
parent
04e58e5ca1
commit
a27f5cc6c9
1 changed files with 4 additions and 11 deletions
|
|
@ -27,7 +27,6 @@ import litellm
|
|||
from litellm import (
|
||||
ModelResponse,
|
||||
RateLimitError,
|
||||
ServiceUnavailableError,
|
||||
Timeout,
|
||||
completion,
|
||||
completion_cost,
|
||||
|
|
@ -2021,16 +2020,10 @@ def test_bedrock_context_window_error():
|
|||
|
||||
def test_bedrock_converse_route():
|
||||
litellm.set_verbose = True
|
||||
try:
|
||||
litellm.completion(
|
||||
model="bedrock/converse/us.amazon.nova-pro-v1:0",
|
||||
messages=[{"role": "user", "content": "Hello, world!"}],
|
||||
)
|
||||
except ServiceUnavailableError as e:
|
||||
if "Too many requests" in str(e):
|
||||
pytest.skip("Skipping test due to AWS Bedrock rate limiting")
|
||||
else:
|
||||
raise
|
||||
litellm.completion(
|
||||
model="bedrock/converse/us.amazon.nova-pro-v1:0",
|
||||
messages=[{"role": "user", "content": "Hello, world!"}],
|
||||
)
|
||||
|
||||
|
||||
def test_bedrock_mapped_converse_models():
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue