Fixes from review

Author: jr

src/api/providers/fetchers/litellm.ts

@@ -1,5 +1,5 @@
 import axios from "axios"
-import { ModelRecord } from "../../../shared/api"
+import { COMPUTER_USE_MODELS, ModelRecord } from "../../../shared/api"
 
 /**
  * Fetches available models from a LiteLLM server
@@ -21,19 +21,25 @@ export async function getLiteLLMModels(apiKey: string, baseUrl: string): Promise
 		const response = await axios.get(`${baseUrl}/v1/model/info`, { headers })
 		const models: ModelRecord = {}
 
+		const computerModels = Array.from(COMPUTER_USE_MODELS)
+
 		// Process the model info from the response
 		if (response.data && response.data.data && Array.isArray(response.data.data)) {
 			for (const model of response.data.data) {
 				const modelName = model.model_name
 				const modelInfo = model.model_info
+				const litellmModelName = model?.litellm_params?.model as string | undefined
 
-				if (!modelName || !modelInfo) continue
+				if (!modelName || !modelInfo || !litellmModelName) continue
 
 				models[modelName] = {
 					maxTokens: modelInfo.max_tokens || 8192,
 					contextWindow: modelInfo.max_input_tokens || 200000,
 					supportsImages: Boolean(modelInfo.supports_vision),
-					supportsComputerUse: Boolean(modelInfo.supports_function_calling || modelInfo.supports_tool_choice),
+					// litellm_params.model may have a prefix like openrouter/
+					supportsComputerUse: computerModels.some((computer_model) =>
+						litellmModelName.endsWith(computer_model),
+					),
 					supportsPromptCache: Boolean(modelInfo.supports_prompt_caching),
 					inputPrice: modelInfo.input_cost_per_token ? modelInfo.input_cost_per_token * 1000000 : undefined,
 					outputPrice: modelInfo.output_cost_per_token

src/api/providers/router-provider.ts

@@ -44,15 +44,7 @@ export abstract class RouterProvider extends BaseProvider {
 	}
 
 	public async fetchModel() {
-		// For LiteLLM, we need to pass the API key and base URL
-		if (this.name === "litellm") {
-			const apiKey = this.client.apiKey
-			const baseUrl = this.client.baseURL
-			this.models = await getModels(this.name, apiKey, baseUrl)
-		} else {
-			// For other routers, just pass the API key
-			this.models = await getModels(this.name, this.client.apiKey)
-		}
+		this.models = await getModels(this.name, this.client.apiKey, this.client.baseURL)
 		return this.getModel()
 	}