scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-06 02:48:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 103

Use CI_CD_DEFAULT_AZURE_MODEL env var in live azure CI tests

Browse source 
Replace the hardcoded azure/gpt-4.1-mini deployment name at the two live
azure call sites in test_custom_logger.py with the CI_CD_DEFAULT_AZURE_MODEL
env var, so the deployment can be updated from CircleCI project settings
instead of editing code. The mocked call and all router config fixtures are
left unchanged.
This commit is contained in:
ryan-crabbe-berri 2026-06-02 17:14:08 -07:00
parent f047b1571e
commit 33f80f3f77
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/local_testing/test_custom_logger.py
View file

@ -213,7 +213,9 @@ def test_async_custom_handler_stream():
async def test_1():
nonlocal complete_streaming_response
response = await litellm.acompletion(
model="azure/gpt-4.1-mini", messages=messages, stream=True
model=f"azure/{os.environ.get('CI_CD_DEFAULT_AZURE_MODEL', 'gpt-4.1-mini')}",
messages=messages,
stream=True,
)
async for chunk in response:
complete_streaming_response += (
@ -257,7 +259,9 @@ def test_azure_completion_stream():
complete_streaming_response = ""
response = litellm.completion(
model="azure/gpt-4.1-mini", messages=messages, stream=True
model=f"azure/{os.environ.get('CI_CD_DEFAULT_AZURE_MODEL', 'gpt-4.1-mini')}",
messages=messages,
stream=True,
)
for chunk in response:
complete_streaming_response += chunk["choices"][0]["delta"]["content"] or ""