phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix(main.py): accepting azure deployment_id

Browse source
This commit is contained in:
Krrish Dholakia 2023-11-09 18:15:54 -08:00
parent 523c540051
commit 249cde3d40
3 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

7
litellm/tests/test_completion.py
View file

@ -445,7 +445,7 @@ def test_completion_openai_litellm_key():
except Exception as e:
pytest.fail(f"Error occurred: {e}")
test_completion_openai_litellm_key()
# test_completion_openai_litellm_key()
def test_completion_openrouter1():
try:
@ -540,6 +540,8 @@ def test_completion_openai_with_more_optional_params():
pytest.fail(f"Error occurred: {e}")
if type(response_str_2) != str:
pytest.fail(f"Error occurred: {e}")
except Timeout as e:
pass
except Exception as e:
pytest.fail(f"Error occurred: {e}")
@ -721,6 +723,7 @@ def test_completion_azure_with_litellm_key():
def test_completion_azure_deployment_id():
try:
litellm.set_verbose = True
response = completion(
deployment_id="chatgpt-v-2",
model="gpt-3.5-turbo",
@ -730,7 +733,7 @@ def test_completion_azure_deployment_id():
print(response)
except Exception as e:
pytest.fail(f"Error occurred: {e}")
# test_completion_azure_deployment_id()
test_completion_azure_deployment_id()
# Only works for local endpoint
# def test_completion_anthropic_openai_proxy():