diff --git a/tests/llm_translation/conftest.py b/tests/llm_translation/conftest.py index 46a3a31771..e258bafc01 100644 --- a/tests/llm_translation/conftest.py +++ b/tests/llm_translation/conftest.py @@ -30,6 +30,9 @@ _SCALAR_DEFAULTS = { "force_ipv4": getattr(litellm, "force_ipv4", False), "drop_params": getattr(litellm, "drop_params", None), "modify_params": getattr(litellm, "modify_params", False), + "api_base": getattr(litellm, "api_base", None), + "api_key": getattr(litellm, "api_key", None), + "cohere_key": getattr(litellm, "cohere_key", None), } diff --git a/tests/local_testing/conftest.py b/tests/local_testing/conftest.py index a6b31caff2..0013f25357 100644 --- a/tests/local_testing/conftest.py +++ b/tests/local_testing/conftest.py @@ -43,6 +43,8 @@ _SCALAR_DEFAULTS = { "force_ipv4": getattr(litellm, "force_ipv4", False), "drop_params": getattr(litellm, "drop_params", None), "modify_params": getattr(litellm, "modify_params", False), + "api_base": getattr(litellm, "api_base", None), + "api_key": getattr(litellm, "api_key", None), }