@burtson-labs/bandit-engine 2.0.102 → 2.0.103

package/dist/index.mjs

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-R3HBSHLE.mjs";
+} from "./chunk-GM5V2RJS.mjs";
 import {
   chat_provider_default
 } from "./chunk-VTKRZNLI.mjs";
@@ -10,7 +10,7 @@ import {
   useGatewayHealth,
   useGatewayMemory,
   useGatewayModels
-} from "./chunk-7NXSMMCU.mjs";
+} from "./chunk-SWLZ7QJB.mjs";
 import "./chunk-KYC7CC6C.mjs";
 import "./chunk-HAUDGBUS.mjs";
 import "./chunk-BN3D45E2.mjs";

package/dist/management/management.js

@@ -20365,7 +20365,7 @@ var init_bandit_chat_logo = __esm({
 });
 
 // src/chat/source-chips.tsx
-var import_react43, import_material35, import_jsx_runtime35, parseWebSources, stripSourcesForDisplay, domainOf, SourceChip, SourceChips, source_chips_default;
+var import_react43, import_material35, import_jsx_runtime35, parseWebSources, stripSourcesForDisplay, linkifyCitations, domainOf, SourceChip, SourceChips, source_chips_default;
 var init_source_chips = __esm({
   "src/chat/source-chips.tsx"() {
     "use strict";
@@ -20400,6 +20400,14 @@ var init_source_chips = __esm({
       );
       return c.replace(/\s+$/u, "");
     };
+    linkifyCitations = (content, sources) => {
+      if (!content || sources.length === 0) return content;
+      const link = (text) => text.replace(/\[(\d{1,2})\]/g, (full, num) => {
+        const src = sources[parseInt(num, 10) - 1];
+        return src ? `[\\[${num}\\]](${src.url})` : full;
+      });
+      return content.split(/(```[\s\S]*?```)/g).map((seg, i) => i % 2 === 0 ? link(seg) : seg).join("");
+    };
     domainOf = (url) => {
       try {
         return new URL(url).hostname.replace(/^www\./, "");
@@ -20608,7 +20616,7 @@ var init_chat_messages = __esm({
                       /* @__PURE__ */ (0, import_jsx_runtime36.jsx)(
                         StreamingMarkdown_default,
                         {
-                          content: stripSourcesForDisplay(content),
+                          content: linkifyCitations(stripSourcesForDisplay(content), parseWebSources(content)),
                           isStreaming: isStreaming && isLast,
                           sources: sourceSummaries
                         }
@@ -24136,6 +24144,29 @@ _This link is temporary and expires in about ${mins} minutes._`;
                     try {
                       const toolResultsText = summarizableResults.map((r) => `## ${r.name}
 ${r.output}`).join("\n\n");
+                      const dedupeSources = (arr) => {
+                        const seen = /* @__PURE__ */ new Set();
+                        return arr.filter((s) => seen.has(s.url) ? false : (seen.add(s.url), true));
+                      };
+                      const citeDomain = (u) => {
+                        try {
+                          return new URL(u).hostname.replace(/^www\./, "");
+                        } catch {
+                          return u;
+                        }
+                      };
+                      const buildCitationGuide = () => {
+                        const cite = dedupeSources(collectedSources).slice(0, 6);
+                        if (!cite.length)
+                          return `
+
+Do NOT add citations, footnotes, superscript reference numbers, or a Sources list \u2014 sources are attached automatically.`;
+                        const list = cite.map((s, i) => `[${i + 1}] ${s.title || citeDomain(s.url)}`).join("\n");
+                        return `
+
+When a statement is supported by one of the sources below, add an inline citation right after it using that source's number in square brackets \u2014 e.g. [1] or [2]. Use ONLY these numbers, never invent one, never use superscripts, and do NOT write your own Sources/References list (the numbered list is attached automatically below your answer):
+${list}`;
+                      };
                       const MAX_CHAIN_ROUNDS = 4;
                       const enabledToolsForChain = getEnabledMCPToolsForAI();
                       const convo = [
@@ -24151,7 +24182,7 @@ ${r.output}`).join("\n\n");
 ${toolResultsText}
 ===END TOOL RESULTS===
 
-Use them to fully complete my original request. If you still need to take an action I asked for (for example, actually create a file I want to download), call the appropriate tool now with a \`\`\`tool_code\`\`\` block. Otherwise give your final answer. Do NOT add citations, footnotes, superscript reference numbers (e.g. \xB9 \xB2 \xB3), or a Sources/References/Citations list \u2014 the sources are attached automatically below your answer. Just write the answer naturally.`
+Use them to fully complete my original request. If you still need to take an action I asked for (for example, actually create a file I want to download), call the appropriate tool now with a \`\`\`tool_code\`\`\` block. Otherwise give your final answer.${buildCitationGuide()}`
                         }
                       ];
                       const streamTurn = (req) => new Promise((resolve) => {
@@ -24297,7 +24328,7 @@ That step failed: ${e instanceof Error ? e.message : String(e)}`);
 ${roundOut.join("\n\n")}
 ===END TOOL RESULTS===
 
-Now give your final answer to my original request, or call another tool if you still genuinely need to. Do NOT add citations, footnotes, superscript reference numbers, or a Sources list \u2014 sources are attached automatically.`
+Now give your final answer to my original request, or call another tool if you still genuinely need to.${buildCitationGuide()}`
                         });
                       }
                       setIsThinking?.(false);
@@ -24310,7 +24341,7 @@ Now give your final answer to my original request, or call another tool if you s
                         const sourcesMd = collectedSources.length ? `
 
 **Sources**
-${collectedSources.slice(0, 6).map((s) => {
+${dedupeSources(collectedSources).slice(0, 6).map((s) => {
                           let domain = s.url;
                           try {
                             domain = new URL(s.url).hostname.replace(/^www\./, "");