@prometheus-ai/agent 0.5.4 → 0.5.8

package/src/exa/render.ts

@@ -1,244 +0,0 @@
-/**
- * Exa TUI Rendering
- *
- * Tree-based rendering with collapsed/expanded states for Exa search results.
- */
-import type { Component } from "@prometheus-ai/tui";
-import { Text } from "@prometheus-ai/tui";
-import { logger } from "@prometheus-ai/utils";
-import type { RenderResultOptions } from "../extensibility/custom-tools/types";
-import type { Theme } from "../modes/theme/theme";
-import {
-	formatCount,
-	formatExpandHint,
-	formatMoreItems,
-	formatStatusIcon,
-	getDomain,
-	getPreviewLines,
-	PREVIEW_LIMITS,
-	TRUNCATE_LENGTHS,
-	truncateToWidth,
-} from "../tools/render-utils";
-import type { ExaRenderDetails } from "./types";
-
-const COLLAPSED_PREVIEW_LINES = PREVIEW_LIMITS.COLLAPSED_LINES;
-const COLLAPSED_PREVIEW_LINE_LEN = TRUNCATE_LENGTHS.LONG;
-const EXPANDED_TEXT_LINES = 5;
-const EXPANDED_TEXT_LINE_LEN = 90;
-const MAX_TITLE_LEN = TRUNCATE_LENGTHS.TITLE;
-const MAX_HIGHLIGHT_LEN = TRUNCATE_LENGTHS.CONTENT;
-
-function renderErrorMessage(message: string, theme: Theme): Text {
-	const clean = message.replace(/^Error:\s*/, "").trim();
-	return new Text(
-		`${formatStatusIcon("error", theme)} ${theme.fg("error", `Error: ${clean || "Unknown error"}`)}`,
-		0,
-		0,
-	);
-}
-
-function renderEmptyMessage(message: string, theme: Theme): Text {
-	return new Text(`${formatStatusIcon("warning", theme)} ${theme.fg("muted", message)}`, 0, 0);
-}
-
-/** Render Exa result with tree-based layout */
-export function renderExaResult(
-	result: { content: Array<{ type: string; text?: string }>; details?: ExaRenderDetails },
-	options: RenderResultOptions,
-	uiTheme: Theme,
-): Component {
-	const { expanded } = options;
-	const details = result.details;
-
-	if (details?.error) {
-		logger.error("Exa render error", { error: details.error, toolName: details.toolName });
-		return renderErrorMessage(details.error, uiTheme);
-	}
-
-	const response = details?.response;
-	if (!response) {
-		if (details?.raw) {
-			const rawText = typeof details.raw === "string" ? details.raw : JSON.stringify(details.raw, null, 2);
-			const rawLines = rawText.split("\n").filter(l => l.trim());
-			const maxLines = expanded ? rawLines.length : Math.min(rawLines.length, COLLAPSED_PREVIEW_LINES);
-			const displayLines = rawLines.slice(0, maxLines);
-			const remaining = rawLines.length - maxLines;
-			const expandHint = formatExpandHint(uiTheme, expanded, remaining > 0);
-
-			let text = `${formatStatusIcon("info", uiTheme)} ${uiTheme.fg("dim", "Raw response")}${expandHint}`;
-
-			for (let i = 0; i < displayLines.length; i++) {
-				const isLast = i === displayLines.length - 1 && remaining === 0;
-				const branch = isLast ? uiTheme.tree.last : uiTheme.tree.branch;
-				text += `\n ${uiTheme.fg("dim", branch)} ${uiTheme.fg(
-					"toolOutput",
-					truncateToWidth(displayLines[i], COLLAPSED_PREVIEW_LINE_LEN),
-				)}`;
-			}
-
-			if (remaining > 0) {
-				text += `\n ${uiTheme.fg("dim", uiTheme.tree.last)} ${uiTheme.fg(
-					"muted",
-					formatMoreItems(remaining, "line"),
-				)}`;
-			}
-
-			return new Text(text, 0, 0);
-		}
-		return renderEmptyMessage("No response data", uiTheme);
-	}
-
-	const results = response.results ?? [];
-	const resultCount = results.length;
-	const cost = response.costDollars?.total;
-	const time = response.searchTime;
-
-	const icon = formatStatusIcon(resultCount > 0 ? "success" : "warning", uiTheme);
-
-	const metaParts = [formatCount("result", resultCount)];
-	if (cost !== undefined) metaParts.push(`cost:$${cost.toFixed(4)}`);
-	if (time !== undefined) metaParts.push(`time:${time.toFixed(2)}s`);
-	const summaryText = metaParts.join(uiTheme.sep.dot);
-
-	let hasMorePreview = false;
-	if (!expanded && resultCount > 0) {
-		const previewText = results[0].text ?? results[0].title ?? "";
-		const totalLines = previewText.split("\n").filter(l => l.trim()).length;
-		hasMorePreview = totalLines > COLLAPSED_PREVIEW_LINES || resultCount > 1;
-	}
-	const expandHint = formatExpandHint(uiTheme, expanded, hasMorePreview);
-
-	let text = `${icon} ${uiTheme.fg("dim", summaryText)}${expandHint}`;
-
-	if (!expanded) {
-		if (resultCount === 0) {
-			text += `\n ${uiTheme.fg("dim", uiTheme.tree.last)} ${uiTheme.fg("muted", "No results")}`;
-			return new Text(text, 0, 0);
-		}
-
-		const first = results[0];
-		const previewText = first.text ?? first.title ?? "";
-		const previewLines = previewText
-			? getPreviewLines(previewText, COLLAPSED_PREVIEW_LINES, COLLAPSED_PREVIEW_LINE_LEN)
-			: [];
-		const safePreviewLines = previewLines.length > 0 ? previewLines : ["No preview text"];
-		const totalLines = previewText.split("\n").filter(l => l.trim()).length;
-		const remainingLines = Math.max(0, totalLines - previewLines.length);
-		const extraItems: string[] = [];
-		if (remainingLines > 0) {
-			extraItems.push(formatMoreItems(remainingLines, "line"));
-		}
-		if (resultCount > 1) {
-			extraItems.push(formatMoreItems(resultCount - 1, "result"));
-		}
-
-		for (let i = 0; i < safePreviewLines.length; i++) {
-			const isLast = i === safePreviewLines.length - 1 && extraItems.length === 0;
-			const branch = isLast ? uiTheme.tree.last : uiTheme.tree.branch;
-			const line = safePreviewLines[i];
-			const color = line === "No preview text" ? "muted" : "toolOutput";
-			text += `\n ${uiTheme.fg("dim", branch)} ${uiTheme.fg(color, line)}`;
-		}
-
-		for (let i = 0; i < extraItems.length; i++) {
-			const isLast = i === extraItems.length - 1;
-			const branch = isLast ? uiTheme.tree.last : uiTheme.tree.branch;
-			text += `\n ${uiTheme.fg("dim", branch)} ${uiTheme.fg("muted", extraItems[i])}`;
-		}
-
-		return new Text(text, 0, 0);
-	}
-
-	if (resultCount === 0) {
-		text += `\n ${uiTheme.fg("dim", uiTheme.tree.last)} ${uiTheme.fg("muted", "No results")}`;
-		return new Text(text, 0, 0);
-	}
-
-	for (let i = 0; i < results.length; i++) {
-		const res = results[i];
-		const isLast = i === results.length - 1;
-		const branch = isLast ? uiTheme.tree.last : uiTheme.tree.branch;
-		const cont = isLast ? " " : uiTheme.tree.vertical;
-
-		const title = truncateToWidth(res.title ?? "Untitled", MAX_TITLE_LEN);
-		const domain = res.url ? getDomain(res.url) : "";
-		const domainPart = domain ? uiTheme.fg("dim", ` (${domain})`) : "";
-
-		text += `\n ${uiTheme.fg("dim", branch)} ${uiTheme.fg("accent", title)}${domainPart}`;
-
-		if (res.url) {
-			text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-				"mdLinkUrl",
-				res.url,
-			)}`;
-		}
-
-		if (res.author) {
-			text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-				"muted",
-				`Author: ${res.author}`,
-			)}`;
-		}
-
-		if (res.publishedDate) {
-			text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-				"muted",
-				`Published: ${res.publishedDate}`,
-			)}`;
-		}
-
-		if (res.text) {
-			const textLines = res.text.split("\n").filter(l => l.trim());
-			const displayLines = textLines.slice(0, EXPANDED_TEXT_LINES);
-			for (const line of displayLines) {
-				text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-					"toolOutput",
-					truncateToWidth(line.trim(), EXPANDED_TEXT_LINE_LEN),
-				)}`;
-			}
-			if (textLines.length > EXPANDED_TEXT_LINES) {
-				text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-					"muted",
-					formatMoreItems(textLines.length - EXPANDED_TEXT_LINES, "line"),
-				)}`;
-			}
-		}
-
-		if (res.highlights?.length) {
-			text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-				"accent",
-				"Highlights",
-			)}`;
-			const maxHighlights = Math.min(res.highlights.length, 3);
-			for (let j = 0; j < maxHighlights; j++) {
-				const h = res.highlights[j];
-				text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-					"muted",
-					`${uiTheme.format.dash} ${truncateToWidth(h, MAX_HIGHLIGHT_LEN)}`,
-				)}`;
-			}
-			if (res.highlights.length > maxHighlights) {
-				text += `\n ${uiTheme.fg("dim", cont)} ${uiTheme.fg("dim", uiTheme.tree.hook)} ${uiTheme.fg(
-					"muted",
-					formatMoreItems(res.highlights.length - maxHighlights, "highlight"),
-				)}`;
-			}
-		}
-	}
-
-	return new Text(text, 0, 0);
-}
-
-/** Render Exa call (query/args preview) */
-export function renderExaCall(args: Record<string, unknown>, toolName: string, uiTheme: Theme): Component {
-	const toolLabel = toolName || "Exa Search";
-	const query = typeof args.query === "string" ? truncateToWidth(args.query, 80) : "?";
-	const numResults = typeof args.num_results === "number" ? args.num_results : undefined;
-
-	let text = `${uiTheme.fg("toolTitle", toolLabel)} ${uiTheme.fg("accent", query)}`;
-	if (numResults !== undefined) {
-		text += ` ${uiTheme.fg("muted", `results:${numResults}`)}`;
-	}
-
-	return new Text(text, 0, 0);
-}

package/src/exa/researcher.ts

@@ -1,36 +0,0 @@
-/**
- * Exa Researcher Tools
- *
- * Async research tasks with polling for completion.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import * as z from "zod/v4";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import { createExaTool } from "./factory";
-import type { ExaRenderDetails } from "./types";
-
-const researcherStartTool = createExaTool(
-	"exa_researcher_start",
-	"Start Deep Research",
-	"Start an asynchronous deep research task using Exa's researcher. Returns a task_id for polling completion.",
-	z.object({
-		query: z.string().describe("research query"),
-		depth: z.number().int().min(1).max(5).describe("research depth (1-5)").optional(),
-		breadth: z.number().int().min(1).max(5).describe("research breadth (1-5)").optional(),
-	}),
-	"deep_researcher_start",
-	{ formatResponse: false },
-);
-
-const researcherPollTool = createExaTool(
-	"exa_researcher_poll",
-	"Poll Research Status",
-	"Poll the status of an asynchronous research task. Returns status (pending|running|completed|failed) and result if completed.",
-	z.object({
-		task_id: z.string().describe("task id"),
-	}),
-	"deep_researcher_check",
-	{ formatResponse: false },
-);
-
-export const researcherTools: CustomTool<TSchema, ExaRenderDetails>[] = [researcherStartTool, researcherPollTool];

package/src/exa/search.ts

@@ -1,47 +0,0 @@
-/**
- * Exa Search Tools
- *
- * Basic neural/keyword search, deep research, code search, and URL crawling.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import * as z from "zod/v4";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import { createExaTool } from "./factory";
-import type { ExaRenderDetails } from "./types";
-
-/** exa_search - Basic neural/keyword search */
-const exaSearchTool = createExaTool(
-	"exa_search",
-	"Exa Search",
-	`Search the web using Exa's neural or keyword search.
-
-Returns structured search results with optional text content and highlights.
-
-Parameters:
-- query: Search query (required)
-- type: Search type - "neural" (semantic), "keyword" (exact), or "auto" (default: auto)
-- include_domains: Array of domains to include in results
-- exclude_domains: Array of domains to exclude from results
-- start_published_date: Filter results published after this date (ISO 8601)
-- end_published_date: Filter results published before this date (ISO 8601)
-- use_autoprompt: Let Exa optimize your query automatically (default: true)
-- text: Include page text content in results (default: false, costs more)
-- highlights: Include highlighted relevant snippets (default: false)
-- num_results: Maximum number of results to return (default: 10, max: 100)`,
-
-	z.object({
-		query: z.string().describe("search query"),
-		type: z.enum(["keyword", "neural", "auto"]).describe("search type").optional(),
-		include_domains: z.array(z.string()).describe("include domains").optional(),
-		exclude_domains: z.array(z.string()).describe("exclude domains").optional(),
-		start_published_date: z.string().describe("published after (iso 8601)").optional(),
-		end_published_date: z.string().describe("published before (iso 8601)").optional(),
-		use_autoprompt: z.boolean().describe("autoprompt").optional(),
-		text: z.boolean().describe("include page text").optional(),
-		highlights: z.boolean().describe("include highlights").optional(),
-		num_results: z.number().int().min(1).max(100).describe("max results (1-100)").optional(),
-	}),
-	"web_search_exa",
-);
-
-export const searchTools: CustomTool<TSchema, ExaRenderDetails>[] = [exaSearchTool];

package/src/exa/websets.ts

@@ -1,248 +0,0 @@
-/**
- * Exa Websets Tools
- *
- * CRUD operations for websets, items, searches, enrichments, and monitoring.
- */
-import type { TSchema } from "@prometheus-ai/ai";
-import * as z from "zod/v4";
-import type { CustomTool } from "../extensibility/custom-tools/types";
-import { callWebsetsTool, findApiKey } from "./mcp-client";
-import type { ExaRenderDetails } from "./types";
-
-/** Helper to create a websets tool with proper execute signature */
-function createWebsetTool(
-	name: string,
-	label: string,
-	description: string,
-	parameters: TSchema,
-	mcpToolName: string,
-): CustomTool<TSchema, ExaRenderDetails> {
-	return {
-		name,
-		label,
-		description,
-		parameters,
-		async execute(_toolCallId, params, _onUpdate, _ctx, _signal) {
-			try {
-				const apiKey = findApiKey();
-				if (!apiKey) {
-					return {
-						content: [{ type: "text" as const, text: "Error: EXA_API_KEY not found" }],
-						details: { error: "EXA_API_KEY not found", toolName: name },
-					};
-				}
-				const result = await callWebsetsTool(apiKey, mcpToolName, params as Record<string, unknown>);
-				return {
-					content: [{ type: "text" as const, text: JSON.stringify(result, null, 2) }],
-					details: { raw: result, toolName: name },
-				};
-			} catch (error) {
-				const message = error instanceof Error ? error.message : String(error);
-				return {
-					content: [{ type: "text" as const, text: `Error: ${message}` }],
-					details: { error: message, toolName: name },
-				};
-			}
-		},
-	};
-}
-
-// CRUD Operations
-const websetCreateTool = createWebsetTool(
-	"webset_create",
-	"Create Webset",
-	"Create a new webset collection for organizing web content.",
-	z.object({
-		name: z.string().describe("webset name"),
-		description: z.string().describe("description").optional(),
-	}),
-	"create_webset",
-);
-
-const websetListTool = createWebsetTool(
-	"webset_list",
-	"List Websets",
-	"List all websets in your account.",
-	z.object({}),
-	"list_websets",
-);
-
-const websetGetTool = createWebsetTool(
-	"webset_get",
-	"Get Webset",
-	"Get details of a specific webset by ID.",
-	z.object({
-		id: z.string().describe("webset id"),
-	}),
-	"get_webset",
-);
-
-const websetUpdateTool = createWebsetTool(
-	"webset_update",
-	"Update Webset",
-	"Update a webset's name or description.",
-	z.object({
-		id: z.string().describe("webset id"),
-		name: z.string().describe("new name").optional(),
-		description: z.string().describe("new description").optional(),
-	}),
-	"update_webset",
-);
-
-const websetDeleteTool = createWebsetTool(
-	"webset_delete",
-	"Delete Webset",
-	"Delete a webset and all its contents.",
-	z.object({
-		id: z.string().describe("webset id"),
-	}),
-	"delete_webset",
-);
-
-// Item Management
-const websetItemsListTool = createWebsetTool(
-	"webset_items_list",
-	"List Webset Items",
-	"List items in a webset with optional pagination.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		limit: z.number().describe("max items").optional(),
-		offset: z.number().describe("offset").optional(),
-	}),
-	"list_webset_items",
-);
-
-const websetItemGetTool = createWebsetTool(
-	"webset_item_get",
-	"Get Webset Item",
-	"Get a specific item from a webset.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		item_id: z.string().describe("item id"),
-	}),
-	"get_item",
-);
-
-// Search Operations
-const websetSearchCreateTool = createWebsetTool(
-	"webset_search_create",
-	"Create Webset Search",
-	"Create a new search within a webset.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		query: z.string().describe("search query"),
-	}),
-	"create_search",
-);
-
-const websetSearchGetTool = createWebsetTool(
-	"webset_search_get",
-	"Get Webset Search",
-	"Get the status and results of a webset search.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		search_id: z.string().describe("search id"),
-	}),
-	"get_search",
-);
-
-const websetSearchCancelTool = createWebsetTool(
-	"webset_search_cancel",
-	"Cancel Webset Search",
-	"Cancel a running webset search.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		search_id: z.string().describe("search id"),
-	}),
-	"cancel_search",
-);
-
-// Enrichment Operations
-const websetEnrichmentCreateTool = createWebsetTool(
-	"webset_enrichment_create",
-	"Create Enrichment",
-	"Create a new enrichment task for a webset.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		name: z.string().describe("enrichment name"),
-		prompt: z.string().describe("enrichment prompt"),
-	}),
-	"create_enrichment",
-);
-
-const websetEnrichmentGetTool = createWebsetTool(
-	"webset_enrichment_get",
-	"Get Enrichment",
-	"Get the status and results of an enrichment task.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		enrichment_id: z.string().describe("enrichment id"),
-	}),
-	"get_enrichment",
-);
-
-const websetEnrichmentUpdateTool = createWebsetTool(
-	"webset_enrichment_update",
-	"Update Enrichment",
-	"Update an enrichment's name or prompt.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		enrichment_id: z.string().describe("enrichment id"),
-		name: z.string().describe("new name").optional(),
-		prompt: z.string().describe("new prompt").optional(),
-	}),
-	"update_enrichment",
-);
-
-const websetEnrichmentDeleteTool = createWebsetTool(
-	"webset_enrichment_delete",
-	"Delete Enrichment",
-	"Delete an enrichment task.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		enrichment_id: z.string().describe("enrichment id"),
-	}),
-	"delete_enrichment",
-);
-
-const websetEnrichmentCancelTool = createWebsetTool(
-	"webset_enrichment_cancel",
-	"Cancel Enrichment",
-	"Cancel a running enrichment task.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		enrichment_id: z.string().describe("enrichment id"),
-	}),
-	"cancel_enrichment",
-);
-
-// Monitoring
-const websetMonitorCreateTool = createWebsetTool(
-	"webset_monitor_create",
-	"Create Monitor",
-	"Create a monitoring task for a webset with optional webhook notifications.",
-	z.object({
-		webset_id: z.string().describe("webset id"),
-		webhook_url: z.string().describe("webhook url").optional(),
-	}),
-	"create_monitor",
-);
-
-export const websetsTools: CustomTool<TSchema, ExaRenderDetails>[] = [
-	websetCreateTool,
-	websetListTool,
-	websetGetTool,
-	websetUpdateTool,
-	websetDeleteTool,
-	websetItemsListTool,
-	websetItemGetTool,
-	websetSearchCreateTool,
-	websetSearchGetTool,
-	websetSearchCancelTool,
-	websetEnrichmentCreateTool,
-	websetEnrichmentGetTool,
-	websetEnrichmentUpdateTool,
-	websetEnrichmentDeleteTool,
-	websetEnrichmentCancelTool,
-	websetMonitorCreateTool,
-];