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

(router) better debugging using config.yaml

Browse source
This commit is contained in:
ishaan-jaff 2023-12-05 18:07:27 -08:00
parent 9693e5ede6
commit cc2bb5daa6
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/router.py
View file

@ -889,7 +889,7 @@ class Router:
if "azure" in model_name: if "azure" in model_name:
if api_base is None: if api_base is None:
raise ValueError("api_base is required for Azure OpenAI. Set it on your config") raise ValueError("api_base is required for Azure OpenAI. Set it on your config")
self.print_verbose(f"Initializing Azure OpenAI Client for {model_name}, {str(api_base)}, {api_key}") self.print_verbose(f"Initializing Azure OpenAI Client for {model_name}, Api Base: {str(api_base)}, Api Key:{api_key}")
if api_version is None: if api_version is None:
api_version = "2023-07-01-preview" api_version = "2023-07-01-preview"
if "gateway.ai.cloudflare.com" in api_base: if "gateway.ai.cloudflare.com" in api_base:
@ -960,7 +960,7 @@ class Router:
) )
else: else:
self.print_verbose(f"Initializing OpenAI Client for {model_name}, {str(api_base)}, {api_key}") self.print_verbose(f"Initializing OpenAI Client for {model_name}, Api Base:{str(api_base)}, Api Key:{api_key}")
model["async_client"] = openai.AsyncOpenAI( model["async_client"] = openai.AsyncOpenAI(
api_key=api_key, api_key=api_key,
base_url=api_base, base_url=api_base,