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.
Hey team,
Thanks for a great framework! Adding integration with Nebius AI Studio, which provides access to state-of-the-art models.