@burtson-labs/bandit-engine 2.0.52 → 2.0.53

package/dist/index.mjs

@@ -1,25 +1,25 @@
 import {
   chat_default
-} from "./chunk-TSQCNHOX.mjs";
+} from "./chunk-GNE4TTSI.mjs";
 import {
   chat_provider_default
-} from "./chunk-MH7WFWCP.mjs";
+} from "./chunk-AVV7HDGR.mjs";
 import "./chunk-ONQMRE2G.mjs";
 import {
   management_default,
   useGatewayHealth,
   useGatewayMemory,
   useGatewayModels
-} from "./chunk-Y5N3NSTU.mjs";
-import "./chunk-YZ2HJFPQ.mjs";
-import "./chunk-37PEP5JK.mjs";
+} from "./chunk-O54PTFJM.mjs";
+import "./chunk-EULV5CHD.mjs";
+import "./chunk-UFSEYVRS.mjs";
 import "./chunk-EHNWQ4T3.mjs";
 import {
   defineCustomElement
 } from "./chunk-IXIM7BNO.mjs";
 import {
   chat_modal_default
-} from "./chunk-RSSJADDD.mjs";
+} from "./chunk-2BGORTWS.mjs";
 import {
   FeedbackButton,
   FeedbackModal,
@@ -37,7 +37,7 @@ import {
   useTTS,
   useVoiceStore,
   voiceService
-} from "./chunk-M3BEAMCC.mjs";
+} from "./chunk-NZKLKZJT.mjs";
 import {
   DEFAULT_TIER_FEATURES,
   FeatureFlagContext,
@@ -57,11 +57,11 @@ import {
   useVectorStore,
   vectorDatabaseService,
   vectorMigrationService
-} from "./chunk-QX6CO7TJ.mjs";
+} from "./chunk-557E5VZ2.mjs";
 import {
   usePackageSettingsStore
 } from "./chunk-7ZDS33S2.mjs";
-import "./chunk-BENL3EF2.mjs";
+import "./chunk-H3BYFEIE.mjs";
 import {
   DebugLogger,
   debugLogger

package/dist/management/management.js

@@ -1709,7 +1709,8 @@ var init_gateway_service = __esm({
                             index: 0,
                             delta: {
                               role: parsed.message.role,
-                              content: parsed.message.content
+                              content: parsed.message.content,
+                              tool_calls: parsed.message.tool_calls
                             },
                             finish_reason: parsed.done ? parsed.done_reason || "stop" : null
                           }]
@@ -2689,6 +2690,7 @@ var init_gateway_provider = __esm({
             }
           }
         }
