feat(utils.py): fix openai-like streaming

2025-04-26 11:14:04 +00:00 · 2024-07-27 15:32:57 -07:00 · 2024-07-27 15:32:57 -07:00 · 56ba0c62f3
commit 56ba0c62f3
parent cfc061e38e
4 changed files with 20 additions and 7 deletions
--- a/litellm/main.py
+++ b/litellm/main.py
@ -1867,6 +1867,7 @@ def completion(
                    custom_prompt_dict=custom_prompt_dict,
                    client=client,  # pass AsyncOpenAI, OpenAI client
                    encoding=encoding,
+                    custom_llm_provider="databricks",
                )
            except Exception as e:
                ## LOGGING - log the original exception returned