@oh-my-pi/pi-coding-agent 11.0.3 → 11.2.0

package/src/web/search/providers/codex.ts

@@ -6,12 +6,14 @@
  * Returns synthesized answers with web search sources.
  */
 import * as os from "node:os";
-import { readSseData } from "@oh-my-pi/pi-utils";
+import { readSseJson } from "@oh-my-pi/pi-utils";
 import packageJson from "../../../../package.json" with { type: "json" };
 import { getAgentDbPath, getConfigDirPaths } from "../../../config";
 import { AgentStorage } from "../../../session/agent-storage";
-import type { WebSearchResponse, WebSearchSource } from "../../../web/search/types";
-import { WebSearchProviderError } from "../../../web/search/types";
+import type { SearchResponse, SearchSource } from "../../../web/search/types";
+import { SearchProviderError } from "../../../web/search/types";
+import type { SearchParams } from "./base";
+import { SearchProvider } from "./base";
 
 const CODEX_BASE_URL = "https://chatgpt.com/backend-api";
 const CODEX_RESPONSES_PATH = "/codex/responses";
@@ -21,6 +23,7 @@ const DEFAULT_INSTRUCTIONS =
 	"You are a helpful assistant with web search capabilities. Search the web to answer the user's question accurately and cite your sources.";
 
 export interface CodexSearchParams {
+	signal?: AbortSignal;
 	query: string;
 	system_prompt?: string;
 	num_results?: number;
@@ -173,15 +176,15 @@ function buildCodexHeaders(accessToken: string, accountId: string): Record<strin
  * @param query - Search query from the user
  * @param options - Search options including system prompt and context size
  * @returns Parsed response with answer, sources, and usage
- * @throws {WebSearchProviderError} If the API request fails
+ * @throws {SearchProviderError} If the API request fails
  */
-async function callCodexWebSearch(
+async function callCodexSearch(
 	auth: { accessToken: string; accountId: string },
 	query: string,
-	options: { systemPrompt?: string; searchContextSize?: "low" | "medium" | "high" },
+	options: { signal?: AbortSignal; systemPrompt?: string; searchContextSize?: "low" | "medium" | "high" },
 ): Promise<{
 	answer: string;
-	sources: WebSearchSource[];
+	sources: SearchSource[];
 	model: string;
 	requestId: string;
 	usage?: { inputTokens: number; outputTokens: number; totalTokens: number };
@@ -217,21 +220,21 @@ async function callCodexWebSearch(
 
 	if (!response.ok) {
 		const errorText = await response.text();
-		throw new WebSearchProviderError("codex", `Codex API error (${response.status}): ${errorText}`, response.status);
+		throw new SearchProviderError("codex", `Codex API error (${response.status}): ${errorText}`, response.status);
 	}
 
 	if (!response.body) {
-		throw new WebSearchProviderError("codex", "Codex API returned no response body", 500);
+		throw new SearchProviderError("codex", "Codex API returned no response body", 500);
 	}
 
 	// Parse SSE stream
 	const answerParts: string[] = [];
-	const sources: WebSearchSource[] = [];
+	const sources: SearchSource[] = [];
 	let model = DEFAULT_MODEL;
 	let requestId = "";
 	let usage: { inputTokens: number; outputTokens: number; totalTokens: number } | undefined;
 
-	for await (const rawEvent of readSseData<Record<string, unknown>>(response.body)) {
+	for await (const rawEvent of readSseJson<Record<string, unknown>>(response.body, options.signal)) {
 		const eventType = typeof rawEvent.type === "string" ? rawEvent.type : "";
 		if (!eventType) continue;
 
@@ -288,11 +291,11 @@ async function callCodexWebSearch(
 		} else if (eventType === "error") {
 			const code = (rawEvent as { code?: string }).code ?? "";
 			const message = (rawEvent as { message?: string }).message ?? "Unknown error";
-			throw new WebSearchProviderError("codex", `Codex error (${code}): ${message}`, 500);
+			throw new SearchProviderError("codex", `Codex error (${code}): ${message}`, 500);
 		} else if (eventType === "response.failed") {
 			const resp = (rawEvent as { response?: { error?: { message?: string } } }).response;
 			const errorMessage = resp?.error?.message ?? "Request failed";
-			throw new WebSearchProviderError("codex", `Codex request failed: ${errorMessage}`, 500);
+			throw new SearchProviderError("codex", `Codex request failed: ${errorMessage}`, 500);
 		}
 	}
 
@@ -312,7 +315,7 @@ async function callCodexWebSearch(
  * @returns Search response with synthesized answer, sources, and usage
  * @throws {Error} If no Codex OAuth credentials are configured
  */
-export async function searchCodex(params: CodexSearchParams): Promise<WebSearchResponse> {
+export async function searchCodex(params: CodexSearchParams): Promise<SearchResponse> {
 	const auth = await findCodexAuth();
 	if (!auth) {
 		throw new Error(
@@ -320,7 +323,7 @@ export async function searchCodex(params: CodexSearchParams): Promise<WebSearchR
 		);
 	}
 
-	const result = await callCodexWebSearch(auth, params.query, {
+	const result = await callCodexSearch(auth, params.query, {
 		systemPrompt: params.system_prompt,
 		searchContextSize: params.search_context_size ?? "high",
 	});
@@ -352,7 +355,26 @@ export async function searchCodex(params: CodexSearchParams): Promise<WebSearchR
  * Checks if Codex web search is available.
  * @returns True if valid OAuth credentials exist for openai-codex
  */
-export async function hasCodexWebSearch(): Promise<boolean> {
+export async function hasCodexSearch(): Promise<boolean> {
 	const auth = await findCodexAuth();
 	return auth !== null;
 }
+
+/** Search provider for OpenAI Codex web search. */
+export class CodexProvider extends SearchProvider {
+	readonly id = "codex";
+	readonly label = "Codex";
+
+	isAvailable(): Promise<boolean> {
+		return Promise.resolve(hasCodexSearch());
+	}
+
+	search(params: SearchParams): Promise<SearchResponse> {
+		return searchCodex({
+			signal: params.signal,
+			query: params.query,
+			system_prompt: params.systemPrompt,
+			num_results: params.numSearchResults ?? params.limit,
+		});
+	}
+}

package/src/web/search/providers/exa.ts

@@ -5,8 +5,11 @@
  * Returns structured search results with optional content extraction.
  */
 import { getEnvApiKey } from "@oh-my-pi/pi-ai";
-import type { WebSearchResponse, WebSearchSource } from "../../../web/search/types";
-import { WebSearchProviderError } from "../../../web/search/types";
+import { findApiKey as findExaKey } from "../../../exa/mcp-client";
+import type { SearchResponse, SearchSource } from "../../../web/search/types";
+import { SearchProviderError } from "../../../web/search/types";
+import type { SearchParams } from "./base";
+import { SearchProvider } from "./base";
 
 const EXA_API_URL = "https://api.exa.ai/search";
 
@@ -79,7 +82,7 @@ async function callExaSearch(apiKey: string, params: ExaSearchParams): Promise<E
 
 	if (!response.ok) {
 		const errorText = await response.text();
-		throw new WebSearchProviderError("exa", `Exa API error (${response.status}): ${errorText}`, response.status);
+		throw new SearchProviderError("exa", `Exa API error (${response.status}): ${errorText}`, response.status);
 	}
 
 	return response.json() as Promise<ExaSearchResponse>;
@@ -98,7 +101,7 @@ function dateToAgeSeconds(dateStr: string | null | undefined): number | undefine
 }
 
 /** Execute Exa web search */
-export async function searchExa(params: ExaSearchParams): Promise<WebSearchResponse> {
+export async function searchExa(params: ExaSearchParams): Promise<SearchResponse> {
 	const apiKey = getEnvApiKey("exa");
 	if (!apiKey) {
 		throw new Error("EXA_API_KEY not found. Set it in environment or .env file.");
@@ -106,8 +109,8 @@ export async function searchExa(params: ExaSearchParams): Promise<WebSearchRespo
 
 	const response = await callExaSearch(apiKey, params);
 
-	// Convert to unified WebSearchResponse
-	const sources: WebSearchSource[] = [];
+	// Convert to unified SearchResponse
+	const sources: SearchSource[] = [];
 
 	if (response.results) {
 		for (const result of response.results) {
@@ -132,3 +135,24 @@ export async function searchExa(params: ExaSearchParams): Promise<WebSearchRespo
 		requestId: response.requestId,
 	};
 }
+
+/** Search provider for Exa. */
+export class ExaProvider extends SearchProvider {
+	readonly id = "exa";
+	readonly label = "Exa";
+
+	isAvailable(): boolean {
+		try {
+			return !!findExaKey();
+		} catch {
+			return false;
+		}
+	}
+
+	search(params: SearchParams): Promise<SearchResponse> {
+		return searchExa({
+			query: params.query,
+			num_results: params.numSearchResults ?? params.limit,
+		});
+	}
+}

package/src/web/search/providers/gemini.ts

@@ -8,8 +8,10 @@
 import { refreshGoogleCloudToken } from "@oh-my-pi/pi-ai";
 import { getAgentDbPath, getConfigDirPaths } from "../../../config";
 import { AgentStorage } from "../../../session/agent-storage";
-import type { WebSearchCitation, WebSearchResponse, WebSearchSource } from "../../../web/search/types";
-import { WebSearchProviderError } from "../../../web/search/types";
+import type { SearchCitation, SearchResponse, SearchSource } from "../../../web/search/types";
+import { SearchProviderError } from "../../../web/search/types";
+import type { SearchParams } from "./base";
+import { SearchProvider } from "./base";
 
 const DEFAULT_ENDPOINT = "https://cloudcode-pa.googleapis.com";
 const ANTIGRAVITY_ENDPOINT = "https://daily-cloudcode-pa.sandbox.googleapis.com";
@@ -41,6 +43,10 @@ export interface GeminiSearchParams {
 	query: string;
 	system_prompt?: string;
 	num_results?: number;
+	/** Maximum output tokens. */
+	max_output_tokens?: number;
+	/** Sampling temperature (0–1). Lower = more focused/factual. */
+	temperature?: number;
 }
 
 /** OAuth credential stored in agent.db */
@@ -67,7 +73,7 @@ interface GeminiAuth {
  * Checks google-antigravity first (daily sandbox, more quota), then google-gemini-cli (prod).
  * @returns OAuth credential with access token and project ID, or null if none found
  */
-async function findGeminiAuth(): Promise<GeminiAuth | null> {
+export async function findGeminiAuth(): Promise<GeminiAuth | null> {
 	const configDirs = getConfigDirPaths("", { project: false });
 	const expiryBuffer = 5 * 60 * 1000; // 5 minutes
 	const now = Date.now();
@@ -191,16 +197,18 @@ interface CloudCodeResponseChunk {
  * @param query - Search query from the user
  * @param systemPrompt - Optional system prompt
  * @returns Parsed response with answer, sources, and usage
- * @throws {WebSearchProviderError} If the API request fails
+ * @throws {SearchProviderError} If the API request fails
  */
 async function callGeminiSearch(
 	auth: GeminiAuth,
 	query: string,
 	systemPrompt?: string,
+	maxOutputTokens?: number,
+	temperature?: number,
 ): Promise<{
 	answer: string;
-	sources: WebSearchSource[];
-	citations: WebSearchCitation[];
+	sources: SearchSource[];
+	citations: SearchCitation[];
 	searchQueries: string[];
 	model: string;
 	usage?: { inputTokens: number; outputTokens: number; totalTokens: number };
@@ -209,7 +217,7 @@ async function callGeminiSearch(
 	const url = `${endpoint}/v1internal:streamGenerateContent?alt=sse`;
 	const headers = auth.isAntigravity ? ANTIGRAVITY_HEADERS : GEMINI_CLI_HEADERS;
 
-	const requestBody = {
+	const requestBody: Record<string, unknown> = {
 		project: auth.projectId,
 		model: DEFAULT_MODEL,
 		request: {
@@ -231,6 +239,17 @@ async function callGeminiSearch(
 		requestId: `search-${Date.now()}-${Math.random().toString(36).slice(2, 11)}`,
 	};
 
+	if (maxOutputTokens !== undefined || temperature !== undefined) {
+		const generationConfig: Record<string, number> = {};
+		if (maxOutputTokens !== undefined) {
+			generationConfig.maxOutputTokens = maxOutputTokens;
+		}
+		if (temperature !== undefined) {
+			generationConfig.temperature = temperature;
+		}
+		(requestBody.request as Record<string, unknown>).generationConfig = generationConfig;
+	}
+
 	const response = await fetch(url, {
 		method: "POST",
 		headers: {
@@ -244,7 +263,7 @@ async function callGeminiSearch(
 
 	if (!response.ok) {
 		const errorText = await response.text();
-		throw new WebSearchProviderError(
+		throw new SearchProviderError(
 			"gemini",
 			`Gemini Cloud Code API error (${response.status}): ${errorText}`,
 			response.status,
@@ -252,13 +271,13 @@ async function callGeminiSearch(
 	}
 
 	if (!response.body) {
-		throw new WebSearchProviderError("gemini", "Gemini API returned no response body", 500);
+		throw new SearchProviderError("gemini", "Gemini API returned no response body", 500);
 	}
 
 	// Parse SSE stream
 	const answerParts: string[] = [];
-	const sources: WebSearchSource[] = [];
-	const citations: WebSearchCitation[] = [];
+	const sources: SearchSource[] = [];
+	const citations: SearchCitation[] = [];
 	const searchQueries: string[] = [];
 	const seenUrls = new Set<string>();
 	let model = DEFAULT_MODEL;
@@ -388,7 +407,7 @@ async function callGeminiSearch(
  * @returns Search response with synthesized answer, sources, and citations
  * @throws {Error} If no Gemini OAuth credentials are configured
  */
-export async function searchGemini(params: GeminiSearchParams): Promise<WebSearchResponse> {
+export async function searchGemini(params: GeminiSearchParams): Promise<SearchResponse> {
 	const auth = await findGeminiAuth();
 	if (!auth) {
 		throw new Error(
@@ -396,7 +415,13 @@ export async function searchGemini(params: GeminiSearchParams): Promise<WebSearc
 		);
 	}
 
-	const result = await callGeminiSearch(auth, params.query, params.system_prompt);
+	const result = await callGeminiSearch(
+		auth,
+		params.query,
+		params.system_prompt,
+		params.max_output_tokens,
+		params.temperature,
+	);
 
 	let sources = result.sources;
 
@@ -416,11 +441,22 @@ export async function searchGemini(params: GeminiSearchParams): Promise<WebSearc
 	};
 }
 
-/**
- * Checks if Gemini web search is available.
- * @returns True if valid OAuth credentials exist for google-gemini-cli or google-antigravity
- */
-export async function hasGeminiWebSearch(): Promise<boolean> {
-	const auth = await findGeminiAuth();
-	return auth !== null;
+/** Search provider for Google Gemini web search. */
+export class GeminiProvider extends SearchProvider {
+	readonly id = "gemini";
+	readonly label = "Gemini";
+
+	isAvailable() {
+		return findGeminiAuth().then(Boolean);
+	}
+
+	search(params: SearchParams): Promise<SearchResponse> {
+		return searchGemini({
+			query: params.query,
+			system_prompt: params.systemPrompt,
+			num_results: params.numSearchResults ?? params.limit,
+			max_output_tokens: params.maxOutputTokens,
+			temperature: params.temperature,
+		});
+	}
 }

package/src/web/search/providers/jina.ts

@@ -6,8 +6,10 @@
  */
 
 import { getEnvApiKey } from "@oh-my-pi/pi-ai";
-import type { WebSearchResponse, WebSearchSource } from "../../../web/search/types";
-import { WebSearchProviderError } from "../../../web/search/types";
+import type { SearchResponse, SearchSource } from "../../../web/search/types";
+import { SearchProviderError } from "../../../web/search/types";
+import type { SearchParams } from "./base";
+import { SearchProvider } from "./base";
 
 const JINA_SEARCH_URL = "https://s.jina.ai";
 
@@ -41,7 +43,7 @@ async function callJinaSearch(apiKey: string, query: string): Promise<JinaSearch
 
 	if (!response.ok) {
 		const errorText = await response.text();
-		throw new WebSearchProviderError("jina", `Jina API error (${response.status}): ${errorText}`, response.status);
+		throw new SearchProviderError("jina", `Jina API error (${response.status}): ${errorText}`, response.status);
 	}
 
 	const data = (await response.json()) as unknown;
@@ -49,14 +51,14 @@ async function callJinaSearch(apiKey: string, query: string): Promise<JinaSearch
 }
 
 /** Execute Jina web search. */
-export async function searchJina(params: JinaSearchParams): Promise<WebSearchResponse> {
+export async function searchJina(params: JinaSearchParams): Promise<SearchResponse> {
 	const apiKey = findApiKey();
 	if (!apiKey) {
 		throw new Error("JINA_API_KEY not found. Set it in environment or .env file.");
 	}
 
 	const response = await callJinaSearch(apiKey, params.query);
-	const sources: WebSearchSource[] = [];
+	const sources: SearchSource[] = [];
 
 	for (const result of response) {
 		if (!result?.url) continue;
@@ -74,3 +76,24 @@ export async function searchJina(params: JinaSearchParams): Promise<WebSearchRes
 		sources: limitedSources,
 	};
 }
+
+/** Search provider for Jina Reader. */
+export class JinaProvider extends SearchProvider {
+	readonly id = "jina";
+	readonly label = "Jina";
+
+	isAvailable() {
+		try {
+			return !!findApiKey();
+		} catch {
+			return false;
+		}
+	}
+
+	search(params: SearchParams): Promise<SearchResponse> {
+		return searchJina({
+			query: params.query,
+			num_results: params.numSearchResults ?? params.limit,
+		});
+	}
+}

package/src/web/search/providers/perplexity.ts

@@ -7,21 +7,34 @@
 
 import { getEnvApiKey } from "@oh-my-pi/pi-ai";
 import type {
+	PerplexityMessageOutput,
 	PerplexityRequest,
 	PerplexityResponse,
-	WebSearchCitation,
-	WebSearchResponse,
-	WebSearchSource,
+	SearchCitation,
+	SearchResponse,
+	SearchSource,
 } from "../../../web/search/types";
-import { WebSearchProviderError } from "../../../web/search/types";
+import { SearchProviderError } from "../../../web/search/types";
+import type { SearchParams } from "./base";
+import { SearchProvider } from "./base";
 
 const PERPLEXITY_API_URL = "https://api.perplexity.ai/chat/completions";
 
+const DEFAULT_MAX_TOKENS = 8192;
+const DEFAULT_TEMPERATURE = 0.2;
+const DEFAULT_NUM_SEARCH_RESULTS = 10;
+
 export interface PerplexitySearchParams {
 	query: string;
 	system_prompt?: string;
-	search_recency_filter?: "day" | "week" | "month" | "year";
+	search_recency_filter?: "hour" | "day" | "week" | "month" | "year";
 	num_results?: number;
+	/** Maximum output tokens. Defaults to 4096. */
+	max_tokens?: number;
+	/** Sampling temperature (0–1). Lower = more focused/factual. Defaults to 0.2. */
+	temperature?: number;
+	/** Number of search results to retrieve. Defaults to 10. */
+	num_search_results?: number;
 }
 
 /** Find PERPLEXITY_API_KEY from environment or .env files (also checks PPLX_API_KEY) */
@@ -42,7 +55,7 @@ async function callPerplexity(apiKey: string, request: PerplexityRequest): Promi
 
 	if (!response.ok) {
 		const errorText = await response.text();
-		throw new WebSearchProviderError(
+		throw new SearchProviderError(
 			"perplexity",
 			`Perplexity API error (${response.status}): ${errorText}`,
 			response.status,
@@ -53,7 +66,7 @@ async function callPerplexity(apiKey: string, request: PerplexityRequest): Promi
 }
 
 /** Calculate age in seconds from ISO date string */
-function dateToAgeSeconds(dateStr: string | undefined): number | undefined {
+function dateToAgeSeconds(dateStr: string | null | undefined): number | undefined {
 	if (!dateStr) return undefined;
 	try {
 		const date = new Date(dateStr);
@@ -64,30 +77,49 @@ function dateToAgeSeconds(dateStr: string | undefined): number | undefined {
 	}
 }
 
-/** Parse API response into unified WebSearchResponse */
-function parseResponse(response: PerplexityResponse): WebSearchResponse {
-	const answer = response.choices[0]?.message?.content ?? "";
+function messageContentToText(content: PerplexityMessageOutput["content"]): string {
+	if (!content) return "";
+	if (typeof content === "string") return content;
+	return content.map(chunk => (chunk.type === "text" ? chunk.text : "")).join("");
+}
+
+/** Parse API response into unified SearchResponse */
+function parseResponse(response: PerplexityResponse): SearchResponse {
+	const messageContent = response.choices[0]?.message?.content ?? null;
+	const answer = messageContentToText(messageContent);
 
 	// Build sources by matching citations to search_results
-	const sources: WebSearchSource[] = [];
-	const citations: WebSearchCitation[] = [];
+	const sources: SearchSource[] = [];
+	const citations: SearchCitation[] = [];
 
 	const citationUrls = response.citations ?? [];
 	const searchResults = response.search_results ?? [];
 
-	for (const url of citationUrls) {
-		const searchResult = searchResults.find(r => r.url === url);
-		sources.push({
-			title: searchResult?.title ?? url,
-			url,
-			snippet: searchResult?.snippet,
-			publishedDate: searchResult?.date,
-			ageSeconds: dateToAgeSeconds(searchResult?.date),
-		});
-		citations.push({
-			url,
-			title: searchResult?.title ?? url,
-		});
+	if (citationUrls.length > 0) {
+		for (const url of citationUrls) {
+			const searchResult = searchResults.find(r => r.url === url);
+			sources.push({
+				title: searchResult?.title ?? url,
+				url,
+				snippet: searchResult?.snippet,
+				publishedDate: searchResult?.date ?? undefined,
+				ageSeconds: dateToAgeSeconds(searchResult?.date),
+			});
+			citations.push({
+				url,
+				title: searchResult?.title ?? url,
+			});
+		}
+	} else {
+		for (const searchResult of searchResults) {
+			sources.push({
+				title: searchResult.title ?? searchResult.url,
+				url: searchResult.url,
+				snippet: searchResult.snippet,
+				publishedDate: searchResult.date ?? undefined,
+				ageSeconds: dateToAgeSeconds(searchResult.date),
+			});
+		}
 	}
 
 	return {
@@ -95,37 +127,46 @@ function parseResponse(response: PerplexityResponse): WebSearchResponse {
 		answer: answer || undefined,
 		sources,
 		citations: citations.length > 0 ? citations : undefined,
-		relatedQuestions: response.related_questions,
-		usage: {
-			inputTokens: response.usage.prompt_tokens,
-			outputTokens: response.usage.completion_tokens,
-			totalTokens: response.usage.total_tokens,
-		},
+		usage: response.usage
+			? {
+					inputTokens: response.usage.prompt_tokens,
+					outputTokens: response.usage.completion_tokens,
+					totalTokens: response.usage.total_tokens,
+				}
+			: undefined,
 		model: response.model,
 		requestId: response.id,
 	};
 }
 
 /** Execute Perplexity web search */
-export async function searchPerplexity(params: PerplexitySearchParams): Promise<WebSearchResponse> {
+export async function searchPerplexity(params: PerplexitySearchParams): Promise<SearchResponse> {
 	const apiKey = findApiKey();
 	if (!apiKey) {
 		throw new Error("PERPLEXITY_API_KEY not found. Set it in environment or .env file.");
 	}
 
+	const systemPrompt = params.system_prompt;
 	const messages: PerplexityRequest["messages"] = [];
-	if (params.system_prompt) {
-		messages.push({ role: "system", content: params.system_prompt });
+	if (systemPrompt) {
+		messages.push({ role: "system", content: systemPrompt });
 	}
 	messages.push({ role: "user", content: params.query });
 
 	const request: PerplexityRequest = {
 		model: "sonar-pro",
 		messages,
-		return_related_questions: false,
+		max_tokens: params.max_tokens ?? DEFAULT_MAX_TOKENS,
+		temperature: params.temperature ?? DEFAULT_TEMPERATURE,
+		search_mode: "web",
+		num_search_results: params.num_search_results ?? DEFAULT_NUM_SEARCH_RESULTS,
 		web_search_options: {
-			search_context_size: "high",
+			search_type: "pro",
+			search_context_size: "medium",
 		},
+		enable_search_classifier: true,
+		reasoning_effort: "medium",
+		language_preference: "en",
 	};
 
 	if (params.search_recency_filter) {
@@ -142,3 +183,29 @@ export async function searchPerplexity(params: PerplexitySearchParams): Promise<
 
 	return result;
 }
+
+/** Search provider for Perplexity. */
+export class PerplexityProvider extends SearchProvider {
+	readonly id = "perplexity";
+	readonly label = "Perplexity";
+
+	isAvailable() {
+		try {
+			return !!findApiKey();
+		} catch {
+			return false;
+		}
+	}
+
+	search(params: SearchParams): Promise<SearchResponse> {
+		return searchPerplexity({
+			query: params.query,
+			temperature: params.temperature,
+			max_tokens: params.maxOutputTokens,
+			num_search_results: params.numSearchResults,
+			system_prompt: params.systemPrompt,
+			search_recency_filter: params.recency,
+			num_results: params.limit,
+		});
+	}
+}