@burtson-labs/bandit-engine 2.0.101 → 2.0.103

package/dist/{chat-PMO3WTM4.mjs → chat-HCZEF3ZM.mjs}

@@ -1,6 +1,6 @@
 import {
   chat_default
-} from "./chunk-R3HBSHLE.mjs";
+} from "./chunk-GM5V2RJS.mjs";
 import "./chunk-ONQMRE2G.mjs";
 import "./chunk-KYC7CC6C.mjs";
 import "./chunk-BN3D45E2.mjs";
@@ -13,4 +13,4 @@ import "./chunk-BJTO5JO5.mjs";
 export {
   chat_default as default
 };
-//# sourceMappingURL=chat-PMO3WTM4.mjs.map
+//# sourceMappingURL=chat-HCZEF3ZM.mjs.map

package/dist/{chunk-R3HBSHLE.mjs → chunk-GM5V2RJS.mjs}

@@ -436,6 +436,14 @@ var stripSourcesForDisplay = (content) => {
   );
   return c.replace(/\s+$/u, "");
 };
+var 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("");
+};
 var domainOf = (url) => {
   try {
     return new URL(url).hostname.replace(/^www\./, "");
@@ -633,7 +641,7 @@ var ChatMessages = ({
                   /* @__PURE__ */ jsx6(
                     StreamingMarkdown_default,
                     {
-                      content: stripSourcesForDisplay(content),
+                      content: linkifyCitations(stripSourcesForDisplay(content), parseWebSources(content)),
                       isStreaming: isStreaming && isLast,
                       sources: sourceSummaries
                     }
@@ -4017,6 +4025,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 = [
@@ -4032,7 +4063,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) => {
@@ -4178,7 +4209,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);
@@ -4191,7 +4222,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\./, "");
@@ -10663,4 +10694,4 @@ var chat_default = Chat;
 export {
   chat_default
 };
-//# sourceMappingURL=chunk-R3HBSHLE.mjs.map
+//# sourceMappingURL=chunk-GM5V2RJS.mjs.map