diff --git a/litellm/llms/cohere.py b/litellm/llms/cohere.py index 4f97ef0c08..001c6ba269 100644 --- a/litellm/llms/cohere.py +++ b/litellm/llms/cohere.py @@ -188,7 +188,7 @@ def completion( def embedding( model: str, input: list, - api_key: str, + api_key: Optional[str] = None, logging_obj=None, model_response=None, encoding=None, diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py index 6aba450c35..93c6a9a089 100644 --- a/litellm/llms/huggingface_restapi.py +++ b/litellm/llms/huggingface_restapi.py @@ -324,8 +324,8 @@ def completion( def embedding( model: str, input: list, - api_key: str, - api_base: str, + api_key: Optional[str] = None, + api_base: Optional[str] = None, logging_obj=None, model_response=None, encoding=None,