Просмотр исходного кода

feat(provider): add NVIDIA to popular providers, docs, and attribution headers (#22927)

Co-authored-by: Kit Langton <kit.langton@gmail.com>
Co-authored-by: Aiden Cline <63023139+rekram1-node@users.noreply.github.com>
Annie Surla 4 месяцев назад
Родитель
Сommit
6e0178655b
2 измененных файлов с 54 добавлено и 17 удалено
  1. 10 0
      packages/opencode/src/provider/provider.ts
  2. 44 17
      packages/web/src/content/docs/providers.mdx

+ 10 - 0
packages/opencode/src/provider/provider.ts

@@ -410,6 +410,16 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
           },
         },
       }),
+    nvidia: () =>
+      Effect.succeed({
+        autoload: false,
+        options: {
+          headers: {
+            "HTTP-Referer": "https://opencode.ai/",
+            "X-Title": "opencode",
+          },
+        },
+      }),
     vercel: () =>
       Effect.succeed({
         autoload: false,

+ 44 - 17
packages/web/src/content/docs/providers.mdx

@@ -1316,30 +1316,57 @@ To use Kimi K2 from Moonshot AI:
 
 ---
 
-### Mistral AI
+### NVIDIA
 
-1.  Head over to the [Mistral AI console](https://console.mistral.ai/), create an account, and generate an API key.
+NVIDIA provides access to Nemotron models and many other open models through [build.nvidia.com](https://build.nvidia.com) for free.
 
-2.  Run the `/connect` command and search for **Mistral AI**.
+1. Head over to [build.nvidia.com](https://build.nvidia.com), create an account, and generate an API key.
 
-    ```txt
-    /connect
-    ```
+2. Run the `/connect` command and search for **NVIDIA**.
 
-3.  Enter your Mistral API key.
+   ```txt
+   /connect
+   ```
+
+3. Enter your NVIDIA API key.
+
+   ```txt
+   ┌ API key
+   │
+   │
+   └ enter
+   ```
+
+4. Run the `/models` command to select a model like nemotron-3-super-120b-a12b.
+
+   ```txt
+   /models
+   ```
 
-    ```txt
-    ┌ API key
-    │
-    │
-    └ enter
-    ```
+#### On-Prem / NIM
 
-4.  Run the `/models` command to select a model like _Mistral Medium_.
+You can also use NVIDIA models locally via [NVIDIA NIM](https://docs.nvidia.com/nim/) by setting a custom base URL.
 
-    ```txt
-    /models
-    ```
+```json title="opencode.json" {6}
+{
+  "$schema": "https://opencode.ai/config.json",
+  "provider": {
+    "nvidia": {
+      "options": {
+        "baseURL": "http://localhost:8000/v1"
+      }
+    }
+  }
+}
+```
+
+#### Environment Variable
+
+Alternatively, set your API key as an environment variable.
+
+```bash frame="none"
+export NVIDIA_API_KEY=nvapi-your-key-here
+```
 
 ---