Skip to content

Conversation

@antoncp
Copy link
Contributor

@antoncp antoncp commented Oct 9, 2025

This PR introduces a new provider Nebius AI Studio, one of the largest AI providers in Europe (headquarters in Amsterdam).

The NebiusProvider enables using models hosted on Nebius AI Studio through the familiar pydantic-ai interface. It integrates with the existing OpenAIChatModel class and supports all core features.

Key changes

  • Added NebiusProvider class (mirroring other provider implementations)
  • Added tests for the new provider
  • Updated documentation and environment variable details

Notes

NebiusProvider uses the default NEBIUS_API_KEY environment variable for authentication, or it can be passed explicitly during initialization.

@antoncp antoncp force-pushed the add-nebius-ai-provider branch 2 times, most recently from a237bf6 to 2d146bb Compare October 9, 2025 10:09
@DouweM DouweM self-assigned this Oct 10, 2025
'deepseek-ai': deepseek_model_profile,
'qwen': qwen_model_profile,
'google': google_model_profile,
'openai': openai_model_profile,
Copy link
Collaborator

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

Copy link
Contributor Author

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,
}
Copy link
Collaborator

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?

Copy link
Contributor Author

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/

@antoncp antoncp force-pushed the add-nebius-ai-provider branch from a8c4ce3 to 5640674 Compare October 13, 2025 08:54
@antoncp antoncp requested a review from DouweM October 13, 2025 09:03
@antoncp antoncp force-pushed the add-nebius-ai-provider branch from 5640674 to d3a63de Compare October 13, 2025 11:30
@antoncp antoncp requested a review from DouweM October 13, 2025 11:32
@DouweM DouweM merged commit e119392 into pydantic:main Oct 13, 2025
32 checks passed
@DouweM
Copy link
Collaborator

DouweM commented Oct 13, 2025

@antoncp Thanks Anton!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants