Integration with Nebius AI Studio added #138
Conversation
|
@ksolo may I kindly ask you to review it before it diverges too much from the main? thanks! |
|
|
||
| provider = NebiusProvider() | ||
| print(provider.api_key) | ||
| response = provider.chat_completions_create(model=selected_model, messages=message_history, top_p=chosen_top_p) |
There was a problem hiding this comment.
@Aktsvigun We have been mocking the interaction with the providers. If you look at an example like this one (https://github.com/andrewyng/aisuite/blob/main/tests/providers/test_sambanova_provider.py) we are mostly testing the setup more so than the actual result.
Can you update this to follow that pattern?
There was a problem hiding this comment.
Sure, fixed! Kindly check.
I've standardized the test according to the example you sent & modified the NebiusProvider to match most of other providers (e.g. Mistral, Sambanova, etc.)
examples/client.ipynb
Outdated
There was a problem hiding this comment.
I recommend resetting the notebook so the outputs are all clean when committing these.
There was a problem hiding this comment.
Could you please specify what outputs you reference? In this notebook (examples/client.ipynb) I can see no outputs.
|
@ksolo could you please approve? Fixed all your suggestions. |
|
@Aktsvigun apologies on the delay I was traveling for the holidays. |
|
@Aktsvigun Can you open a follow up PR with documentation? Examples can be found here (https://github.com/andrewyng/aisuite/tree/main/guides) |
|
Hi @ksolo, thank you! Sure, will do. |
Hey team,
Thanks for a great framework! Adding integration with Nebius AI Studio, which provides access to state-of-the-art models.