@burtson-labs/bandit-engine 2.0.65 → 2.0.66

package/dist/index.js

@@ -710,6 +710,8 @@ var init_preferencesStore = __esm({
       sttEnabled: true,
       banditModelsEnabled: true,
       feedbackEnabled: true,
+      interests: [],
+      useKnowledgeForStarters: false,
       homeUrl: ""
     };
     sanitizePreferences = (preferences) => {
@@ -807,6 +809,8 @@ var init_preferencesStore = __esm({
             sttEnabled: typeof mergedPreferences.sttEnabled === "boolean" ? mergedPreferences.sttEnabled : defaultPreferences.sttEnabled,
             banditModelsEnabled: typeof mergedPreferences.banditModelsEnabled === "boolean" ? mergedPreferences.banditModelsEnabled : defaultPreferences.banditModelsEnabled,
             feedbackEnabled: typeof mergedPreferences.feedbackEnabled === "boolean" ? mergedPreferences.feedbackEnabled : defaultPreferences.feedbackEnabled,
+            interests: Array.isArray(mergedPreferences.interests) ? mergedPreferences.interests.filter((i) => typeof i === "string") : defaultPreferences.interests,
+            useKnowledgeForStarters: typeof mergedPreferences.useKnowledgeForStarters === "boolean" ? mergedPreferences.useKnowledgeForStarters : defaultPreferences.useKnowledgeForStarters,
             homeUrl: typeof mergedPreferences.homeUrl === "string" ? mergedPreferences.homeUrl : defaultPreferences.homeUrl
           };
           set({ preferences: validatedPreferences });
@@ -6620,7 +6624,7 @@ var init_util = __esm({
 });
 
 // src/prompts/getStableQuestionPrompt.ts
-var TOPICS, getRandomTopicOfInterest, getStableQuestionPrompt;
+var TOPICS, getRandomTopicOfInterest, pickDistinctRandomTopics, getStableQuestionPrompt;
 var init_getStableQuestionPrompt = __esm({
   "src/prompts/getStableQuestionPrompt.ts"() {
     "use strict";
@@ -6633,42 +6637,65 @@ var init_getStableQuestionPrompt = __esm({
       const randomTopic = TOPICS[randomIndex];
       return randomTopic;
     };
+    pickDistinctRandomTopics = (count) => {
+      const pool = [...TOPICS];
+      const picked = [];
+      const n = Math.min(Math.max(count, 0), pool.length);
+      for (let i = 0; i < n; i++) {
+        const idx = randomRange(0, pool.length - 1);
+        picked.push(pool[idx]);
+        pool.splice(idx, 1);
+      }
+      return picked;
+    };
     getStableQuestionPrompt = (args) => {
-      const { limit, topicOfInterest, modelSystemPrompt } = args;
+      const { limit, topicOfInterest, modelSystemPrompt, interests, knowledgeTopics, webSearchAvailable } = args;
       if (limit < 1 || limit > 10) {
         throw new Error("Limit must be between 1 and 10");
       }
       const seed = generateSeed();
-      let prompt = `You are a helpful assistant.
-
-The following seed uniquely identifies the topic: "${seed}"`;
+      const lines = [
+        `You are crafting fresh, engaging conversation starters for a chat app's home screen.`,
+        ``,
+        `Variation seed (make this set different from any previous set): "${seed}"`
+      ];
       if (modelSystemPrompt && modelSystemPrompt.trim()) {
-        prompt += `
-
-Based on this specialized assistant profile: "${modelSystemPrompt.trim()}"`;
+        lines.push(``, `Tailor the questions to this assistant's role and expertise: "${modelSystemPrompt.trim()}"`);
       }
-      prompt += `
-
-Generate ${limit} concise (5\u201320 words, try to use this entire range), natural-sounding questions a user might ask${modelSystemPrompt ? " this specialized assistant" : " an AI assistant"}. These should be:
-
-- Relevant to ${topicOfInterest}`;
-      if (modelSystemPrompt && modelSystemPrompt.trim()) {
-        prompt += `
-- Aligned with the assistant's specialized capabilities and knowledge area`;
-      }
-      prompt += `
-- Specific enough to be practical
-- Easy to understand and not abstract
-
-Do not:
-- Refer to yourself or use phrases like "As an AI..."
-- Include greetings, explanations, or personality
-- Include jokes, fiction, or quotes
-- Number, bullet, or otherwise prefix the questions with extra characters
-- Repeat the same idea phrased differently \u2014 each question must explore a distinct angle or subtopic
-
-Output only ${limit} questions \u2014 one per line, with no leading numbers, bullets, or prefixes.`;
-      return prompt.trim();
+      if (interests && interests.length) {
+        lines.push(
+          ``,
+          `The user is especially interested in: ${interests.join(", ")}. Lean toward these, but do NOT make every question about the same one.`
+        );
+      }
+      if (knowledgeTopics && knowledgeTopics.length) {
+        lines.push(
+          ``,
+          `The user keeps these documents in their library: ${knowledgeTopics.join("; ")}. Include one or two questions that draw on this material.`
+        );
+      }
+      lines.push(
+        ``,
+        `Spread the set across a VARIETY of these topics so it never feels repetitive (do not cluster on one): ${topicOfInterest}.`
+      );
+      if (webSearchAvailable) {
+        lines.push(
+          ``,
+          `This assistant can search the web for live information. Include 2\u20133 questions that invite current, up-to-date answers (latest news, recent developments, "today" / "this week") so the user discovers that capability.`
+        );
+      }
+      lines.push(
+        ``,
+        `Generate ${limit} concise (5\u201320 words), natural-sounding questions a user might ask${modelSystemPrompt ? " this specialized assistant" : " an AI assistant"}. Requirements:`,
+        `- Each question must explore a DISTINCT topic or angle \u2014 no two may be similar or rephrasings of each other.`,
+        `- Vary the type across the set: practical how-to, curious exploration, current/topical, creative, and learning.`,
+        `- Specific and concrete, easy to understand, not abstract.`,
+        ``,
+        `Do not refer to yourself, say "As an AI", add greetings, explanations, jokes, fiction, quotes, or number/bullet the lines.`,
+        ``,
+        `Output only ${limit} questions \u2014 one per line, with no leading numbers, bullets, or prefixes.`
+      );
+      return lines.join("\n").trim();
     };
   }
 });
@@ -25596,6 +25623,9 @@ var init_query_suggestion_picker = __esm({
     init_prompts();
     init_getStableQuestionPrompt();
     init_modelStore();
+    init_preferencesStore();
+    init_knowledgeStore();
+    init_mcpToolsStore();
     import_jsx_runtime25 = require("react/jsx-runtime");
     markdownComponents = {
       p: ({ node, ...props }) => /* @__PURE__ */ (0, import_jsx_runtime25.jsx)("span", { ...props }),
@@ -25622,13 +25652,19 @@ var init_query_suggestion_picker = __esm({
         if (hasGenerated.current || isLoading) return;
         hasGenerated.current = true;
         const currentModel = getCurrentModel();
+        const prefs = usePreferencesStore.getState().preferences;
+        const interests = (prefs.interests ?? []).filter((i) => i && i.trim());
+        const randomTopics = pickDistinctRandomTopics(interests.length ? 3 : 5);
+        const topicOfInterest = Array.from(/* @__PURE__ */ new Set([...interests, ...randomTopics])).join(", ") || getRandomTopicOfInterest();
+        const webSearchAvailable = useMCPToolsStore.getState().getEnabledTools().some((t) => t.name === "web_search");
+        const knowledgeTopics = prefs.useKnowledgeForStarters ? useKnowledgeStore.getState().docs.map((d) => d.name).filter((n) => n && n.trim()).slice(0, 12) : void 0;
         const args = {
-          // keep responses quick and snappy, server may be handling concurrent requests adjust as needed
           limit: 9,
-          // pick a random topic of interest from the list, consider using the users preference topics dynamically, otherwise get a random one
-          topicOfInterest: getRandomTopicOfInterest(),
-          // Pass the current model's system prompt to tailor suggestions
-          modelSystemPrompt: currentModel?.systemPrompt
+          topicOfInterest,
+          modelSystemPrompt: currentModel?.systemPrompt,
+          interests,
+          knowledgeTopics,
+          webSearchAvailable
         };
         generateConversationStarters(args).then((prompts) => {
           if (prompts.length > 0) {
@@ -35302,6 +35338,7 @@ var PersonalitiesTab_default = PersonalitiesTab;
 var import_react47 = require("react");
 var import_material38 = require("@mui/material");
 init_preferencesStore();
+init_getStableQuestionPrompt();
 init_modelStore();
 init_voiceStore();
 init_useFeatures();
@@ -35759,6 +35796,49 @@ var PreferencesTab = ({
         children: "Control which AI features are enabled to optimize performance for your device. Disabling features can help reduce resource usage on machines with limited capabilities."
       }
     ),
+    /* @__PURE__ */ (0, import_jsx_runtime38.jsxs)(import_material38.Paper, { sx: { p: { xs: 1.75, sm: 2.5 }, mb: { xs: 2, md: 3 } }, children: [
+      /* @__PURE__ */ (0, import_jsx_runtime38.jsxs)(import_material38.Box, { children: [
+        /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_material38.Typography, { variant: "h6", sx: { fontWeight: 600, color: "text.primary" }, children: "Conversation Starters" }),
+        /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_material38.Typography, { variant: "body2", color: "text.secondary", children: "Personalize the suggested prompts on your home screen. Pick the topics you care about \u2014 starters lean toward them (blended with fresh ones each time) instead of generic blanket questions." })
+      ] }),
+      /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_material38.Box, { sx: { mt: 2, display: "flex", flexWrap: "wrap", gap: 1 }, children: TOPICS.map((topic) => {
+        const selected = (preferences.interests ?? []).includes(topic);
+        return /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(
+          import_material38.Chip,
+          {
+            label: topic,
+            size: "small",
+            color: selected ? "primary" : "default",
+            variant: selected ? "filled" : "outlined",
+            onClick: () => {
+              const current = preferences.interests ?? [];
+              const next = selected ? current.filter((t) => t !== topic) : [...current, topic];
+              updatePreference("interests", next);
+            },
+            sx: { textTransform: "capitalize" }
+          },
+          topic
+        );
+      }) }),
+      /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_material38.Box, { sx: { mt: 2.5 }, children: /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(
+        import_material38.FormControlLabel,
+        {
+          control: /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(
+            import_material38.Switch,
+            {
+              checked: !!preferences.useKnowledgeForStarters,
+              onChange: (e) => updatePreference("useKnowledgeForStarters", e.target.checked),
+              color: "primary"
+            }
+          ),
+          label: /* @__PURE__ */ (0, import_jsx_runtime38.jsxs)(import_material38.Box, { textAlign: "left", children: [
+            /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_material38.Typography, { variant: "body1", sx: { fontWeight: 600, color: "text.primary" }, children: "Use my knowledge documents" }),
+            /* @__PURE__ */ (0, import_jsx_runtime38.jsx)(import_material38.Typography, { variant: "body2", color: "text.secondary", children: "Let some starters reference the documents in your library." })
+          ] }),
+          sx: { alignSelf: "flex-start", ml: 0 }
+        }
+      ) })
+    ] }),
     /* @__PURE__ */ (0, import_jsx_runtime38.jsxs)(import_material38.Paper, { sx: { p: { xs: 1.75, sm: 2.5 }, mb: { xs: 2, md: 3 } }, children: [
       /* @__PURE__ */ (0, import_jsx_runtime38.jsxs)(import_material38.Box, { sx: { display: "flex", flexDirection: { xs: "column", sm: "row" }, justifyContent: "space-between", gap: 2 }, children: [
         /* @__PURE__ */ (0, import_jsx_runtime38.jsxs)(import_material38.Box, { children: [