clone/opencode
1
0
Fork 0
mirror of https://github.com/anomalyco/opencode.git synced 2026-05-17 18:12:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(provider): add NVIDIA endpoints origin header (#27394)

Browse Source
This commit is contained in:
nv-kasikritc
2026-05-14 15:21:58 -07:00
committed by GitHub
parent 43310f4d8c
commit d34a0194ec
4 changed files with 151 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
packages/opencode/src/provider/provider.ts
View File

@@ -427,13 +427,14 @@ function custom(dep: CustomDep): Record<string, CustomLoader> {
},
},
}),
nvidia: () =>
nvidia: (provider) =>
Effect.succeed({
autoload: false,
autoload: provider.source === "config",
options: {
headers: {
"HTTP-Referer": "https://opencode.ai/",
"X-Title": "opencode",
"X-BILLING-INVOKE-ORIGIN": "OpenCode",
},
},
}),

94
packages/opencode/test/provider/provider.test.ts
View File

@@ -1868,6 +1868,100 @@ test("provider options are deeply merged", async () => {
})
})
test("hosted nvidia provider adds billing origin header", async () => {
await using tmp = await tmpdir({
init: async (dir) => {
await Bun.write(
path.join(dir, "opencode.json"),
JSON.stringify({
$schema: "https://opencode.ai/config.json",
provider: {
nvidia: {
options: {
apiKey: "test-api-key",
},
},
},
}),
)
},
})
await WithInstance.provide({
directory: tmp.path,
fn: async () => {
const providers = await list()
expect(providers[ProviderID.make("nvidia")].options.headers).toEqual({
"HTTP-Referer": "https://opencode.ai/",
"X-Title": "opencode",
"X-BILLING-INVOKE-ORIGIN": "OpenCode",
})
},
})
})
test("custom nvidia baseURL adds billing origin header", async () => {
await using tmp = await tmpdir({
init: async (dir) => {
await Bun.write(
path.join(dir, "opencode.json"),
JSON.stringify({
$schema: "https://opencode.ai/config.json",
provider: {
nvidia: {
options: {
apiKey: "test-api-key",
baseURL: "http://localhost:8000/v1",
},
},
},
}),
)
},
})
await WithInstance.provide({
directory: tmp.path,
fn: async () => {
const providers = await list()
expect(providers[ProviderID.make("nvidia")].options.headers).toEqual({
"HTTP-Referer": "https://opencode.ai/",
"X-Title": "opencode",
"X-BILLING-INVOKE-ORIGIN": "OpenCode",
})
},
})
})
test("explicit nvidia billing origin header is preserved", async () => {
await using tmp = await tmpdir({
init: async (dir) => {
await Bun.write(
path.join(dir, "opencode.json"),
JSON.stringify({
$schema: "https://opencode.ai/config.json",
provider: {
nvidia: {
options: {
apiKey: "test-api-key",
baseURL: "http://localhost:8000/v1",
headers: {
"X-BILLING-INVOKE-ORIGIN": "CustomOrigin",
},
},
},
},
}),
)
},
})
await WithInstance.provide({
directory: tmp.path,
fn: async () => {
const providers = await list()
expect(providers[ProviderID.make("nvidia")].options.headers["X-BILLING-INVOKE-ORIGIN"]).toBe("CustomOrigin")
},
})
})
test("custom model inherits npm package from models.dev provider config", async () => {
await using tmp = await tmpdir({
init: async (dir) => {