Merge pull request #678 from shivaylamba/nebius-provider

Nebius provider

Author: dgageot

README.md

@@ -100,7 +100,7 @@ See [the USAGE docs](./docs/USAGE.md#tool-configuration) for more detailed infor
 - **🔄 Smart delegation** - Agents can automatically route tasks to the most suitable specialist.
 - **📝 YAML configuration** - Declarative model and agent configuration.
 - **💭 Advanced reasoning** - Built-in "think", "todo" and "memory" tools for complex problem-solving.
-- **🌐 Multiple AI providers** - Support for OpenAI, Anthropic, Gemini, xAI and [Docker Model Runner](https://docs.docker.com/ai/model-runner/).
+- **🌐 Multiple AI providers** - Support for OpenAI, Anthropic, Gemini, xA, Nebius and [Docker Model Runner](https://docs.docker.com/ai/model-runner/).
 
 ## 🚀 Quick Start 🚀
 
@@ -145,6 +145,9 @@ export GOOGLE_API_KEY=your_api_key_here
 
 # For xAI models
 export XAI_API_KEY=your_api_key_here
+
+# For Nebius models
+export NEBIUS_API_KEY=your_api_key_here
 ```
 
 ### Run Agents!

examples/nebius.yaml

@@ -0,0 +1,17 @@
+version: "2"
+
+agents:
+  root:
+    model: grok-model
+    description: A helpful AI assistant powered by Nebius
+    instruction: |
+      You are a LLM model, a helpful and maximally truthful AI built by Moonshotai.
+      Be helpful, accurate, and concise in your responses.
+      You are not based on any other companies and their models.
+
+models:
+  grok-model:
+    provider: nebius
+    model: moonshotai/Kimi-K2-Instruct
+    max_tokens: 110000
+    temperature: 0.7

pkg/model/provider/provider.go

@@ -40,6 +40,11 @@ var ProviderAliases = map[string]Alias{
 		BaseURL:     "https://api.x.ai/v1",
 		TokenEnvVar: "XAI_API_KEY",
 	},
+	"nebius": {
+		APIType:     "openai",
+		BaseURL:     "https://api.studio.nebius.com/v1",
+		TokenEnvVar: "NEBIUS_API_KEY",
+	},
 }
 
 // Provider defines the interface for model providers