-
Notifications
You must be signed in to change notification settings - Fork 2.6k
Make Databricks models streaming in OpenAI Compatible provider #2495
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Conversation
|
src/api/providers/openai.ts
Outdated
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Consider extracting the (isGrokXAI || isDatabricksAI) check into a helper function for clarity and maintainability if similar conditions are repeated in the future.
|
Should we make azure profiler for this ? |
e05d769 to
784bded
Compare
7d086fe to
abcf613
Compare
@samhvw8 Databricks is hosted on multiple cloud platforms. They have the same API usage. So I don't think we need to create the Azure profiler for it. How do you think? |
* Added Sambanova Deepseek-V3-0324 * Added changeset
|
Thanks for your contribution. We genuinely apologize the delay on reviewing your PR. We're closing this PR because we'd prefer have an issue for this since we now shifted to a clearer issue-first workflow to avoid this kind of situation going forward. Please open a new issue if you'd like to revisit this PR. We truly appreciate your patience and would be happy to have you continue contributing. |
Context
update the condition to detect the Databrick Azure provider in OpenAI Compatible provider config to make it stream the response.

WHY: The stream option does not work for the Databricks model
Implementation
Based on #2449 , add condition isDatabricksAI
Screenshots
How to Test
Configure Databricks Azure API endpoint in the Open AI Compatible provider and enable streaming support.
After the fix streaming works with models are provided by Databricks can stream as expected.
I didn't try on Databrick instance on AWS.
Get in Touch
Important
Add support for streaming Databricks models in OpenAI Compatible provider by excluding
stream_optionsfor Databricks AI.OpenAiHandlerinopenai.tsto support streaming for Databricks models by excludingstream_optionswhenisDatabricksAIis true._isDatabricksAI()method to detect Databricks AI based onopenAiBaseUrl.openai.test.tsto verify Databricks AI configuration and ensurestream_optionsis excluded during streaming.This description was created by
for e05d7696193a1833d10c11898d328c7d7dbe5bd8. It will automatically update as commits are pushed.