-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Add Nebius AI Studio provider support #3124
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
a237bf6 to
2d146bb
Compare
| 'deepseek-ai': deepseek_model_profile, | ||
| 'qwen': qwen_model_profile, | ||
| 'google': google_model_profile, | ||
| 'openai': openai_model_profile, |
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.
If this is just gpt-oss, this should be harmony_model_profile
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.
Thanks for the tip on how to do it correctly. Changed.
| 'openai': openai_model_profile, | ||
| 'mistralai': mistral_model_profile, | ||
| 'moonshotai': moonshotai_model_profile, | ||
| } |
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.
Are these all of the providers Nebius supports?
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.
Yes, all currently supported Nebius providers are included here. They are listed at https://studio.nebius.com/
2d146bb to
a8c4ce3
Compare
a8c4ce3 to
5640674
Compare
5640674 to
d3a63de
Compare
|
@antoncp Thanks Anton! |
This PR introduces a new provider Nebius AI Studio, one of the largest AI providers in Europe (headquarters in Amsterdam).
The
NebiusProviderenables using models hosted on Nebius AI Studio through the familiarpydantic-aiinterface. It integrates with the existingOpenAIChatModelclass and supports all core features.Key changes
NebiusProviderclass (mirroring other provider implementations)Notes
NebiusProvideruses the defaultNEBIUS_API_KEYenvironment variable for authentication, or it can be passed explicitly during initialization.