diff --git a/litellm/llms/palm.py b/litellm/llms/palm.py index ee1fc19bb1..266f7611ee 100644 --- a/litellm/llms/palm.py +++ b/litellm/llms/palm.py @@ -19,8 +19,8 @@ def completion( model: str, messages: list, model_response: ModelResponse, - api_key: str, print_verbose: Callable, + api_key, encoding, logging_obj, optional_params=None, diff --git a/litellm/main.py b/litellm/main.py index a17a8a707a..3779df87c3 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -794,7 +794,7 @@ def completion( return response response = model_response elif custom_llm_provider == "palm": - api_key = ( + palm_api_key = ( api_key or get_secret("PALM_API_KEY") or litellm.api_key @@ -809,7 +809,7 @@ def completion( litellm_params=litellm_params, logger_fn=logger_fn, encoding=encoding, - api_key=api_key, + api_key=palm_api_key, logging_obj=logging ) if "stream_tokens" in optional_params and optional_params["stream_tokens"] == True: