From 8cfdaed175bdc88d65b6320a0261535533f371ec Mon Sep 17 00:00:00 2001 From: Ariya Hidayat Date: Fri, 18 Oct 2024 10:17:04 -0700 Subject: [PATCH] Support Nebius AI Studio --- .github/workflows/test-nebius.yml | 24 ++++++++++++++++++++++++ README.md | 10 +++++++++- 2 files changed, 33 insertions(+), 1 deletion(-) create mode 100644 .github/workflows/test-nebius.yml diff --git a/.github/workflows/test-nebius.yml b/.github/workflows/test-nebius.yml new file mode 100644 index 0000000..7fd1a3f --- /dev/null +++ b/.github/workflows/test-nebius.yml @@ -0,0 +1,24 @@ +name: Test on Nebius + +on: + push: + branches: [main] + workflow_dispatch: + +jobs: + test: + runs-on: ubuntu-22.04 + timeout-minutes: 5 + steps: + - uses: actions/checkout@v4 + + - run: node --version + + - run: echo 'Which planet in our solar system is the largest?' | ./ask-llm.js | tee output.txt | grep -i jupiter + timeout-minutes: 3 + env: + LLM_API_BASE_URL: 'https://api.studio.nebius.ai/v1' + LLM_API_KEY: ${{ secrets.NEBIUS_API_KEY }} + LLM_CHAT_MODEL: 'meta-llama/Meta-Llama-3.1-8B-Instruct' + + - run: cat output.txt diff --git a/README.md b/README.md index 0c14b4e..a6f7a87 100644 --- a/README.md +++ b/README.md @@ -82,13 +82,14 @@ export LLM_API_BASE_URL=http://127.0.0.1:10002/v1 [![Test on Hyperbolic](https://github.com/ariya/ask-llm/actions/workflows/test-hyperbolic.yml/badge.svg)](https://github.com/ariya//ask-llm/actions/workflows/test-hyperbolic.yml) [![Test on Lepton](https://github.com/ariya/ask-llm/actions/workflows/test-lepton.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-lepton.yml) [![Test on Mistral](https://github.com/ariya/ask-llm/actions/workflows/test-mistral.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-mistral.yml) +[![Test on Nebius](https://github.com/ariya/ask-llm/actions/workflows/test-nebius.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-nebius.yml) [![Test on Novita](https://github.com/ariya/ask-llm/actions/workflows/test-novita.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-novita.yml) [![Test on Octo](https://github.com/ariya/ask-llm/actions/workflows/test-octo.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-octo.yml) [![Test on OpenAI](https://github.com/ariya/ask-llm/actions/workflows/test-openai.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-openai.yml) [![Test on OpenRouter](https://github.com/ariya/ask-llm/actions/workflows/test-openrouter.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-openrouter.yml) [![Test on Together](https://github.com/ariya/ask-llm/actions/workflows/test-together.yml/badge.svg)](https://github.com/ariya/ask-llm/actions/workflows/test-together.yml) -Supported LLM services include [AI21](https://studio.ai21.com), [Deep Infra](https://deepinfra.com), [DeepSeek](https://platform.deepseek.com/), [Fireworks](https://fireworks.ai), [Groq](https://groq.com), [Hyperbolic](https://www.hyperbolic.xyz), [Lepton](https://lepton.ai), [Mistral](https://console.mistral.ai), [Novita](https://novita.ai), [Octo](https://octo.ai), [OpenAI](https://platform.openai.com), [OpenRouter](https://openrouter.ai), and [Together](https://www.together.ai). +Supported LLM services include [AI21](https://studio.ai21.com), [Deep Infra](https://deepinfra.com), [DeepSeek](https://platform.deepseek.com/), [Fireworks](https://fireworks.ai), [Groq](https://groq.com), [Hyperbolic](https://www.hyperbolic.xyz), [Lepton](https://lepton.ai), [Mistral](https://console.mistral.ai), [Nebius](https://studio.nebius.ai), [Novita](https://novita.ai), [Octo](https://octo.ai), [OpenAI](https://platform.openai.com), [OpenRouter](https://openrouter.ai), and [Together](https://www.together.ai). For configuration specifics, refer to the relevant section. The quality of answers can vary based on the model's performance. @@ -148,6 +149,13 @@ export LLM_API_KEY="yourownapikey" export LLM_CHAT_MODEL="open-mistral-7b" ``` +* [Nebius](https://studio.nebius.ai) +```bash +export LLM_API_BASE_URL=https://api.studio.nebius.ai/v1 +export LLM_API_KEY="yourownapikey" +export LLM_CHAT_MODEL="meta-llama/Meta-Llama-3.1-8B-Instruct" +``` + * [Novita](https://novita.ai) ```bash export LLM_API_BASE_URL=https://api.novita.ai/v3/openai