From 491e17734856fbd28260bbd3222b3ca9c49a86e0 Mon Sep 17 00:00:00 2001
From: Krrish Dholakia <krrishdholakia@gmail.com>
Date: Thu, 9 May 2024 18:44:19 -0700
Subject: [PATCH] fix(predibase.py): fix async completion call

---
 litellm/llms/predibase.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/litellm/llms/predibase.py b/litellm/llms/predibase.py
index ef9c6b0ba53..1f96e7c67e9 100644
--- a/litellm/llms/predibase.py
+++ b/litellm/llms/predibase.py
@@ -390,7 +390,7 @@ class PredibaseChatCompletion(BaseLLM):
                     model=model,
                     messages=messages,
                     data=data,
-                    api_base=api_base,
+                    api_base=completion_url,
                     model_response=model_response,
                     print_verbose=print_verbose,
                     encoding=encoding,