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

use api_base instead of custom_api_base

Browse source
This commit is contained in:
ishaan-jaff 2023-09-02 17:11:30 -07:00
parent bd0a118552
commit 4a994dc498
9 changed files with 39 additions and 39 deletions
Download patch file Download diff file Expand all files Collapse all files

6
litellm/testing.py
View file

@ -31,9 +31,9 @@ def testing_batch_completion(*args, **kwargs):
if isinstance(model, dict) and "custom_llm_provider" in model
else None
)
kwargs_modified["custom_api_base"] = (
model["custom_api_base"]
if isinstance(model, dict) and "custom_api_base" in model
kwargs_modified["api_base"] = (
model["api_base"]
if isinstance(model, dict) and "api_base" in model
else None
)
for message_list in batch_messages: