chat-agent-toolkit 1.2.300 → 1.2.301

package/dist/tools/search/meta-search-types.d.ts

@@ -8,7 +8,9 @@ export interface ChatTurnMessage {
 /** A `[role, content]` tuple used for few-shot prompt examples. */
 export type FewShotExample = [role: "user" | "assistant", content: string];
 export interface MetaSearchAgentType {
-    searchAndAnswer: (message: string, history: ChatTurnMessage[], llm: LanguageModel, optimizationMode: "speed" | "balanced" | "quality", fileIds: string[], systemInstructions: string, category?: string, sourceExtractionEnabled?: boolean, thinkingTimeLimit?: number) => Promise<EventEmitter>;
+    searchAndAnswer: (message: string, history: ChatTurnMessage[], llm: LanguageModel, optimizationMode: "speed" | "balanced" | "quality", fileIds: string[], systemInstructions: string, category?: string, sourceExtractionEnabled?: boolean, thinkingTimeLimit?: number, 
+    /** User-authored replacement for the focus mode's query-expansion prompt. */
+    queryExpansionPrompt?: string) => Promise<EventEmitter>;
 }
 /** Emitted on the EventEmitter data channel to report live search progress. */
 export interface SearchingEvent {

package/dist/tools/search/metaSearchAgent.d.ts

@@ -16,6 +16,6 @@ declare class MetaSearchAgent implements MetaSearchAgentType {
      * "response", "searching", "end", and "error" events on the emitter.
      */
     private runPipeline;
-    searchAndAnswer(message: string, history: ChatTurnMessage[], llm: LanguageModel, optimizationMode: "speed" | "balanced" | "quality", fileIds: string[], systemInstructions: string, category?: string, sourceExtractionEnabled?: boolean, thinkingTimeLimit?: number): Promise<EventEmitter<any>>;
+    searchAndAnswer(message: string, history: ChatTurnMessage[], llm: LanguageModel, optimizationMode: "speed" | "balanced" | "quality", fileIds: string[], systemInstructions: string, category?: string, sourceExtractionEnabled?: boolean, thinkingTimeLimit?: number, queryExpansionPrompt?: string): Promise<EventEmitter<any>>;
 }
 export default MetaSearchAgent;

package/dist/tools/search/suggestionGeneratorAgent.d.ts

@@ -3,6 +3,12 @@ import { ChatTurnMessage } from './meta-search-types';
 type SuggestionGeneratorInput = {
     chat_history: ChatTurnMessage[];
     maxQuestions?: number;
+    /**
+     * Optional user-authored replacement for {@link followUpSuggestionsPrompt}
+     * (edited in Settings → Search Settings). Blank or whitespace-only values
+     * fall back to the built-in template.
+     */
+    promptTemplate?: string;
 };
 declare const generateSuggestions: (input: SuggestionGeneratorInput, llm: LanguageModel) => Promise<string[]>;
 export default generateSuggestions;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chat-agent-toolkit",
-  "version": "1.2.300",
+  "version": "1.2.301",
   "description": "Multi-provider AI agent toolkit: generate language responses, search the web, extract content, and manage memory across 10+ LLM providers.",
   "author": "vtempest <grokthiscontact@gmail.com>",
   "license": "rights.institute/PROSPER",
@@ -98,7 +98,7 @@
     "html-entities": "^2.6.0",
     "manage-storage": "^0.0.47",
     "marked": "^18.0.12",
-    "qwksearch-api-client": "^0.9.242",
+    "qwksearch-api-client": "^0.9.243",
     "write-language": "^0.1.153",
     "zod": "^4.5.4"
   },

package/src/prompts/search-prompts.ts

@@ -92,6 +92,30 @@ https://example.com
   ],
 ];
 
