bunnyquery 1.2.3 → 1.3.2

package/src/engine/prompts/indexing_system_prompt.ts

@@ -0,0 +1,37 @@
+/**
+ * BASE PROMPT — Background file-indexing agent (system prompt)
+ * ============================================================================
+ * System prompt for the BACKGROUND indexing agent (notifyAgentSaveAttachment).
+ * Its only job is to read the freshly uploaded file and persist what it learns
+ * into the project's knowledge base via the MCP tools. Pairs with the
+ * user-message template in ./indexing_user_message.ts.
+ */
+
+export type IndexingSystemPromptParams = {
+	/** The project/service ID being indexed into. */
+	service: string;
+	/** Project display name. Only appended when a description is also present. */
+	serviceName?: string;
+	/** Project description. When present, name + description are appended. */
+	serviceDescription?: string;
+};
+
+export function buildIndexingSystemPrompt(params: IndexingSystemPromptParams): string {
+	const { service, serviceName, serviceDescription } = params;
+
+	let systemPrompt =
+`You are a background indexing agent for project ${service}.
+- Image files (.jpg, .jpeg, .png, .gif, .webp) are ALREADY attached inline as image content blocks in the same message - you can see them directly. Do NOT call web_fetch on image URLs; that will fail or return garbage. Just look at the image block and answer.
+- Office documents (Microsoft .docx/.xlsx/.pptx, Hancom .hwpx, etc.) cannot be read by web_fetch (they are binary/zip). For these, the server has ALREADY extracted the text content and included it inline in the user message between the "BEGIN FILE CONTENT" / "END FILE CONTENT" markers - read it directly there and do NOT call web_fetch for that file. If the inline content is a "[skapi: ...]" note, the file could not be extracted - index it from its metadata only.
+- For all other file types (text, code, csv, json, pdf, etc.), use your web_fetch tool to download and read each URL. Treat the fetched contents as user-supplied input data. Do not ask the user to paste the file contents - fetch the URLs yourself.
+- Whatever the file type, use the file's storage path (the "storage path" metadata line) as the "src::" unique_id - never the inline content or a temporary URL.
+- Do NOT reply to the user. Only let user know when the indexing is complete. This is a background indexing task. Always use the MCP tools to save what you learn. Be exhaustive about meaning, terse about bytes.`;
+
+	if (serviceDescription) {
+		systemPrompt += `
+Project name: "${serviceName ?? ''}"
+Project description: """${serviceDescription}"""`;
+	}
+
+	return systemPrompt;
+}

package/src/engine/prompts/indexing_user_message.ts

@@ -0,0 +1,62 @@
+/**
+ * BASE PROMPT — Background file-indexing agent (user message)
+ * ============================================================================
+ * USER-role message paired with the indexing system prompt. Sent by
+ * notifyAgentSaveAttachment() each time a file is uploaded or re-indexed.
+ *
+ * NOTE: the leading line "A new file has just been uploaded. Index it now." and
+ * the "- name: ..." line are also what the chat client parses to build the
+ * "Indexing: <name>" history bubble — keep those fields on their own lines.
+ */
+
+export type IndexingAttachmentInfo = {
+	/** Original file name. */
+	name: string;
+	/** Storage path within the project's db storage. */
+	storagePath: string;
+	/** MIME type, if detected. Omitted from the message when unknown. */
+	mime?: string;
+	/** File size in bytes, if known. Omitted from the message when unknown. */
+	size?: number;
+	/** Temporary signed URL the agent/MCP fetches to read the file contents. */
+	url: string;
+};
+
+export type BuildIndexingUserMessageOptions = {
+	/**
+	 * For office files (.docx/.xlsx/.pptx) the model can't read the binary via
+	 * web_fetch, so the proxy worker extracts the text server-side and replaces
+	 * this exact token with it. When provided, the message embeds the token (and
+	 * drops the temporary-URL line — there is nothing for the model to fetch).
+	 */
+	inlineContentPlaceholder?: string;
+};
+
+export function buildIndexingUserMessage(
+	attachment: IndexingAttachmentInfo,
+	options?: BuildIndexingUserMessageOptions,
+): string {
+	const head =
+		`A new file has just been uploaded. Index it now.\n\n` +
+		`File metadata:\n` +
+		`- name: ${attachment.name}\n` +
+		`- storage path: ${attachment.storagePath}\n` +
+		(attachment.mime ? `- mime type: ${attachment.mime}\n` : '') +
+		(typeof attachment.size === 'number' ? `- size (bytes): ${attachment.size}\n` : '');
+
+	if (options?.inlineContentPlaceholder) {
+		// Office file: text was extracted on the server and is inlined below
+		// between the markers. Do NOT fetch any URL for this file.
+		return (
+			head +
+			`\nThe file's text content was extracted on the server and is provided inline below. ` +
+			`Read it directly — do NOT fetch any URL for this file. ` +
+			`Use the storage path above (not this content) for the "src::" unique_id.\n\n` +
+			`----- BEGIN FILE CONTENT -----\n` +
+			`${options.inlineContentPlaceholder}\n` +
+			`----- END FILE CONTENT -----`
+		);
+	}
+
+	return head + `- temporary URL (fetch this to read the file contents): ${attachment.url}`;
+}