From 10dd931eb72ea268426cdac9628ebc37d4e4e75b Mon Sep 17 00:00:00 2001
From: Pugazhendhi <pugazhendhi@unboundsecurity.ai>
Date: Wed, 5 Mar 2025 16:28:04 +0530
Subject: [PATCH] Adds unbound metadata to non streaming requests

---
 src/api/providers/unbound.ts | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/api/providers/unbound.ts b/src/api/providers/unbound.ts
index e3894f8718..a6c506abfc 100644
--- a/src/api/providers/unbound.ts
+++ b/src/api/providers/unbound.ts
@@ -147,6 +147,7 @@ export class UnboundHandler extends BaseProvider implements SingleCompletionHand
 
 	async completePrompt(prompt: string): Promise<string> {
 		try {
+			console.log("completePrompt running in Unbound: ", prompt)
 			const requestOptions: OpenAI.Chat.Completions.ChatCompletionCreateParamsNonStreaming = {
 				model: this.getModel().id.split("/")[1],
 				messages: [{ role: "user", content: prompt }],
@@ -157,7 +158,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) {