+        const toolAwareRequest = request;
         const gatewayRequest = {
           model: request.model,
           messages,
@@ -2697,7 +2699,8 @@ var init_gateway_provider = __esm({
           max_tokens: request.maxTokens,
           provider: this.config.provider,
           // Only include top-level images for Ollama (fallback)
-          images: this.config.provider === "ollama" ? request.images : void 0
+          images: this.config.provider === "ollama" ? request.images : void 0,
+          tools: toolAwareRequest.tools?.length ? toolAwareRequest.tools : void 0
         };
         debugLogger.debug("Gateway provider chat request", {
           model: request.model,
@@ -2714,13 +2717,18 @@ var init_gateway_provider = __esm({
           }))
         });
         return this.gatewayService.chat(gatewayRequest).pipe(
-          (0, import_rxjs7.map)((response) => ({
-            message: {
-              content: response.choices?.[0]?.message?.content || response.choices?.[0]?.delta?.content || "",
-              role: "assistant"
-            },
-            done: response.choices?.[0]?.finish_reason === "stop" || response.choices?.[0]?.finish_reason === "length"
-          }))
+          (0, import_rxjs7.map)((response) => {
+            const choice = response.choices?.[0];
+            const toolCalls = choice?.message?.tool_calls ?? choice?.delta?.tool_calls;
+            return {
+              message: {
+                content: choice?.message?.content ?? choice?.delta?.content ?? "",
+                role: "assistant",
+                tool_calls: toolCalls
+              },
+              done: choice?.finish_reason === "stop" || choice?.finish_reason === "length" || choice?.finish_reason === "tool_calls"
+            };
+          })
         );
       }
       generate(request) {
@@ -20065,6 +20073,7 @@ var init_chat_messages = __esm({
       scrollTargetRef,
       responseStarted,
       isStreaming,
+      isThinking = false,
       isNetworkSlow = false,
       showInstantFeedback = true,
       selectedModel,
@@ -20081,6 +20090,7 @@ var init_chat_messages = __esm({
           const isLast = index === lastIndex;
           const isPlaceholder = entry.answer === "...";
           const showLoader = isLast && isStreaming && streamBuffer.trim() === "";
+          const showThinking = showLoader && isThinking;
           const content = isLast ? isStreaming ? streamBuffer || "" : isPlaceholder ? "" : entry.answer : entry.answer;
           const rawSources = entry.sourceFiles;
           const sourceSummaries = rawSources ? rawSources.filter((doc) => doc && typeof doc.name === "string" && doc.name.trim()).map((doc) => ({ id: doc.id || doc.name, name: doc.name.trim() })) : void 0;
@@ -20108,11 +20118,27 @@ var init_chat_messages = __esm({
                       pointerEvents: showLoader ? "auto" : "none",
                       zIndex: showLoader ? 1 : 0
                     },
-                    children: /* @__PURE__ */ (0, import_jsx_runtime32.jsx)(import_material32.Box, { sx: { display: "flex", alignItems: "center", minHeight: "40px", pl: 2 }, children: /* @__PURE__ */ (0, import_jsx_runtime32.jsxs)("div", { className: "typing-only", children: [
-                      /* @__PURE__ */ (0, import_jsx_runtime32.jsx)("span", { className: "dot" }),
-                      /* @__PURE__ */ (0, import_jsx_runtime32.jsx)("span", { className: "dot" }),
-                      /* @__PURE__ */ (0, import_jsx_runtime32.jsx)("span", { className: "dot" })
-                    ] }) })
+                    children: /* @__PURE__ */ (0, import_jsx_runtime32.jsxs)(import_material32.Box, { sx: { display: "flex", alignItems: "center", gap: 1, minHeight: "40px", pl: 2 }, children: [
+                      /* @__PURE__ */ (0, import_jsx_runtime32.jsxs)("div", { className: "typing-only", children: [
+                        /* @__PURE__ */ (0, import_jsx_runtime32.jsx)("span", { className: "dot" }),
+                        /* @__PURE__ */ (0, import_jsx_runtime32.jsx)("span", { className: "dot" }),
+                        /* @__PURE__ */ (0, import_jsx_runtime32.jsx)("span", { className: "dot" })
+                      ] }),
+                      showThinking && /* @__PURE__ */ (0, import_jsx_runtime32.jsx)(
+                        import_material32.Box,
+                        {
+                          component: "span",
+                          sx: {
+                            fontSize: "0.75rem",
+                            color: "text.secondary",
+                            opacity: 0.6,
+                            fontStyle: "italic",
+                            userSelect: "none"
+                          },
+                          children: "Thinking\u2026"
+                        }
+                      )
+                    ] })
                   }
                 ),
                 /* @__PURE__ */ (0, import_jsx_runtime32.jsx)(
@@ -22251,6 +22277,7 @@ var init_useAIProvider = __esm({
       setIsSubmitting,
       setResponseStarted,
       setIsStreaming,
+      setIsThinking,
       setResponse,
       setPastedImages,
       setPendingMessage,
@@ -22708,6 +22735,12 @@ ${protocol}`;
           let fullMessage = "";
           let latestDisplayMessage = "";
           let sawToolBlock = false;
+          const stripThinking = (text) => {
+            let result = text.replace(/<think>[\s\S]*?<\/think>/g, "");
+            const openIdx = result.indexOf("<think>");
+            if (openIdx !== -1) result = result.slice(0, openIdx);
+            return result.trimStart();
+          };
           const flushNow = () => {
             clearFlushTimer();
             if (!sawToolBlock) {
@@ -22735,13 +22768,16 @@ ${protocol}`;
           }
           const sub = stream.subscribe({
             next: (data) => {
-              if (!data?.message?.content) return;
-              fullMessage += data.message.content;
-              if (/```(?:tool_code|TOOL_CODE)/.test(fullMessage)) {
+              if (!data?.message?.content && !data?.message?.tool_calls) return;
+              if (data.message.content) fullMessage += data.message.content;
+              const inThinkBlock = /<think>/.test(fullMessage) && !/<think>[\s\S]*<\/think>/.test(fullMessage);
+              setIsThinking?.(inThinkBlock);
+              const visibleMessage = stripThinking(fullMessage);
+              if (/```(?:tool_code|TOOL_CODE)/.test(visibleMessage)) {
                 sawToolBlock = true;
                 clearFlushTimer();
               }
-              latestDisplayMessage = fullMessage;
+              latestDisplayMessage = visibleMessage;
               if (!sawToolBlock) {
                 scheduleFlush();
               }
@@ -22762,6 +22798,7 @@ ${protocol}`;
                 setResponse(partial);
               }
               setStreamBuffer("");
+              setIsThinking?.(false);
               setPendingMessage(null);
               setLogoVisible(false);
               if (onError) {
@@ -22770,7 +22807,8 @@ ${protocol}`;
             },
             complete: async () => {
               try {
-                latestDisplayMessage = fullMessage;
+                setIsThinking?.(false);
+                latestDisplayMessage = stripThinking(fullMessage);
                 if (!sawToolBlock) {
                   flushNow();
                 }
@@ -28132,6 +28170,7 @@ var init_chat2 = __esm({
       const [streamBuffer, setStreamBuffer] = (0, import_react55.useState)("");
       const [responseStarted, setResponseStarted] = (0, import_react55.useState)(false);
       const [isStreaming, setIsStreaming] = (0, import_react55.useState)(false);
+      const [isThinking, setIsThinking] = (0, import_react55.useState)(false);
       const initialLogoState = history.length === 0;
       const [logoVisible, setLogoVisible] = (0, import_react55.useState)(initialLogoState);
       const [logoShouldRender, setLogoShouldRender] = (0, import_react55.useState)(initialLogoState);
@@ -28615,6 +28654,7 @@ var init_chat2 = __esm({
         overrideComponentStatus: setComponentStatus,
         setIsSubmitting,
         setIsStreaming,
+        setIsThinking,
         setResponseStarted,
         setResponse,
         setStreamBuffer,
@@ -29021,6 +29061,7 @@ var init_chat2 = __esm({
                               chat_messages_default,
                               {
                                 isStreaming,
+                                isThinking,
                                 history,
                                 pendingMessage,
                                 streamBuffer,