@burtson-labs/bandit-engine 2.0.64 → 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();
     };
   }
 });
@@ -11015,8 +11042,8 @@ var init_chat_scroll_to_bottom_button = __esm({
       drawerOpen = false,
       isMobile = false
     }) => {
-      const verticalBuffer = isMobile ? 28 : 48;
-      const bottomOffset = Math.max(inputHeight + verticalBuffer, verticalBuffer + 64);
+      const verticalBuffer = isMobile ? 44 : 80;
+      const bottomOffset = Math.max(inputHeight + verticalBuffer, verticalBuffer + 72);
       return /* @__PURE__ */ (0, import_jsx_runtime6.jsx)(
         import_material4.IconButton,
         {
@@ -14491,7 +14518,7 @@ ${sourcesMarkdownList.join("\n")}`;
 });
 
 // src/components/StreamingMarkdown.tsx
-var import_react16, import_material9, import_react_markdown2, import_remark_gfm2, import_rehype_raw2, import_rehype_sanitize3, import_styles5, import_jsx_runtime12, StreamingMarkdown, StreamingMarkdown_default;
+var import_react16, import_material9, import_react_markdown2, import_remark_gfm2, import_rehype_raw2, import_rehype_sanitize3, import_styles5, import_jsx_runtime12, StreamingMarkdown, arePropsEqual, StreamingMarkdown_default;
 var init_StreamingMarkdown = __esm({
   "src/components/StreamingMarkdown.tsx"() {
     "use strict";
@@ -14925,8 +14952,8 @@ ${listMarkdown}`;
         {
           ref: containerRef,
           sx: {
-            // Base transition for minor layout changes
-            transition: "opacity 120ms ease-out, transform 120ms ease-out",
+            // Settle the whole block to full color a beat after streaming ends.
+            transition: "opacity 380ms ease-out, transform 220ms ease-out",
             "& .cursor": {
               display: showCursor ? "inline" : "none",
               animation: "blink 1s step-start infinite"
@@ -14934,16 +14961,18 @@ ${listMarkdown}`;
             "@keyframes blink": {
               "50%": { opacity: 0 }
             },
+            // Each newly-written word focuses in — blur→sharp, a tiny rise, and fade —
+            // like ink settling onto the page as the typewriter writes.
             "& .bl-fade-word": {
               opacity: 0,
-              animation: "bl-fade-in 420ms ease-out forwards"
+              animation: "bl-fade-in 480ms cubic-bezier(0.22, 1, 0.36, 1) forwards"
             },
             "@keyframes bl-fade-in": {
-              from: { opacity: 0, transform: "translateY(1.5px)" },
-              to: { opacity: 1, transform: "translateY(0)" }
+              from: { opacity: 0, filter: "blur(3px)", transform: "translateY(2px)" },
+              to: { opacity: 1, filter: "blur(0)", transform: "translateY(0)" }
             },
-            // Subtle fade-in for each render while streaming to reduce choppiness perception
-            opacity: isStreaming ? 0.985 : 1,
+            // Dimmed while streaming, then brightens to full color once the answer lands.
+            opacity: isStreaming ? 0.86 : 1,
             transform: isStreaming ? "translateY(0.25px)" : "none",
             // Reduce layout jumpiness between updates
             "& p:last-child": { marginBottom: 0 },
@@ -14969,7 +14998,17 @@ ${listMarkdown}`;
         }
       );
     };
-    StreamingMarkdown_default = StreamingMarkdown;
+    arePropsEqual = (prev, next) => {
+      if (prev.content !== next.content || prev.isStreaming !== next.isStreaming) {
+        return false;
+      }
+      const a = prev.sources;
+      const b = next.sources;
+      if (a === b) return true;
+      if (!a || !b || a.length !== b.length) return false;
+      return a.every((s, i) => s.id === b[i].id && s.name === b[i].name);
+    };
+    StreamingMarkdown_default = import_react16.default.memo(StreamingMarkdown, arePropsEqual);
   }
 });
 
@@ -25584,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 }),
@@ -25610,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) {
@@ -35290,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();
@@ -35747,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: [