diff --git a/litellm/tests/test_router.py b/litellm/tests/test_router.py index 8024c9dd2d..ae9a06c41f 100644 --- a/litellm/tests/test_router.py +++ b/litellm/tests/test_router.py @@ -329,4 +329,39 @@ def test_azure_aembedding_on_router(): except Exception as e: traceback.print_exc() pytest.fail(f"Error occurred: {e}") -# test_azure_aembedding_on_router() \ No newline at end of file +# test_azure_aembedding_on_router() + + +def test_bedrock_on_router(): + litellm.set_verbose = True + print("\n Testing bedrock on router\n") + try: + model_list = [ + { + "model_name": "claude-v1", + "litellm_params": { + "model": "bedrock/anthropic.claude-instant-v1", + }, + "tpm": 100000, + "rpm": 10000, + }, + ] + + async def test(): + router = Router(model_list=model_list) + response = await router.acompletion( + model="claude-v1", + messages=[ + { + "role": "user", + "content": "hello from litellm test", + } + ] + ) + print(response) + router.reset() + asyncio.run(test()) + except Exception as e: + traceback.print_exc() + pytest.fail(f"Error occurred: {e}") +# test_bedrock_on_router() \ No newline at end of file