+/**
+ * Default template for generating follow-up question suggestions after an
+ * answer. Interpolated with `{maxQuestions}` and `{chat_history}`.
+ *
+ * Exported (rather than inlined in the suggestion agent) so the settings UI can
+ * show the live default in an editable textarea and users can override it.
+ */
+export const followUpSuggestionsPrompt = `
+You are an AI suggestion generator for an AI powered search engine. You will be given a conversation below. You need to generate {maxQuestions} suggestions based on the conversation. The suggestion should be relevant to the conversation that can be used by the user to ask the chat model for more information.
+You need to make sure the suggestions are relevant to the conversation and are helpful to the user. Keep a note that the user might use these suggestions to ask a chat model for more information.
+Make sure the suggestions are medium in length and are informative and relevant to the conversation.
+
+Provide these suggestions separated by newlines between the XML tags <suggestions> and </suggestions>. For example:
+
+<suggestions>
+Tell me more about SpaceX and their recent projects
+What is the latest news on SpaceX?
+Who is the CEO of SpaceX?
+</suggestions>
+
+Conversation:
+{chat_history}
+`;
+
 export const webSearchResponsePrompt = `
     You are QwkSearch, an AI model skilled in web search and crafting detailed, engaging, and well-structured answers. You excel at summarizing web pages and extracting relevant information to create professional, blog-style responses.
 
@@ -163,4 +187,5 @@ export default {
   webSearchRetrieverPrompt,
   webSearchRetrieverFewShots,
   writingAssistantPrompt,
+  followUpSuggestionsPrompt,
 };

package/src/tools/search/meta-search-types.ts

@@ -25,6 +25,8 @@ export interface MetaSearchAgentType {
     category?: string,
     sourceExtractionEnabled?: boolean,
     thinkingTimeLimit?: number,
+    /** User-authored replacement for the focus mode's query-expansion prompt. */
+    queryExpansionPrompt?: string,
   ) => Promise<EventEmitter>;
 }
 

package/src/tools/search/metaSearchAgent.ts

@@ -88,11 +88,19 @@ class MetaSearchAgent implements MetaSearchAgentType {
     sourceExtractionEnabled = false,
     thinkingTimeLimit = 0,
     emitter?: EventEmitter,
+    queryGeneratorPromptOverride?: string,
   ): Promise<{ query: string; docs: Document[] }> {
+    // A user-authored query-expansion prompt (Settings → Search Settings)
+    // replaces the focus mode's built-in one. Blank or whitespace-only text
+    // means "use the built-in prompt".
+    const queryGeneratorPrompt = queryGeneratorPromptOverride?.trim()
+      ? queryGeneratorPromptOverride
+      : this.config.queryGeneratorPrompt;
+
     const { text: retrieverOutput } = await generateText({
       model: llm,
       temperature: 0,
-      system: this.config.queryGeneratorPrompt,
+      system: queryGeneratorPrompt,
       messages: [
         ...this.config.queryGeneratorFewShots.map(([role, content]) => ({
           role,
@@ -333,6 +341,7 @@ class MetaSearchAgent implements MetaSearchAgentType {
     category: string,
     sourceExtractionEnabled: boolean,
     thinkingTimeLimit: number,
+    queryExpansionPrompt?: string,
   ): Promise<void> {
     try {
       let docs: Document[] | null = null;
@@ -347,6 +356,7 @@ class MetaSearchAgent implements MetaSearchAgentType {
           sourceExtractionEnabled,
           thinkingTimeLimit,
           emitter,
+          queryExpansionPrompt,
         );
         query = result.query;
         docs = result.docs;
@@ -454,6 +464,7 @@ class MetaSearchAgent implements MetaSearchAgentType {
     category: string = "general",
     sourceExtractionEnabled = false,
     thinkingTimeLimit = 0,
+    queryExpansionPrompt?: string,
   ) {
     const emitter = new EventEmitter();
 
@@ -471,6 +482,7 @@ class MetaSearchAgent implements MetaSearchAgentType {
         category,
         sourceExtractionEnabled,
         thinkingTimeLimit,
+        queryExpansionPrompt,
       );
     }, 0);
 

package/src/tools/search/suggestionGeneratorAgent.ts

@@ -6,28 +6,18 @@
 import { generateText, type LanguageModel } from "ai";
 import { LineListOutputParser } from "../../utils/outputParser";
 import { formatChatHistoryAsString } from "../../utils";
+import { followUpSuggestionsPrompt } from "../../prompts/search-prompts";
 import type { ChatTurnMessage } from "./meta-search-types";
 
-const createSuggestionGeneratorPrompt = (maxQuestions: number = 4) => `
-You are an AI suggestion generator for an AI powered search engine. You will be given a conversation below. You need to generate ${maxQuestions} suggestions based on the conversation. The suggestion should be relevant to the conversation that can be used by the user to ask the chat model for more information.
-You need to make sure the suggestions are relevant to the conversation and are helpful to the user. Keep a note that the user might use these suggestions to ask a chat model for more information.
-Make sure the suggestions are medium in length and are informative and relevant to the conversation.
-
-Provide these suggestions separated by newlines between the XML tags <suggestions> and </suggestions>. For example:
-
-<suggestions>
-Tell me more about SpaceX and their recent projects
-What is the latest news on SpaceX?
-Who is the CEO of SpaceX?
-</suggestions>
-
-Conversation:
-{chat_history}
-`;
-
 type SuggestionGeneratorInput = {
   chat_history: ChatTurnMessage[];
   maxQuestions?: number;
+  /**
+   * Optional user-authored replacement for {@link followUpSuggestionsPrompt}
+   * (edited in Settings → Search Settings). Blank or whitespace-only values
+   * fall back to the built-in template.
+   */
+  promptTemplate?: string;
 };
 
 const outputParser = new LineListOutputParser({
@@ -39,10 +29,15 @@ const generateSuggestions = async (
   llm: LanguageModel,
 ): Promise<string[]> => {
   const maxQuestions = input.maxQuestions ?? 4;
-  const prompt = createSuggestionGeneratorPrompt(maxQuestions).replace(
-    "{chat_history}",
-    formatChatHistoryAsString(input.chat_history),
-  );
+  const template = input.promptTemplate?.trim()
+    ? input.promptTemplate
+    : followUpSuggestionsPrompt;
+
+  const prompt = template
+    .split("{maxQuestions}")
+    .join(String(maxQuestions))
+    .split("{chat_history}")
+    .join(formatChatHistoryAsString(input.chat_history));
 
   const { text } = await generateText({
     model: llm,