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

2025-04-25 18:54:30 +00:00 · 2024-07-27 15:32:57 -07:00 · 2024-07-27 15:32:57 -07:00 · c85ed01756
commit c85ed01756
parent 3c77f39751
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