add pearai provider

Author: Himanshu-Singh-Chauhan

.gitignore

@@ -15,3 +15,5 @@ roo-cline-*.vsix
 # Test environment
 .test_env
 .vscode-test/
+.aider*
+.env

src/api/index.ts

@@ -14,6 +14,7 @@ import { DeepSeekHandler } from "./providers/deepseek"
 import { MistralHandler } from "./providers/mistral"
 import { VsCodeLmHandler } from "./providers/vscode-lm"
 import { ApiStream } from "./transform/stream"
+import { PearAiHandler } from "./providers/pearai"
 
 export interface SingleCompletionHandler {
 	completePrompt(prompt: string): Promise<string>
@@ -53,6 +54,8 @@ export function buildApiHandler(configuration: ApiConfiguration): ApiHandler {
 			return new VsCodeLmHandler(options)
 		case "mistral":
 			return new MistralHandler(options)
+		case "pearai":
+			return new PearAiHandler(options)
 		default:
 			return new AnthropicHandler(options)
 	}

src/api/providers/pearai.ts

@@ -0,0 +1,19 @@
+import { OpenAiHandler } from "./openai"
+import { ApiHandlerOptions } from "../../shared/api"
+
+export class PearAiHandler extends OpenAiHandler {
+	constructor(options: ApiHandlerOptions) {
+		if (!options.pearaiApiKey) {
+			throw new Error("PearAI API key is required. Please provide it in the settings.")
+		}
+		super({
+			...options,
+			// Map PearAI specific options to OpenAI options for compatibility
+			openAiApiKey: options.pearaiApiKey,
+			openAiBaseUrl:
+				options.pearaiBaseUrl ??
+				"https://stingray-app-gb2an.ondigitalocean.app/pearai-server-api2/integrations/cline",
+			openAiStreamingEnabled: true,
+		})
+	}
+}

src/core/webview/ClineProvider.ts

@@ -62,6 +62,7 @@ type SecretKey =
 	| "openAiNativeApiKey"
 	| "deepSeekApiKey"
 	| "mistralApiKey"
+	| "pearaiApiKey"
 type GlobalStateKey =
 	| "apiProvider"
 	| "apiModelId"
@@ -96,6 +97,10 @@ type GlobalStateKey =
 	| "openRouterModelId"
 	| "openRouterModelInfo"
 	| "openRouterBaseUrl"
+	| "pearaiModelId"
+	| "pearaiModelInfo"
+	| "pearaiBaseUrl"
+	| "pearaiApiKey"
 	| "openRouterUseMiddleOutTransform"
 	| "allowedCommands"
 	| "soundEnabled"
@@ -1347,6 +1352,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 			openRouterUseMiddleOutTransform,
 			vsCodeLmModelSelector,
 			mistralApiKey,
+			pearaiApiKey,
+			pearaiBaseUrl,
+			pearaiModelId,
+			pearaiModelInfo,
 		} = apiConfiguration
 		await this.updateGlobalState("apiProvider", apiProvider)
 		await this.updateGlobalState("apiModelId", apiModelId)
@@ -1385,6 +1394,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 		await this.updateGlobalState("openRouterUseMiddleOutTransform", openRouterUseMiddleOutTransform)
 		await this.updateGlobalState("vsCodeLmModelSelector", vsCodeLmModelSelector)
 		await this.storeSecret("mistralApiKey", mistralApiKey)
+		await this.updateGlobalState("pearaiModelId", pearaiModelId)
+		await this.updateGlobalState("pearaiModelInfo", pearaiModelInfo)
+		await this.storeSecret("pearaiApiKey", pearaiApiKey)
+		await this.updateGlobalState("pearaiBaseUrl", pearaiBaseUrl)
 		if (this.cline) {
 			this.cline.api = buildApiHandler(apiConfiguration)
 		}
@@ -2003,6 +2016,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 			openAiNativeApiKey,
 			deepSeekApiKey,
 			mistralApiKey,
+			pearaiApiKey,
+			pearaiBaseUrl,
+			pearaiModelId,
+			pearaiModelInfo,
 			azureApiVersion,
 			openAiStreamingEnabled,
 			openRouterModelId,
@@ -2073,6 +2090,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 			this.getSecret("openAiNativeApiKey") as Promise<string | undefined>,
 			this.getSecret("deepSeekApiKey") as Promise<string | undefined>,
 			this.getSecret("mistralApiKey") as Promise<string | undefined>,
+			this.getSecret("pearaiApiKey") as Promise<string | undefined>,
+			this.getGlobalState("pearaiBaseUrl") as Promise<string | undefined>,
+			this.getGlobalState("pearaiModelId") as Promise<string | undefined>,
+			this.getGlobalState("pearaiModelInfo") as Promise<ModelInfo | undefined>,
 			this.getGlobalState("azureApiVersion") as Promise<string | undefined>,
 			this.getGlobalState("openAiStreamingEnabled") as Promise<boolean | undefined>,
 			this.getGlobalState("openRouterModelId") as Promise<string | undefined>,
