argilla-io / distilabel

Distilabel is a framework for synthetic data and AI feedback for engineers who need fast, reliable and scalable pipelines based on verified research papers.
https://distilabel.argilla.io
Apache License 2.0
1.57k stars 123 forks source link

[FEATURE] Add `O1AILLM` #663

Open gabrielmbmb opened 5 months ago

gabrielmbmb commented 5 months ago

Is your feature request related to a problem? Please describe. 01.AI has recently released their API platform which allows access to models such as Yi-Large, currently in the top 10 in LMSYS Chatbot Arena.

Describe the solution you'd like Add a new O1AILLM which allows using their API platform as an LLM.

Additional context The platform is not fully open yet and you need to join a waitlist: https://platform.01.ai/contactform

Josephrp commented 3 months ago

this is cool , i'll surely do it :-)

anything you might want to point me to as reference , clients , and more ?

Josephrp commented 3 months ago

linking this pull request (WiP) : https://github.com/argilla-io/distilabel/pull/793