pearaiAgentModels-null-fix

Author: Himanshu-Singh-Chauhan

src/api/providers/pearai/pearai.ts

@@ -11,7 +11,7 @@ import { OpenAiHandler } from "../openai"
 import { PearAIGenericHandler } from "./pearaiGeneric"
 import { PEARAI_URL } from "../../../shared/pearaiApi"
 
-interface PearAiModelsResponse {
+export interface PearAIAgentModelsConfig {
 	models: {
 		[key: string]: {
 			underlyingModel?: { [key: string]: any }
@@ -23,7 +23,7 @@ interface PearAiModelsResponse {
 
 export class PearAiHandler extends BaseProvider implements SingleCompletionHandler {
 	private handler!: AnthropicHandler | PearAIGenericHandler
-	private pearAiModelsResponse: PearAiModelsResponse | null = null
+	private pearAIAgentModels: PearAIAgentModelsConfig | null = null
 	private options: ApiHandlerOptions
 
 	constructor(options: ApiHandlerOptions) {
@@ -64,15 +64,17 @@ export class PearAiHandler extends BaseProvider implements SingleCompletionHandl
 
 		if (modelId.startsWith("pearai")) {
 			try {
-				const response = await fetch(`${PEARAI_URL}/getPearAIAgentModels`)
-				if (!response.ok) {
-					throw new Error(`Failed to fetch models: ${response.statusText}`)
+				if (!options.pearaiAgentModels) {
+					console.log("NOT AMAZING")
+					throw new Error("PearAI models not found")
+				} else {
+					console.log("AMAZING")
 				}
-				const data = (await response.json()) as PearAiModelsResponse
-				this.pearAiModelsResponse = data
+				const pearaiAgentModels = options.pearaiAgentModels
+				this.pearAIAgentModels = pearaiAgentModels
 				const underlyingModel =
-					data.models[modelId]?.underlyingModelUpdated?.underlyingModel ||
-					data.models[modelId]?.underlyingModel ||
+					pearaiAgentModels.models[modelId]?.underlyingModelUpdated?.underlyingModel ||
+					pearaiAgentModels.models[modelId]?.underlyingModel ||
 					"claude-3-5-sonnet-20241022"
 				if (underlyingModel.startsWith("claude") || modelId.startsWith("anthropic/")) {
 					// Default to Claude
@@ -120,9 +122,9 @@ export class PearAiHandler extends BaseProvider implements SingleCompletionHandl
 		if (this.options.apiModelId) {
 			let modelInfo = null
 			if (this.options.apiModelId.startsWith("pearai")) {
-				modelInfo = this.pearAiModelsResponse?.models[this.options.apiModelId].underlyingModelUpdated
-			} else if (this.pearAiModelsResponse) {
-				modelInfo = this.pearAiModelsResponse.models[this.options.apiModelId || "pearai-model"]
+				modelInfo = this.pearAIAgentModels?.models[this.options.apiModelId].underlyingModelUpdated
+			} else if (this.pearAIAgentModels) {
+				modelInfo = this.pearAIAgentModels.models[this.options.apiModelId || "pearai-model"]
 			}
 			if (modelInfo) {
 				return {

src/core/webview/ClineProvider.ts

@@ -78,6 +78,8 @@ import { getUri } from "./getUri"
 import { telemetryService } from "../../services/telemetry/TelemetryService"
 import { TelemetrySetting } from "../../shared/TelemetrySetting"
 import { getWorkspacePath } from "../../utils/path"
+import { PEARAI_URL } from "../../shared/pearaiApi"
+import { PearAIAgentModelsConfig } from "../../api/providers/pearai/pearai"
 
 /**
  * https://github.com/microsoft/vscode-webview-ui-toolkit-samples/blob/main/default/weather-webview/src/providers/WeatherViewProvider.ts
@@ -459,6 +461,15 @@ export class ClineProvider extends EventEmitter<ClineProviderEvents> implements
 		this.outputChannel.appendLine("Webview view resolved")
 	}
 
+	public async getPearAIAgentModels() {
+		const response = await fetch(`${PEARAI_URL}/getPearAIAgentModels`)
+		if (!response.ok) {
+			throw new Error(`Failed to fetch models: ${response.statusText}`)
+		}
+		const data = (await response.json()) as PearAIAgentModelsConfig
+		return data
+	}
+
 	public async initClineWithSubTask(parent: Cline, task?: string, images?: string[]) {
 		return this.initClineWithTask(task, images, parent)
 	}
@@ -482,9 +493,11 @@ export class ClineProvider extends EventEmitter<ClineProviderEvents> implements
 		const modePrompt = customModePrompts?.[mode] as PromptComponent
 		const effectiveInstructions = [globalInstructions, modePrompt?.customInstructions].filter(Boolean).join("\n\n")
 
+		const pearaiAgentModels = await this.getPearAIAgentModels()
+
 		const cline = new Cline({
 			provider: this,
-			apiConfiguration,
+			apiConfiguration: { ...apiConfiguration, pearaiAgentModels },
 			customInstructions: effectiveInstructions,
 			enableDiff,
 			enableCheckpoints,
@@ -549,9 +562,11 @@ export class ClineProvider extends EventEmitter<ClineProviderEvents> implements
 			}
 		}
 
+		const pearaiAgentModels = await this.getPearAIAgentModels()
+
 		const cline = new Cline({
 			provider: this,
-			apiConfiguration,
+			apiConfiguration: { ...apiConfiguration, pearaiAgentModels },
 			customInstructions: effectiveInstructions,
 			enableDiff,
 			...checkpoints,

src/shared/api.ts

@@ -1,4 +1,5 @@
 import * as vscode from "vscode"
+import { PearAIAgentModelsConfig } from "../api/providers/pearai/pearai"
 
 export type ApiProvider =
 	| "anthropic"
@@ -81,6 +82,7 @@ export interface ApiHandlerOptions {
 	pearaiBaseUrl?: string
 	pearaiModelId?: string
 	pearaiModelInfo?: ModelInfo
+	pearaiAgentModels?: PearAIAgentModelsConfig
 	modelMaxThinkingTokens?: number
 	fakeAi?: unknown
 }