scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-08-28 05:27:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1414 from websentry-ai/pm/add-unbound-metadata

Browse source 
Adds Unbound metadata for non-streaming requests
This commit is contained in:
Chris Estreich 2025-03-05 21:07:27 -08:00 • committed by GitHub
commit c9e21eb091
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 22 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/api/providers/__tests__/unbound.test.ts
View file

@ -192,6 +192,11 @@ describe("UnboundHandler", () => {
temperature: 0,
max_tokens: 8192,
}),
expect.objectContaining({
headers: expect.objectContaining({
"X-Unbound-Metadata": expect.stringContaining("roo-code"),
}),
}),
)
})
@ -233,6 +238,11 @@ describe("UnboundHandler", () => {
messages: [{ role: "user", content: "Test prompt" }],
temperature: 0,
}),
expect.objectContaining({
headers: expect.objectContaining({
"X-Unbound-Metadata": expect.stringContaining("roo-code"),
}),
}),
)
expect(mockCreate.mock.calls[0][0]).not.toHaveProperty("max_tokens")
})

13
src/api/providers/unbound.ts
View file

@ -157,7 +157,18 @@ export class UnboundHandler extends BaseProvider implements SingleCompletionHand
requestOptions.max_tokens = this.getModel().info.maxTokens
}
const response = await this.client.chat.completions.create(requestOptions)
const response = await this.client.chat.completions.create(requestOptions, {
headers: {
"X-Unbound-Metadata": JSON.stringify({
labels: [
{
key: "app",
value: "roo-code",
},
],
}),
},
})
return response.choices[0]?.message.content || ""
} catch (error) {
if (error instanceof Error) {