Pass task ID in the Roo provider request headers (#7303)

Author: cte

src/api/providers/__tests__/chutes.spec.ts

@@ -398,6 +398,7 @@ describe("ChutesHandler", () => {
 				stream: true,
 				stream_options: { include_usage: true },
 			}),
+			undefined,
 		)
 	})
 

src/api/providers/__tests__/fireworks.spec.ts

@@ -356,6 +356,7 @@ describe("FireworksHandler", () => {
 				stream: true,
 				stream_options: { include_usage: true },
 			}),
+			undefined,
 		)
 	})
 

src/api/providers/__tests__/groq.spec.ts

@@ -145,6 +145,7 @@ describe("GroqHandler", () => {
 				stream: true,
 				stream_options: { include_usage: true },
 			}),
+			undefined,
 		)
 	})
 
@@ -178,6 +179,7 @@ describe("GroqHandler", () => {
 				messages: expect.arrayContaining([{ role: "system", content: systemPrompt }]),
 				stream: true,
 			}),
+			undefined,
 		)
 
 		// Verify temperature is NOT included
@@ -216,6 +218,7 @@ describe("GroqHandler", () => {
 				messages: expect.arrayContaining([{ role: "system", content: systemPrompt }]),
 				stream: true,
 			}),
+			undefined,
 		)
 	})
 })

src/api/providers/__tests__/roo.spec.ts

@@ -261,6 +261,7 @@ describe("RooHandler", () => {
 						expect.objectContaining({ role: "user", content: "Second message" }),
 					]),
 				}),
+				undefined,
 			)
 		})
 	})
@@ -365,6 +366,7 @@ describe("RooHandler", () => {
 				expect.not.objectContaining({
 					temperature: expect.anything(),
 				}),
+				undefined,
 			)
 		})
 
@@ -382,6 +384,7 @@ describe("RooHandler", () => {
 				expect.objectContaining({
 					temperature: 0.9,
 				}),
+				undefined,
 			)
 		})
 

src/api/providers/__tests__/sambanova.spec.ts

@@ -148,6 +148,7 @@ describe("SambaNovaHandler", () => {
 				stream: true,
 				stream_options: { include_usage: true },
 			}),
+			undefined,
 		)
 	})
 })

src/api/providers/__tests__/zai.spec.ts

@@ -224,6 +224,7 @@ describe("ZAiHandler", () => {
 					stream: true,
 					stream_options: { include_usage: true },
 				}),
+				undefined,
 			)
 		})
 	})

src/api/providers/base-openai-compatible-provider.ts

@@ -66,6 +66,7 @@ export abstract class BaseOpenAiCompatibleProvider<ModelName extends string>
 		systemPrompt: string,
 		messages: Anthropic.Messages.MessageParam[],
 		metadata?: ApiHandlerCreateMessageMetadata,
+		requestOptions?: OpenAI.RequestOptions,
 	) {
 		const {
 			id: model,
@@ -85,7 +86,7 @@ export abstract class BaseOpenAiCompatibleProvider<ModelName extends string>
 			params.temperature = this.options.modelTemperature
 		}
 
-		return this.client.chat.completions.create(params)
+		return this.client.chat.completions.create(params, requestOptions)
 	}
 
 	override async *createMessage(

src/api/providers/roo.ts

@@ -36,7 +36,12 @@ export class RooHandler extends BaseOpenAiCompatibleProvider<RooModelId> {
 		messages: Anthropic.Messages.MessageParam[],
 		metadata?: ApiHandlerCreateMessageMetadata,
 	): ApiStream {
-		const stream = await this.createStream(systemPrompt, messages, metadata)
+		const stream = await this.createStream(
+			systemPrompt,
+			messages,
+			metadata,
+			metadata?.taskId ? { headers: { "X-Roo-Task-ID": metadata.taskId } } : undefined,
+		)
 
 		for await (const chunk of stream) {
 			const delta = chunk.choices[0]?.delta