test: fix test

This commit is contained in:
Krrish Dholakia 2024-04-28 09:45:01 -07:00
parent 39244cd517
commit 1f6c342e94

View file

@ -2722,7 +2722,7 @@ def test_unified_auth_params(provider, model, project, region_name, token):
@pytest.mark.asyncio
async def test_acompletion_watsonx():
litellm.set_verbose = True
model_name = "watsonx/deployment/" + os.getenv("WATSONX_DEPLOYMENT_ID")
model_name = "watsonx/ibm/granite-13b-chat-v2"
print("testing watsonx")
try:
response = await litellm.acompletion(