chore: adding x-title header and testing for litellm

Author: andrewshu2000

src/api/providers/fetchers/__tests__/litellm.spec.ts

@@ -4,6 +4,7 @@ vi.mock("axios")
 import type { Mock } from "vitest"
 import axios from "axios"
 import { getLiteLLMModels } from "../litellm"
+import { DEFAULT_HEADERS } from "../../constants"
 
 const mockedAxios = axios as typeof axios & {
 	get: Mock
@@ -32,7 +33,10 @@ describe("getLiteLLMModels", () => {
 			headers: {
 				Authorization: "Bearer test-api-key",
 				"Content-Type": "application/json",
+								...DEFAULT_HEADERS,
+
 			},
+			
 			timeout: 5000,
 		})
 	})
@@ -83,6 +87,8 @@ describe("getLiteLLMModels", () => {
 			headers: {
 				Authorization: "Bearer test-api-key",
 				"Content-Type": "application/json",
+								...DEFAULT_HEADERS,
+
 			},
 			timeout: 5000,
 		})
@@ -125,6 +131,8 @@ describe("getLiteLLMModels", () => {
 		expect(mockedAxios.get).toHaveBeenCalledWith("http://localhost:4000/v1/model/info", {
 			headers: {
 				"Content-Type": "application/json",
+								...DEFAULT_HEADERS,
+
 			},
 			timeout: 5000,
 		})

src/api/providers/fetchers/litellm.ts

@@ -4,6 +4,9 @@ import { LITELLM_COMPUTER_USE_MODELS } from "@roo-code/types"
 
 import type { ModelRecord } from "../../../shared/api"
 
+import { DEFAULT_HEADERS } from "../constants"
+
+
 /**
  * Fetches available models from a LiteLLM server
  *
@@ -16,6 +19,8 @@ export async function getLiteLLMModels(apiKey: string, baseUrl: string): Promise
 	try {
 		const headers: Record<string, string> = {
 			"Content-Type": "application/json",
+						...DEFAULT_HEADERS,
+
 		}
 
 		if (apiKey) {

src/api/providers/router-provider.ts

@@ -7,6 +7,9 @@ import { ApiHandlerOptions, RouterName, ModelRecord } from "../../shared/api"
 import { BaseProvider } from "./base-provider"
 import { getModels } from "./fetchers/modelCache"
 
+import { DEFAULT_HEADERS } from "./constants"
+
+
 type RouterProviderOptions = {
 	name: RouterName
 	baseURL: string
@@ -43,8 +46,14 @@ export abstract class RouterProvider extends BaseProvider {
 		this.defaultModelId = defaultModelId
 		this.defaultModelInfo = defaultModelInfo
 
-		this.client = new OpenAI({ baseURL, apiKey })
-	}
+		this.client = new OpenAI({
+			baseURL,
+			apiKey,
+			defaultHeaders: {
+				...DEFAULT_HEADERS,
+				...(options.openAiHeaders || {})
+			}
+		})	}
 
 	public async fetchModel() {
 		this.models = await getModels({ provider: this.name, apiKey: this.client.apiKey, baseUrl: this.client.baseURL })