@prometheus-ai/agent 0.5.3 → 0.5.8

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

@@ -4,7 +4,7 @@
  * Calls Z.AI's remote MCP server (`webSearchPrime`) and adapts results into
  * the unified SearchResponse shape used by the web search tool.
  */
-import { type AuthStorage, getEnvApiKey } from "@prometheus-ai/ai";
+import { type ApiKey, type AuthStorage, type FetchImpl, getEnvApiKey, withAuth } from "@prometheus-ai/ai";
 import { asRecord, asString } from "../../../web/scrapers/utils";
 import type { SearchResponse, SearchSource } from "../../../web/search/types";
 import { SearchProviderError } from "../../../web/search/types";
@@ -21,6 +21,7 @@ export interface ZaiSearchParams {
 	query: string;
 	num_results?: number;
 	signal?: AbortSignal;
+	fetch?: FetchImpl;
 	authStorage: AuthStorage;
 	sessionId?: string;
 }
@@ -62,8 +63,13 @@ export async function findApiKey(
 	return (await authStorage.getApiKey("zai", sessionId, { signal })) ?? null;
 }
 
-async function callZaiTool(apiKey: string, args: Record<string, unknown>, signal?: AbortSignal): Promise<unknown> {
-	const response = await fetch(ZAI_MCP_URL, {
+async function callZaiTool(
+	apiKey: string,
+	args: Record<string, unknown>,
+	signal: AbortSignal | undefined,
+	fetchImpl: FetchImpl,
+): Promise<unknown> {
+	const response = await fetchImpl(ZAI_MCP_URL, {
 		method: "POST",
 		headers: {
 			Authorization: `Bearer ${apiKey}`,
@@ -158,6 +164,7 @@ async function callZaiTool(apiKey: string, args: Record<string, unknown>, signal
 
 async function callZaiSearch(apiKey: string, params: ZaiSearchParams): Promise<unknown> {
 	const count = params.num_results ?? DEFAULT_NUM_RESULTS;
+	const fetchImpl = params.fetch ?? fetch;
 	const attempts: Record<string, unknown>[] = [
 		{ query: params.query, count },
 		{ search_query: params.query, count },
@@ -167,7 +174,7 @@ async function callZaiSearch(apiKey: string, params: ZaiSearchParams): Promise<u
 	let lastError: unknown;
 	for (let i = 0; i < attempts.length; i++) {
 		try {
-			return await callZaiTool(apiKey, attempts[i], params.signal);
+			return await callZaiTool(apiKey, attempts[i], params.signal, fetchImpl);
 		} catch (error) {
 			lastError = error;
 			const isLastAttempt = i === attempts.length - 1;
@@ -278,12 +285,14 @@ function toSources(results: ZaiSearchResult[]): SearchSource[] {
 
 /** Execute Z.AI web search via remote MCP endpoint. */
 export async function searchZai(params: ZaiSearchParams): Promise<SearchResponse> {
-	const apiKey = await findApiKey(params.authStorage, params.sessionId, params.signal);
-	if (!apiKey) {
-		throw new Error("Z.AI credentials not found. Set ZAI_API_KEY or login with 'prometheus /login zai'.");
-	}
+	const keyOrResolver: ApiKey = params.authStorage.resolver("zai", {
+		sessionId: params.sessionId,
+	});
 
-	const rawResult = await callZaiSearch(apiKey, params);
+	const rawResult = await withAuth(keyOrResolver, key => callZaiSearch(key, params), {
+		signal: params.signal,
+		missingKeyMessage: "Z.AI credentials not found. Set ZAI_API_KEY or login with 'prometheus /login zai'.",
+	});
 	const payload = parseSearchPayload(rawResult);
 	let sources = toSources(payload.results);
 
@@ -299,6 +308,8 @@ export async function searchZai(params: ZaiSearchParams): Promise<SearchResponse
 	};
 }
 
+type ZaiProviderSearchParams = SearchParams & { fetch?: FetchImpl };
+
 /** Search provider for Z.AI web search MCP. */
 export class ZaiProvider extends SearchProvider {
 	readonly id = "zai";
@@ -309,12 +320,14 @@ export class ZaiProvider extends SearchProvider {
 	}
 
 	search(params: SearchParams): Promise<SearchResponse> {
+		const { fetch: fetchOverride } = params as ZaiProviderSearchParams;
 		return searchZai({
 			query: params.query,
 			num_results: params.numSearchResults ?? params.limit,
 			signal: params.signal,
 			authStorage: params.authStorage,
 			sessionId: params.sessionId,
+			fetch: fetchOverride,
 		});
 	}
 }

package/src/web/search/render.ts

@@ -15,23 +15,16 @@ import {
 	formatMoreItems,
 	formatStatusIcon,
 	getDomain,
-	getPreviewLines,
 	PREVIEW_LIMITS,
-	TRUNCATE_LENGTHS,
+	replaceTabs,
 	truncateToWidth,
 } from "../../tools/render-utils";
-import { renderStatusLine, renderTreeList } from "../../tui";
+import { renderStatusLine, renderTreeList, urlHyperlink } from "../../tui";
 import { CachedOutputBlock, markFramedBlockComponent } from "../../tui/output-block";
 import { getSearchProviderLabel } from "./provider";
 import type { SearchResponse } from "./types";
 
-const MAX_COLLAPSED_ANSWER_LINES = PREVIEW_LIMITS.COLLAPSED_LINES;
-const MAX_SNIPPET_LINES = 2;
-const MAX_SNIPPET_LINE_LEN = TRUNCATE_LENGTHS.LINE;
 const MAX_COLLAPSED_ITEMS = PREVIEW_LIMITS.COLLAPSED_ITEMS;
-const MAX_QUERY_PREVIEW = 2;
-const MAX_QUERY_LEN = 90;
-const MAX_REQUEST_ID_LEN = 36;
 
 function renderFallbackText(contentText: string, expanded: boolean, theme: Theme): Component {
 	const lines = contentText.split("\n").filter(line => line.trim());
@@ -66,6 +59,21 @@ export interface SearchRenderDetails {
 	error?: string;
 }
 
+/** Render a web search failure as a framed error panel, matching the success layout. */
+function renderSearchErrorPanel(message: string, providerLabel: string | undefined, theme: Theme): Component {
+	const header = renderStatusLine({ icon: "error", title: "Web Search", description: providerLabel }, theme);
+	const body = theme.fg("error", `Error: ${replaceTabs(message)}`);
+	const outputBlock = new CachedOutputBlock();
+	return markFramedBlockComponent({
+		render(width: number): readonly string[] {
+			return outputBlock.render({ header, state: "error", sections: [{ lines: [body] }], width }, theme);
+		},
+		invalidate() {
+			outputBlock.invalidate();
+		},
+	});
+}
+
 /** Render web search result with tree-based layout */
 export function renderSearchResult(
 	result: { content: Array<{ type: string; text?: string }>; details?: SearchRenderDetails },
@@ -78,9 +86,12 @@ export function renderSearchResult(
 ): Component {
 	const details = result.details;
 
-	// Handle error case
+	// Handle error case as a framed panel, matching the success layout.
 	if (details?.error) {
-		return new Text(theme.fg("error", `Error: ${details.error}`), 0, 0);
+		const errorProvider = details.response?.provider;
+		const errorProviderLabel =
+			errorProvider && errorProvider !== "none" ? getSearchProviderLabel(errorProvider) : undefined;
+		return renderSearchErrorPanel(details.error, errorProviderLabel, theme);
 	}
 
 	const rawText = result.content?.find(block => block.type === "text")?.text?.trim() ?? "";
@@ -91,8 +102,6 @@ export function renderSearchResult(
 
 	const sources = Array.isArray(response.sources) ? response.sources : [];
 	const sourceCount = sources.length;
-	const citations = Array.isArray(response.citations) ? response.citations : [];
-	const citationCount = citations.length;
 	const searchQueries = Array.isArray(response.searchQueries)
 		? response.searchQueries.filter(item => typeof item === "string")
 		: [];
@@ -108,26 +117,29 @@ export function renderSearchResult(
 		: searchQueries[0]
 			? truncateToWidth(searchQueries[0], 80)
 			: undefined;
+	const success = sourceCount > 0;
 	const header = renderStatusLine(
-		{
-			icon: sourceCount > 0 ? "success" : "warning",
-			title: "Web Search",
-			description: providerLabel,
-			meta: [formatCount("source", sourceCount)],
-		},
+		success
+			? {
+					iconOverride: theme.styledSymbol("tool.webSearch", "accent"),
+					title: "Web Search",
+					description: providerLabel,
+					meta: [formatCount("source", sourceCount)],
+				}
+			: {
+					icon: "warning",
+					title: "Web Search",
+					description: providerLabel,
+					meta: [formatCount("source", sourceCount)],
+				},
 		theme,
 	);
 
-	const metaLines: string[] = [];
-	metaLines.push(`${theme.fg("muted", "Provider:")} ${theme.fg("text", providerLabel)}`);
-	if (response.authMode)
-		metaLines.push(
-			`${theme.fg("muted", "Auth:")} ${theme.fg("text", response.authMode === "oauth" ? "OAuth" : response.authMode === "api_key" ? "API key" : response.authMode)}`,
-		);
-	if (response.model) metaLines.push(`${theme.fg("muted", "Model:")} ${theme.fg("text", response.model)}`);
-	metaLines.push(`${theme.fg("muted", "Sources:")} ${theme.fg("text", String(sourceCount))}`);
-	if (citationCount > 0)
-		metaLines.push(`${theme.fg("muted", "Citations:")} ${theme.fg("text", String(citationCount))}`);
+	const authShort =
+		response.authMode === "oauth" ? "OAuth" : response.authMode === "api_key" ? "API" : response.authMode;
+	let providerInfo = response.model ? `${response.model} @ ${providerLabel}` : providerLabel;
+	if (authShort) providerInfo += ` (${authShort})`;
+	const metaLines: string[] = [`${theme.fg("muted", "Provider:")} ${theme.fg("text", providerInfo)}`];
 	if (response.usage) {
 		const usageParts: string[] = [];
 		if (response.usage.inputTokens !== undefined) usageParts.push(`in ${response.usage.inputTokens}`);
@@ -137,41 +149,32 @@ export function renderSearchResult(
 		if (usageParts.length > 0)
 			metaLines.push(`${theme.fg("muted", "Usage:")} ${theme.fg("text", usageParts.join(theme.sep.dot))}`);
 	}
-	if (response.requestId) {
-		metaLines.push(
-			`${theme.fg("muted", "Request:")} ${theme.fg("text", truncateToWidth(response.requestId, MAX_REQUEST_ID_LEN))}`,
-		);
-	}
-	if (searchQueries.length > 0) {
-		const queriesPreview = searchQueries.slice(0, MAX_QUERY_PREVIEW);
-		const queryList = queriesPreview.map(q => truncateToWidth(q, MAX_QUERY_LEN));
-		const suffix = searchQueries.length > queriesPreview.length ? "…" : "";
-		metaLines.push(`${theme.fg("muted", "Queries:")} ${theme.fg("text", queryList.join("; "))}${suffix}`);
-	}
 
 	const answerMarkdown = contentText ? new Markdown(contentText, 0, 0, getMarkdownTheme()) : undefined;
 	const outputBlock = new CachedOutputBlock();
 
 	return markFramedBlockComponent({
-		render(width: number): string[] {
+		render(width: number): readonly string[] {
 			// Read mutable state at render time
 			const { expanded } = options;
 
 			// Answer lines: full markdown when expanded, capped markdown preview when collapsed.
 			const answerWidth = Math.max(20, width - 3);
 			const renderedAnswer = answerMarkdown ? answerMarkdown.render(answerWidth) : [];
-			let answerLines: string[];
+			let answerLines: readonly string[];
 			if (renderedAnswer.length === 0) {
 				answerLines = [theme.fg("muted", "No answer text returned")];
-			} else if (expanded) {
-				answerLines = renderedAnswer;
-			} else {
-				const collapsedCap = args?.maxAnswerLines ?? MAX_COLLAPSED_ANSWER_LINES;
-				answerLines = renderedAnswer.slice(0, collapsedCap);
-				const remaining = renderedAnswer.length - answerLines.length;
+			} else if (args?.maxAnswerLines !== undefined && !expanded) {
+				// CLI compact mode (`prometheus q`) caps the answer; the TUI passes no cap and shows it in full.
+				// `renderedAnswer` is the Markdown component's shared cache — slice copies before appending.
+				const capped = renderedAnswer.slice(0, args.maxAnswerLines);
+				const remaining = renderedAnswer.length - capped.length;
 				if (remaining > 0) {
-					answerLines.push(theme.fg("muted", formatMoreItems(remaining, "line")));
+					capped.push(theme.fg("muted", formatMoreItems(remaining, "line")));
 				}
+				answerLines = capped;
+			} else {
+				answerLines = renderedAnswer;
 			}
 
 			const sourceTree = renderTreeList(
@@ -187,30 +190,22 @@ export function renderSearchResult(
 								: typeof src.url === "string" && src.url.trim()
 									? src.url
 									: "Untitled";
-						const title = truncateToWidth(titleText, MAX_SNIPPET_LINE_LEN);
 						const url = typeof src.url === "string" ? src.url : "";
 						const domain = url ? getDomain(url) : "";
 						const age =
 							formatAge(src.ageSeconds) || (typeof src.publishedDate === "string" ? src.publishedDate : "");
 						const metaParts: string[] = [];
 						if (domain) metaParts.push(theme.fg("dim", `(${domain})`));
-						if (typeof src.author === "string" && src.author.trim())
-							metaParts.push(theme.fg("muted", truncateToWidth(src.author.trim(), 40)));
 						if (age) metaParts.push(theme.fg("muted", age));
 						const metaSep = theme.fg("dim", theme.sep.dot);
 						const metaSuffix = metaParts.length > 0 ? ` ${metaParts.join(metaSep)}` : "";
-						const srcLines: string[] = [
-							truncateToWidth(`${theme.fg("accent", title)}${metaSuffix}`, MAX_SNIPPET_LINE_LEN),
-						];
-						const snippetText = typeof src.snippet === "string" ? src.snippet : "";
-						if (snippetText.trim()) {
-							const snippetLines = getPreviewLines(snippetText, MAX_SNIPPET_LINES, MAX_SNIPPET_LINE_LEN);
-							for (const snippetLine of snippetLines) {
-								srcLines.push(theme.fg("muted", `${theme.format.dash} ${snippetLine}`));
-							}
-						}
-						if (url) srcLines.push(theme.fg("mdLinkUrl", truncateToWidth(url, MAX_SNIPPET_LINE_LEN)));
-						return srcLines;
+						// One line per source: the title links to its URL, followed by domain · age.
+						// Reserve room for the box borders, the tree branch, and the meta suffix.
+						const lineBudget = Math.max(24, width - 6);
+						const titleBudget = Math.max(12, lineBudget - Bun.stringWidth(metaSuffix));
+						const title = theme.fg("accent", truncateToWidth(titleText, titleBudget));
+						const linkedTitle = url ? urlHyperlink(url, title) : title;
+						return [`${linkedTitle}${metaSuffix}`];
 					},
 				},
 				theme,

package/src/web/search/types.ts

@@ -33,7 +33,11 @@ export const SEARCH_PROVIDER_OPTIONS = [
 		description: "Automatically uses the first configured web-search provider",
 	},
 	{ value: "tavily", label: "Tavily", description: "Requires TAVILY_API_KEY" },
-	{ value: "perplexity", label: "Perplexity", description: "Requires PERPLEXITY_COOKIES or PERPLEXITY_API_KEY" },
+	{
+		value: "perplexity",
+		label: "Perplexity",
+		description: "Uses auth when configured; explicit selection falls back to anonymous search",
+	},
 	{ value: "brave", label: "Brave", description: "Requires BRAVE_API_KEY" },
 	{ value: "jina", label: "Jina", description: "Requires JINA_API_KEY" },
 	{ value: "kimi", label: "Kimi", description: "Requires MOONSHOT_SEARCH_API_KEY or MOONSHOT_API_KEY" },

package/dist/types/discovery/context-files.d.ts

@@ -1,17 +0,0 @@
-import type { ContextFile } from "../capability/context-file";
-import type { LoadContext } from "../capability/types";
-export declare const PROMETHEUS_GENERAL_CONTEXT_FILES: readonly ["AGENTS.md", "USER.md", "SOUL.md"];
-export declare const STANDARD_CONTEXT_FILES: readonly ["AGENTS.md"];
-export type GeneralContextFileName = (typeof PROMETHEUS_GENERAL_CONTEXT_FILES)[number];
-export declare function mergeContextFileContent(parts: Array<{
-    name: string;
-    content: string;
-}>): string;
-export declare function loadMergedContextFileGroup(options: {
-    ctx: LoadContext;
-    dir: string;
-    level: "user" | "project";
-    providerId: string;
-    depth?: number;
-    fileNames?: readonly GeneralContextFileName[];
-}): Promise<ContextFile | null>;

package/dist/types/exa/factory.d.ts

@@ -1,13 +0,0 @@
-/**
- * Shared factory for creating Exa tools with consistent error handling and response formatting.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import type { ExaRenderDetails } from "./types";
-/** Creates an Exa tool with standardized API key handling, error wrapping, and optional search response formatting. */
-export declare function createExaTool(name: string, label: string, description: string, parameters: TSchema, mcpToolName: string, options?: {
-    /** When true, checks isSearchResponse and formats with formatSearchResults. Default: true */
-    formatResponse?: boolean;
-    /** Transform params before passing to callExaTool */
-    transformParams?: (params: Record<string, unknown>) => Record<string, unknown>;
-}): CustomTool<TSchema, ExaRenderDetails>;

package/dist/types/exa/render.d.ts

@@ -1,19 +0,0 @@
-/**
- * Exa TUI Rendering
- *
- * Tree-based rendering with collapsed/expanded states for Exa search results.
- */
-import type { Component } from "@prometheus-ai/tui";
-import type { RenderResultOptions } from "../extensibility/custom-tools/types";
-import type { Theme } from "../modes/theme/theme";
-import type { ExaRenderDetails } from "./types";
-/** Render Exa result with tree-based layout */
-export declare function renderExaResult(result: {
-    content: Array<{
-        type: string;
-        text?: string;
-    }>;
-    details?: ExaRenderDetails;
-}, options: RenderResultOptions, uiTheme: Theme): Component;
-/** Render Exa call (query/args preview) */
-export declare function renderExaCall(args: Record<string, unknown>, toolName: string, uiTheme: Theme): Component;

package/dist/types/exa/researcher.d.ts

@@ -1,9 +0,0 @@
-/**
- * Exa Researcher Tools
- *
- * Async research tasks with polling for completion.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import type { ExaRenderDetails } from "./types";
-export declare const researcherTools: CustomTool<TSchema, ExaRenderDetails>[];

package/dist/types/exa/search.d.ts

@@ -1,9 +0,0 @@
-/**
- * Exa Search Tools
- *
- * Basic neural/keyword search, deep research, code search, and URL crawling.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import type { ExaRenderDetails } from "./types";
-export declare const searchTools: CustomTool<TSchema, ExaRenderDetails>[];

package/dist/types/exa/websets.d.ts

@@ -1,9 +0,0 @@
-/**
- * Exa Websets Tools
- *
- * CRUD operations for websets, items, searches, enrichments, and monitoring.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import type { ExaRenderDetails } from "./types";
-export declare const websetsTools: CustomTool<TSchema, ExaRenderDetails>[];