diff --git a/litellm/llms/huggingface_restapi.py b/litellm/llms/huggingface_restapi.py index 7db1b16a1e..2511068286 100644 --- a/litellm/llms/huggingface_restapi.py +++ b/litellm/llms/huggingface_restapi.py @@ -255,7 +255,7 @@ def completion( logging_obj.pre_call( input=input_text, api_key=api_key, - additional_args={"complete_input_dict": data, "task": task, "headers": headers}, + additional_args={"complete_input_dict": data, "task": task, "headers": headers, "api_base": completion_url}, ) ## COMPLETION CALL if "stream" in optional_params and optional_params["stream"] == True: