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

fix(utils.py): support dynamic api key for azure_ai route

Browse source
This commit is contained in:
Krrish Dholakia 2024-06-11 17:51:29 -07:00
parent 6305d2dbcf
commit a9cbf2209c
2 changed files with 5 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/tests/test_completion.py
View file

@ -744,10 +744,7 @@ def test_completion_claude_3_function_plus_image():
@pytest.mark.parametrize(
"provider",
[
"azure",
"azure_ai"
],
["azure", "azure_ai"],
)
def test_completion_azure_mistral_large_function_calling(provider):
"""
@ -780,6 +777,7 @@ def test_completion_azure_mistral_large_function_calling(provider):
"content": "What's the weather like in Boston today in Fahrenheit?",
}
]
response = completion(
model="{}/mistral-large-latest".format(provider),
api_base=os.getenv("AZURE_MISTRAL_API_BASE"),