From 32ecc1a6775afc5973a46a053cd415845e0be6fc Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Mon, 4 Dec 2023 13:43:03 -0800 Subject: [PATCH] (feat) replicate/deployments: add POST Req view --- litellm/llms/replicate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/replicate.py b/litellm/llms/replicate.py index d639a8d1e4..0f8e23e2ad 100644 --- a/litellm/llms/replicate.py +++ b/litellm/llms/replicate.py @@ -100,7 +100,7 @@ def start_prediction(version_id, input_data, api_token, api_base, logging_obj, p logging_obj.pre_call( input=input_data["prompt"], api_key="", - additional_args={"complete_input_dict": initial_prediction_data, "headers": headers}, + additional_args={"complete_input_dict": initial_prediction_data, "headers": headers, "api_base": base_url}, ) response = requests.post(f"{base_url}/predictions", json=initial_prediction_data, headers=headers)