fix(predibase.py): fix async completion call

2024-05-09 18:44:19 -07:00 · 2024-05-09 18:44:19 -07:00 · 491e177348
commit 491e177348
parent 425efc60f4
1 changed files with 1 additions and 1 deletions
--- 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,