Added working

Author: nang-dev

src/api/providers/pearai.ts

@@ -1,29 +1,17 @@
-import { OpenAiHandler } from "./openai"
 import * as vscode from "vscode"
-import { ApiHandlerOptions, PEARAI_URL } from "../../shared/api"
+import { ApiHandlerOptions, PEARAI_URL, AnthropicModelId, ModelInfo } from "../../shared/api"
 import { AnthropicHandler } from "./anthropic"
 
 export class PearAiHandler extends AnthropicHandler {
 	constructor(options: ApiHandlerOptions) {
-		if (options.pearaiModelInfo) {
-			options.pearaiModelInfo = {
-				...options.pearaiModelInfo,
-				inputPrice: options.pearaiModelInfo.inputPrice ? options.pearaiModelInfo.inputPrice * 1.025 : 0,
-				outputPrice: options.pearaiModelInfo.outputPrice ? options.pearaiModelInfo.outputPrice * 1.025 : 0,
-			}
-		}
-
 		if (!options.pearaiApiKey) {
 			vscode.window.showErrorMessage("PearAI API key not found.", "Login to PearAI").then(async (selection) => {
 				if (selection === "Login to PearAI") {
 					const extensionUrl = `${vscode.env.uriScheme}://pearai.pearai/auth`
 					const callbackUri = await vscode.env.asExternalUri(vscode.Uri.parse(extensionUrl))
-
 					vscode.env.openExternal(
 						await vscode.env.asExternalUri(
-							vscode.Uri.parse(
-								`https://trypear.ai/signin?callback=${callbackUri.toString()}`, // Change to localhost if running locally
-							),
+							vscode.Uri.parse(`https://trypear.ai/signin?callback=${callbackUri.toString()}`),
 						),
 					)
 				}
@@ -36,4 +24,18 @@ export class PearAiHandler extends AnthropicHandler {
 			anthropicBaseUrl: PEARAI_URL,
 		})
 	}
+
+	override getModel(): { id: AnthropicModelId; info: ModelInfo } {
+		const baseModel = super.getModel()
+		return {
+			id: baseModel.id,
+			info: {
+				...baseModel.info,
+				inputPrice: (baseModel.info.inputPrice || 0) * 1.025,
+				outputPrice: (baseModel.info.outputPrice || 0) * 1.025,
+				cacheWritesPrice: baseModel.info.cacheWritesPrice ? baseModel.info.cacheWritesPrice * 1.025 : undefined,
+				cacheReadsPrice: baseModel.info.cacheReadsPrice ? baseModel.info.cacheReadsPrice * 1.025 : undefined,
+			},
+		}
+	}
 }

webview-ui/src/components/settings/ApiOptions.tsx

@@ -1633,15 +1633,15 @@ export function normalizeApiConfiguration(apiConfiguration?: ApiConfiguration) {
 			// Create PearAI model info with 1.03x price multiplier
 			const pearaiModelInfo: ModelInfo = {
 				...baseModelInfo,
-				inputPrice: baseModelInfo.inputPrice * 1.03,
-				outputPrice: baseModelInfo.outputPrice * 1.03,
-				cacheWritesPrice: baseModelInfo.cacheWritesPrice ? baseModelInfo.cacheWritesPrice * 1.03 : undefined,
-				cacheReadsPrice: baseModelInfo.cacheReadsPrice ? baseModelInfo.cacheReadsPrice * 1.03 : undefined,
+				inputPrice: baseModelInfo.inputPrice,
+				outputPrice: baseModelInfo.outputPrice,
+				cacheWritesPrice: baseModelInfo.cacheWritesPrice ? baseModelInfo.cacheWritesPrice : undefined,
+				cacheReadsPrice: baseModelInfo.cacheWritesPrice ? baseModelInfo.cacheReadsPrice : undefined,
 			}
 			return {
 				selectedProvider: provider,
-				selectedModelId: apiConfiguration?.pearaiModelId || anthropicDefaultModelId,
-				selectedModelInfo: apiConfiguration?.pearaiModelInfo || pearaiModelInfo,
+				selectedModelId: anthropicDefaultModelId,
+				selectedModelInfo: pearaiModelInfo,
 			}
 		}
 		case "unbound":