@@ -2160,6 +2181,10 @@ export class ClineProvider implements vscode.WebviewViewProvider {
 				openAiNativeApiKey,
 				deepSeekApiKey,
 				mistralApiKey,
+				pearaiApiKey,
+				pearaiBaseUrl,
+				pearaiModelId,
+				pearaiModelInfo,
 				azureApiVersion,
 				openAiStreamingEnabled,
 				openRouterModelId,

src/shared/api.ts

@@ -14,6 +14,7 @@ export type ApiProvider =
 	| "deepseek"
 	| "vscode-lm"
 	| "mistral"
+	| "pearai"
 
 export interface ApiHandlerOptions {
 	apiModelId?: string
@@ -57,6 +58,10 @@ export interface ApiHandlerOptions {
 	deepSeekBaseUrl?: string
 	deepSeekApiKey?: string
 	includeMaxTokens?: boolean
+	pearaiApiKey?: string
+	pearaiBaseUrl?: string
+	pearaiModelId?: string
+	pearaiModelInfo?: ModelInfo
 }
 
 export type ApiConfiguration = ApiHandlerOptions & {

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

@@ -1,6 +1,12 @@
 import { Checkbox, Dropdown, Pane } from "vscrui"
 import type { DropdownOption } from "vscrui"
-import { VSCodeLink, VSCodeRadio, VSCodeRadioGroup, VSCodeTextField } from "@vscode/webview-ui-toolkit/react"
+import {
+	VSCodeButton,
+	VSCodeLink,
+	VSCodeRadio,
+	VSCodeRadioGroup,
+	VSCodeTextField,
+} from "@vscode/webview-ui-toolkit/react"
 import { Fragment, memo, useCallback, useEffect, useMemo, useState } from "react"
 import { useEvent, useInterval } from "react-use"
 import {
@@ -39,6 +45,8 @@ import OpenRouterModelPicker, {
 import OpenAiModelPicker from "./OpenAiModelPicker"
 import GlamaModelPicker from "./GlamaModelPicker"
 
+const PEARAI_DEFAULT_URL = "https://stingray-app-gb2an.ondigitalocean.app/pearai-server-api2/integrations/cline"
+
 interface ApiOptionsProps {
 	apiErrorMessage?: string
 	modelIdErrorMessage?: string
@@ -52,6 +60,7 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage }: ApiOptionsProps) =
 	const [anthropicBaseUrlSelected, setAnthropicBaseUrlSelected] = useState(!!apiConfiguration?.anthropicBaseUrl)
 	const [azureApiVersionSelected, setAzureApiVersionSelected] = useState(!!apiConfiguration?.azureApiVersion)
 	const [openRouterBaseUrlSelected, setOpenRouterBaseUrlSelected] = useState(!!apiConfiguration?.openRouterBaseUrl)
+	const [pearaiBaseUrlSelected, setPearaiBaseUrlSelected] = useState(!!apiConfiguration?.pearaiBaseUrl)
 	const [isDescriptionExpanded, setIsDescriptionExpanded] = useState(false)
 
 	const { selectedProvider, selectedModelId, selectedModelInfo } = useMemo(() => {
@@ -134,6 +143,7 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage }: ApiOptionsProps) =
 					}}
 					style={{ minWidth: 130, position: "relative", zIndex: OPENROUTER_MODEL_PICKER_Z_INDEX + 1 }}
 					options={[
+						{ value: "pearai", label: "PearAI" },
 						{ value: "openrouter", label: "OpenRouter" },
 						{ value: "anthropic", label: "Anthropic" },
 						{ value: "gemini", label: "Google Gemini" },
@@ -151,6 +161,47 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage }: ApiOptionsProps) =
 				/>
 			</div>
 
+			{selectedProvider === "pearai" && (
+				<div>
+					<VSCodeTextField
+						value={apiConfiguration?.pearaiApiKey || ""}
+						style={{ width: "100%" }}
+						type="password"
+						onInput={handleInputChange("pearaiApiKey")}
+						placeholder="Enter API Key...">
+						<span style={{ fontWeight: 500 }}>PearAI API Key</span>
+					</VSCodeTextField>
+					<VSCodeTextField
+						value={apiConfiguration?.pearaiBaseUrl || PEARAI_DEFAULT_URL}
+						style={{ width: "100%" }}
+						type="url"
+						onInput={handleInputChange("pearaiBaseUrl")}
+						placeholder={PEARAI_DEFAULT_URL}>
+						<span style={{ fontWeight: 500 }}>Base URL</span>
+					</VSCodeTextField>
+					{apiConfiguration?.pearaiBaseUrl && apiConfiguration.pearaiBaseUrl !== PEARAI_DEFAULT_URL && (
+						<VSCodeButton
+							onClick={() => {
+								handleInputChange("pearaiBaseUrl")({
+									target: {
+										value: PEARAI_DEFAULT_URL,
+									},
+								})
+							}}>
+							Reset to default URL
+						</VSCodeButton>
+					)}
+					<p
+						style={{
+							fontSize: "12px",
+							marginTop: "5px",
+							color: "var(--vscode-descriptionForeground)",
+						}}>
+						This key is stored locally and only used to make API requests from this extension.
+					</p>
+				</div>
+			)}
+
 			{selectedProvider === "anthropic" && (
 				<div>
 					<VSCodeTextField
@@ -1302,7 +1353,8 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage }: ApiOptionsProps) =
 				selectedProvider !== "openrouter" &&
 				selectedProvider !== "openai" &&
 				selectedProvider !== "ollama" &&
-				selectedProvider !== "lmstudio" && (
+				selectedProvider !== "lmstudio" &&
+				selectedProvider !== "pearai" && (
 					<>
 						<div className="dropdown-container">
 							<label htmlFor="model-id">
@@ -1552,6 +1604,12 @@ export function normalizeApiConfiguration(apiConfiguration?: ApiConfiguration) {
 					supportsImages: false, // VSCode LM API currently doesn't support images
 				},
 			}
+		case "pearai":
+			return {
+				selectedProvider: provider,
+				selectedModelId: apiConfiguration?.pearaiModelId || "",
+				selectedModelInfo: apiConfiguration?.pearaiModelInfo || openAiModelInfoSaneDefaults,
+			}
 		default:
 			return getProviderData(anthropicModels, anthropicDefaultModelId)
 	}