mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-24 02:25:29 +00:00
test: remove redundant test
This commit is contained in:
@@ -1136,7 +1136,7 @@ async def test_aimg_gen_on_router():
|
||||
"api_base": os.getenv("AZURE_SWEDEN_API_BASE"),
|
||||
"api_key": os.getenv("AZURE_SWEDEN_API_KEY"),
|
||||
},
|
||||
}
|
||||
},
|
||||
]
|
||||
router = Router(model_list=model_list, num_retries=3)
|
||||
response = await router.aimage_generation(
|
||||
@@ -1308,41 +1308,6 @@ def test_azure_embedding_on_router():
|
||||
# test_azure_embedding_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()
|
||||
|
||||
|
||||
@@ -2786,4 +2751,4 @@ def test_router_get_model_group_info():
|
||||
assert model_group_info is not None
|
||||
assert model_group_info.model_group == "gpt-4"
|
||||
assert model_group_info.input_cost_per_token > 0
|
||||
assert model_group_info.output_cost_per_token > 0
|
||||
assert model_group_info.output_cost_per_token > 0
|
||||
|
||||
Reference in New Issue
Block a user