@burtson-labs/bandit-engine 2.0.168 → 2.1.1

package/dist/{chunk-GRPTCHFN.mjs → chunk-5KC42IIR.mjs}

@@ -1,6 +1,6 @@
 import {
   debugLogger
-} from "./chunk-HNXPDIWC.mjs";
+} from "./chunk-NI7BKLC7.mjs";
 
 // src/store/aiProviderStore.ts
 import { create as create2 } from "zustand";
@@ -9,7 +9,18 @@ import { create as create2 } from "zustand";
 import { map as map2 } from "rxjs";
 
 // src/services/ollama/ollama.service.ts
-import { catchError, from, lastValueFrom, map, Observable, of, shareReplay, switchMap, throwError, timeout } from "rxjs";
+import {
+  catchError,
+  from,
+  lastValueFrom,
+  map,
+  Observable,
+  of,
+  shareReplay,
+  switchMap,
+  throwError,
+  timeout
+} from "rxjs";
 
 // src/services/stream/streamLineParser.ts
 var SKIP = { kind: "skip" };
@@ -42,7 +53,9 @@ var StreamLineParser = class {
 // src/services/ollama/ollama.service.ts
 var handleError = () => (obs) => obs.pipe(
   switchMap(
-    (response) => response.ok ? of(response) : throwError(() => new Error(`Request failed: ${response.status} ${response.statusText}`))
+    (response) => response.ok ? of(response) : throwError(
+      () => new Error(`Request failed: ${response.status} ${response.statusText}`)
+    )
   )
 );
 var parseResponseBody = async (response, responseType) => {
@@ -97,7 +110,9 @@ var OllamaService = class {
       }
       return availablility;
     } catch (e) {
-      debugLogger.warn(`Service not available at ${this._baseUrl}, trying fallback URL: ${fallbackUrl}`);
+      debugLogger.warn(
+        `Service not available at ${this._baseUrl}, trying fallback URL: ${fallbackUrl}`
+      );
       try {
         availablility.url = fallbackUrl;
         availablility.isAvailable = await lastValueFrom(
@@ -124,7 +139,10 @@ var OllamaService = class {
       const task = fetch(url, {
         method: "POST",
         headers: this._getHeaders(),
-        body: JSON.stringify({ ...request, stream: request.stream === false ? false : true })
+        body: JSON.stringify({
+          ...request,
+          stream: request.stream === false ? false : true
+        })
       });
       task.then((response) => {
         this._throwErrorIfNotOk(url, response);
@@ -141,7 +159,9 @@ var OllamaService = class {
                   observer.next(JSON.parse(tail.text));
                 } catch (err) {
                   observer.error(err);
-                  debugLogger.error("Final chunk parsing error (generate):", { buffer });
+                  debugLogger.error("Final chunk parsing error (generate):", {
+                    buffer
+                  });
                 }
               }
               observer.complete();
@@ -238,18 +258,24 @@ var OllamaService = class {
     };
     const response = from(fetch(url, requestInit));
     const handleFetchError = response.pipe(handleError());
-    const data = handleFetchError.pipe(switchMap((res) => from(parseResponseBody(res, responseType))));
+    const data = handleFetchError.pipe(
+      switchMap((res) => from(parseResponseBody(res, responseType)))
+    );
     const result = data.pipe(shareReplay(1));
     return result;
   }
   _post(url, body) {
-    const response = from(fetch(url, {
-      method: "POST",
-      headers: this._getHeaders(),
-      body: JSON.stringify(body)
-    }));
+    const response = from(
+      fetch(url, {
+        method: "POST",
+        headers: this._getHeaders(),
+        body: JSON.stringify(body)
+      })
+    );
     const handleFetchError = response.pipe(handleError());
-    const json = handleFetchError.pipe(switchMap((res) => from(parseResponseBody(res, "json"))));
+    const json = handleFetchError.pipe(
+      switchMap((res) => from(parseResponseBody(res, "json")))
+    );
     const result = json.pipe(shareReplay(1));
     return result;
   }
@@ -261,27 +287,37 @@ var OllamaService = class {
   _getHeaders() {
     const token = this._tokenFactory();
     if (!token) {
-      debugLogger.warn("OllamaService: No token found, using empty string for Authorization header");
+      debugLogger.warn(
+        "OllamaService: No token found, using empty string for Authorization header"
+      );
     }
     return {
       "Content-Type": "application/json",
-      "Authorization": `Bearer ${token || ""}`
+      Authorization: `Bearer ${token || ""}`
     };
   }
 };
 
 // src/services/ai-provider/providers/deprecated.ts
 function deprecatedOpenAIProvider() {
-  debugLogger.warn('\u26A0\uFE0F  DEPRECATION WARNING: Direct OpenAI provider is deprecated. Please migrate to GatewayProvider with provider: "openai". See GATEWAY_MIGRATION_GUIDE.md for details.');
+  debugLogger.warn(
+    '\u26A0\uFE0F  DEPRECATION WARNING: Direct OpenAI provider is deprecated. Please migrate to GatewayProvider with provider: "openai". See GATEWAY_MIGRATION_GUIDE.md for details.'
+  );
 }
 function deprecatedAzureOpenAIProvider() {
-  debugLogger.warn('\u26A0\uFE0F  DEPRECATION WARNING: Direct Azure OpenAI provider is deprecated. Please migrate to GatewayProvider with provider: "azure-openai". See GATEWAY_MIGRATION_GUIDE.md for details.');
+  debugLogger.warn(
+    '\u26A0\uFE0F  DEPRECATION WARNING: Direct Azure OpenAI provider is deprecated. Please migrate to GatewayProvider with provider: "azure-openai". See GATEWAY_MIGRATION_GUIDE.md for details.'
+  );
 }
 function deprecatedAnthropicProvider() {
-  debugLogger.warn('\u26A0\uFE0F  DEPRECATION WARNING: Direct Anthropic provider is deprecated. Please migrate to GatewayProvider with provider: "anthropic". See GATEWAY_MIGRATION_GUIDE.md for details.');
+  debugLogger.warn(
+    '\u26A0\uFE0F  DEPRECATION WARNING: Direct Anthropic provider is deprecated. Please migrate to GatewayProvider with provider: "anthropic". See GATEWAY_MIGRATION_GUIDE.md for details.'
+  );
 }
 function deprecatedOllamaProvider() {
-  debugLogger.info('\u2139\uFE0F  INFO: Direct Ollama provider is available for local development. For production environments, consider using GatewayProvider with provider: "ollama" for enhanced management. See GATEWAY_MIGRATION_GUIDE.md for details.');
+  debugLogger.info(
+    '\u2139\uFE0F  INFO: Direct Ollama provider is available for local development. For production environments, consider using GatewayProvider with provider: "ollama" for enhanced management. See GATEWAY_MIGRATION_GUIDE.md for details.'
+  );
 }
 
 // src/services/ai-provider/providers/ollama.provider.ts
@@ -297,10 +333,7 @@ var OllamaProvider = class {
       finalBaseUrl: baseUrl,
       hasTokenFactory: !!config.tokenFactory
     });
-    this.ollamaService = new OllamaService(
-      baseUrl,
-      config.tokenFactory || (() => null)
-    );
+    this.ollamaService = new OllamaService(baseUrl, config.tokenFactory || (() => null));
   }
   chat(request) {
     const ollamaRequest = {
@@ -339,13 +372,15 @@ var OllamaProvider = class {
   }
   listModels() {
     return this.ollamaService.listModels().pipe(
-      map2((models) => models.map((model) => ({
-        name: model.name,
-        size: model.size,
-        details: model.details,
-        digest: model.digest,
-        modified_at: model.modified_at
-      })))
+      map2(
+        (models) => models.map((model) => ({
+          name: model.name,
+          size: model.size,
+          details: model.details,
+          digest: model.digest,
+          modified_at: model.modified_at
+        }))
+      )
     );
   }
   async validateServiceAvailability(args) {
@@ -403,9 +438,11 @@ var OpenAIProvider = class {
   }
   listModels() {
     const url = `${this.baseUrl}/models`;
-    return from2(fetch(url, {
-      headers: this.getHeaders()
-    })).pipe(
+    return from2(
+      fetch(url, {
+        headers: this.getHeaders()
+      })
+    ).pipe(
       switchMap2((response) => {
         if (!response.ok) {
           return throwError2(() => new Error(`Failed to list models: ${response.status}`));
@@ -520,7 +557,10 @@ var OpenAIProvider = class {
                     });
                   }
                 } catch (err) {
-                  debugLogger.error("Error parsing OpenAI stream data:", { data, error: err });
+                  debugLogger.error("Error parsing OpenAI stream data:", {
+                    data,
+                    error: err
+                  });
                 }
               }
             }
@@ -532,14 +572,16 @@ var OpenAIProvider = class {
     });
   }
   nonStreamChatRequest(url, payload) {
-    return from2(fetch(url, {
-      method: "POST",
-      headers: {
-        ...this.getHeaders(),
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(payload)
-    })).pipe(
+    return from2(
+      fetch(url, {
+        method: "POST",
+        headers: {
+          ...this.getHeaders(),
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(payload)
+      })
+    ).pipe(
       switchMap2((response) => {
         if (!response.ok) {
           return throwError2(() => new Error(`OpenAI request failed: ${response.status}`));
@@ -728,7 +770,10 @@ var AzureOpenAIProvider = class {
                     });
                   }
                 } catch (err) {
-                  debugLogger.error("Error parsing Azure OpenAI stream data:", { data, error: err });
+                  debugLogger.error("Error parsing Azure OpenAI stream data:", {
+                    data,
+                    error: err
+                  });
                 }
               }
             }
@@ -740,17 +785,21 @@ var AzureOpenAIProvider = class {
     });
   }
   nonStreamChatRequest(url, payload) {
-    return from3(fetch(url, {
-      method: "POST",
-      headers: {
-        ...this.getHeaders(),
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(payload)
-    })).pipe(
+    return from3(
+      fetch(url, {
+        method: "POST",
+        headers: {
+          ...this.getHeaders(),
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(payload)
+      })
+    ).pipe(
       switchMap3((response) => {
         if (!response.ok) {
-          return throwError3(() => new Error(`Azure OpenAI request failed: ${response.status}`));
+          return throwError3(
+            () => new Error(`Azure OpenAI request failed: ${response.status}`)
+          );
         }
         return from3(response.json());
       }),
@@ -979,7 +1028,10 @@ var AnthropicProvider = class {
                     });
                   }
                 } catch (err) {
-                  debugLogger.error("Error parsing Anthropic stream data:", { data, error: err });
+                  debugLogger.error("Error parsing Anthropic stream data:", {
+                    data,
+                    error: err
+                  });
                 }
               }
             }
@@ -991,17 +1043,21 @@ var AnthropicProvider = class {
     });
   }
   nonStreamChatRequest(url, payload) {
-    return from4(fetch(url, {
-      method: "POST",
-      headers: {
-        ...this.getHeaders(),
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(payload)
-    })).pipe(
+    return from4(
+      fetch(url, {
+        method: "POST",
+        headers: {
+          ...this.getHeaders(),
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(payload)
+      })
+    ).pipe(
       switchMap4((response) => {
         if (!response.ok) {
-          return throwError4(() => new Error(`Anthropic request failed: ${response.status}`));
+          return throwError4(
+            () => new Error(`Anthropic request failed: ${response.status}`)
+          );
         }
         return from4(response.json());
       }),
@@ -1053,9 +1109,42 @@ var AnthropicProvider = class {
 // src/services/ai-provider/providers/gateway.provider.ts
 import { map as map12 } from "rxjs";
 
+// src/services/ai-provider/context-sources.ts
+var MAX_LABEL = 120;
+var clean = (value) => {
+  if (typeof value !== "string") return void 0;
+  const trimmed = value.trim();
+  if (!trimmed) return void 0;
+  return trimmed.length > MAX_LABEL ? `${trimmed.slice(0, MAX_LABEL - 1)}\u2026` : trimmed;
+};
+var readContextSourcesField = (chunk) => {
+  if (!chunk || typeof chunk !== "object") return void 0;
+  const record = chunk;
+  const raw = record.context_sources ?? record.contextSources;
+  if (!Array.isArray(raw)) return void 0;
+  const out = [];
+  for (const entry of raw) {
+    if (!entry || typeof entry !== "object") continue;
+    const label = clean(entry.label);
+    if (!label) continue;
+    const detail = clean(entry.detail);
+    out.push(detail ? { label, detail } : { label });
+  }
+  return out.length ? out : void 0;
+};
+
 // src/services/gateway/gateway.service.ts
 import axios, { AxiosHeaders } from "axios";
-import { catchError as catchError2, from as from5, lastValueFrom as lastValueFrom2, map as map6, Observable as Observable6, of as of2, shareReplay as shareReplay2, timeout as timeout2 } from "rxjs";
+import {
+  catchError as catchError2,
+  from as from5,
+  lastValueFrom as lastValueFrom2,
+  map as map6,
+  Observable as Observable6,
+  of as of2,
+  shareReplay as shareReplay2,
+  timeout as timeout2
+} from "rxjs";
 
 // src/store/authenticationStore.ts
 import { create } from "zustand";
@@ -1122,9 +1211,11 @@ function emitAuthTokenChanged(token) {
   if (typeof window === "undefined") {
     return;
   }
-  window.dispatchEvent(new CustomEvent(AUTH_TOKEN_CHANGED_EVENT, {
-    detail: { token }
-  }));
+  window.dispatchEvent(
+    new CustomEvent(AUTH_TOKEN_CHANGED_EVENT, {
+      detail: { token }
+    })
+  );
 }
 var readLocalStorage = () => {
   try {
@@ -1238,8 +1329,10 @@ var AuthenticationService = class {
     }
   }
   parseJwtClaims(token) {
+    const parts = token ? token.split(".") : [];
+    if (parts.length !== 3) return null;
     try {
-      const base64Url = token.split(".")[1];
+      const base64Url = parts[1];
       const base64 = base64Url.replace(/-/g, "+").replace(/_/g, "/");
       const jsonPayload = decodeURIComponent(
         atob(base64).split("").map((c) => {
@@ -1323,7 +1416,8 @@ var GatewayService = class {
    */
   async _retryAfterRefresh(instance, error) {
     const config = error.config;
-    if (!config || config._banditAuthRetried || error.response?.status !== 401) return null;
+    if (!config || config._banditAuthRetried || error.response?.status !== 401)
+      return null;
     if (!authenticationService.canRefresh()) return null;
     const data = error.response?.data;
     if (data?.code && data.code !== "TOKEN_EXPIRED") return null;
@@ -1418,7 +1512,9 @@ var GatewayService = class {
       return availability;
     } catch (e) {
       if (fallbackUrl) {
-        debugLogger.warn(`Gateway service not available at ${this._baseUrl}, trying fallback URL: ${fallbackUrl}`);
+        debugLogger.warn(
+          `Gateway service not available at ${this._baseUrl}, trying fallback URL: ${fallbackUrl}`
+        );
         try {
           availability.url = fallbackUrl.replace(/\/$/, "");
           availability.isAvailable = await lastValueFrom2(
@@ -1434,7 +1530,9 @@ var GatewayService = class {
           this._setBaseUrl(availability.url);
           return availability;
         } catch (e2) {
-          debugLogger.error(`Gateway service not available at fallback URL: ${fallbackUrl}`);
+          debugLogger.error(
+            `Gateway service not available at fallback URL: ${fallbackUrl}`
+          );
           throw e2;
         }
       } else {
@@ -1460,7 +1558,11 @@ var GatewayService = class {
       const trimmed = (request.model ?? "").replace(/^bandit:/, "").trim();
       return trimmed !== "" ? trimmed : "bandit-core-1";
     })() : request.model;
-    const requestBody = { ...request, model: normalizedModel, stream: request.stream !== false };
+    const requestBody = {
+      ...request,
+      model: normalizedModel,
+      stream: request.stream !== false
+    };
     return new Observable6((observer) => {
       const controller = new AbortController();
       const handleStreamingResponse = async (response) => {
@@ -1477,7 +1579,10 @@ var GatewayService = class {
                   const finalResponse = JSON.parse(tail.text);
                   observer.next(finalResponse);
                 } catch (err) {
-                  debugLogger.error("GatewayService chat final chunk parsing error", { buffer, error: err });
+                  debugLogger.error("GatewayService chat final chunk parsing error", {
+                    buffer,
+                    error: err
+                  });
                   observer.error(err);
                 }
               }
@@ -1502,17 +1607,21 @@ var GatewayService = class {
                     const transformed = {
                       id: `chatcmpl-${Date.now()}`,
                       object: "chat.completion.chunk",
-                      created: Math.floor(new Date(parsed.created_at || Date.now()).getTime() / 1e3),
+                      created: Math.floor(
+                        new Date(parsed.created_at || Date.now()).getTime() / 1e3
+                      ),
                       model: parsed.model,
-                      choices: [{
-                        index: 0,
-                        delta: {
-                          role: parsed.message.role,
-                          content: parsed.message.content,
-                          tool_calls: parsed.message.tool_calls
-                        },
-                        finish_reason: parsed.done ? parsed.done_reason || "stop" : null
-                      }]
+                      choices: [
+                        {
+                          index: 0,
+                          delta: {
+                            role: parsed.message.role,
+                            content: parsed.message.content,
+                            tool_calls: parsed.message.tool_calls
+                          },
+                          finish_reason: parsed.done ? parsed.done_reason || "stop" : null
+                        }
+                      ]
                     };
                     if (parsed.done && parsed.total_duration) {
                       transformed.usage = {
@@ -1542,25 +1651,33 @@ var GatewayService = class {
       };
       const sendRequest = (targetEndpoint, allowFallback) => {
         const url = `${this._baseUrl}${targetEndpoint}`;
-        debugLogger.debug(`Gateway chat request to ${url} with provider: ${request.provider || "default"}`, {
-          model: normalizedModel,
-          messageCount: request.messages.length,
-          hasImages: !!(request.images && request.images.length > 0),
-          imageCount: request.images?.length || 0
-        });
+        debugLogger.debug(
+          `Gateway chat request to ${url} with provider: ${request.provider || "default"}`,
+          {
+            model: normalizedModel,
+            messageCount: request.messages.length,
+            hasImages: !!(request.images && request.images.length > 0),
+            imageCount: request.images?.length || 0
+          }
+        );
         fetch(url, {
           method: "POST",
           headers: this._getHeaders(),
           body: JSON.stringify(requestBody),
           signal: controller.signal
         }).then(async (response) => {
-          debugLogger.debug(`Gateway chat response status: ${response.status} for provider: ${request.provider || "default"}`);
+          debugLogger.debug(
+            `Gateway chat response status: ${response.status} for provider: ${request.provider || "default"}`
+          );
           if (response.status === 404 && allowFallback && fallbackEndpoint) {
-            debugLogger.warn("GatewayService chat endpoint returned 404, attempting fallback route", {
-              provider: request.provider,
-              attemptedEndpoint: targetEndpoint,
-              fallbackEndpoint
-            });
+            debugLogger.warn(
+              "GatewayService chat endpoint returned 404, attempting fallback route",
+              {
+                provider: request.provider,
+                attemptedEndpoint: targetEndpoint,
+                fallbackEndpoint
+              }
+            );
             sendRequest(fallbackEndpoint, false);
             return;
           }
@@ -1576,7 +1693,10 @@ var GatewayService = class {
                 body: errorText
               });
             } catch (readError) {
-              debugLogger.error("GatewayService chat failed to read error response body", { error: readError });
+              debugLogger.error(
+                "GatewayService chat failed to read error response body",
+                { error: readError }
+              );
               errorText = `Request failed with status ${response.status}`;
             }
             try {
@@ -1626,14 +1746,21 @@ var GatewayService = class {
       const trimmed = (request.model ?? "").replace(/^bandit:/, "").trim();
       return trimmed !== "" ? trimmed : "bandit-core-1";
     })() : request.model;
-    debugLogger.debug(`Gateway generate request to ${url} with provider: ${request.provider || "default"}`, {
-      model: normalizedModel
-    });
+    debugLogger.debug(
+      `Gateway generate request to ${url} with provider: ${request.provider || "default"}`,
+      {
+        model: normalizedModel
+      }
+    );
     return new Observable6((observer) => {
       const task = fetch(url, {
         method: "POST",
         headers: this._getHeaders(),
-        body: JSON.stringify({ ...request, model: normalizedModel, stream: request.stream !== false })
+        body: JSON.stringify({
+          ...request,
+          model: normalizedModel,
+          stream: request.stream !== false
+        })
       });
       task.then(async (response) => {
         if (!response.ok) {
@@ -1673,7 +1800,9 @@ var GatewayService = class {
                   observer.next(JSON.parse(tail.text));
                 } catch (err) {
                   observer.error(err);
-                  debugLogger.error("Final chunk parsing error (gateway generate):", { buffer });
+                  debugLogger.error("Final chunk parsing error (gateway generate):", {
+                    buffer
+                  });
                 }
               }
               observer.complete();
@@ -1693,7 +1822,9 @@ var GatewayService = class {
                   observer.next(JSON.parse(classified.text));
                 } catch (err) {
                   observer.error(err);
-                  debugLogger.error("Error parsing JSON line (gateway generate):", { line });
+                  debugLogger.error("Error parsing JSON line (gateway generate):", {
+                    line
+                  });
                 }
               }
             }
@@ -1817,7 +1948,9 @@ var GatewayService = class {
    * Generate a mailto URL as fallback for feedback submission
    */
   _generateMailtoUrl(feedback) {
-    const subject = encodeURIComponent(`[${feedback.category.toUpperCase()}] ${feedback.title}`);
+    const subject = encodeURIComponent(
+      `[${feedback.category.toUpperCase()}] ${feedback.title}`
+    );
     let body = `Category: ${feedback.category}
 `;
     body += `Priority: ${feedback.priority}
@@ -2341,7 +2474,9 @@ var GatewayProvider = class {
       throw new Error("Gateway provider requires gatewayUrl in config");
     }
     if (!config.provider) {
-      throw new Error("Gateway provider requires provider field to specify backend (openai, azure-openai, anthropic, ollama)");
+      throw new Error(
+        "Gateway provider requires provider field to specify backend (openai, azure-openai, anthropic, ollama)"
+      );
     }
     const tokenFactory = config.tokenFactory || (() => null);
     this.gatewayService = new GatewayService(config.gatewayUrl, tokenFactory);
@@ -2358,7 +2493,9 @@ var GatewayProvider = class {
         break;
       case "azure-openai":
         if (!this.config.deploymentName || !this.config.apiVersion) {
-          throw new Error("Azure OpenAI gateway provider requires deploymentName and apiVersion");
+          throw new Error(
+            "Azure OpenAI gateway provider requires deploymentName and apiVersion"
+          );
         }
         this.providerSpecificService = new AzureOpenAIGatewayService(
           gatewayUrl,
@@ -2370,10 +2507,16 @@ var GatewayProvider = class {
         );
         break;
       case "anthropic":
-        this.providerSpecificService = new AnthropicGatewayService(gatewayUrl, tokenFactory);
+        this.providerSpecificService = new AnthropicGatewayService(
+          gatewayUrl,
+          tokenFactory
+        );
         break;
       case "bandit":
-        this.providerSpecificService = new BanditAIGatewayService(gatewayUrl, tokenFactory);
+        this.providerSpecificService = new BanditAIGatewayService(
+          gatewayUrl,
+          tokenFactory
+        );
         break;
       case "ollama":
         this.providerSpecificService = new OllamaGatewayService(gatewayUrl, tokenFactory);
@@ -2411,7 +2554,9 @@ var GatewayProvider = class {
             images: request.images
           };
         }
-      } else if (["openai", "azure-openai", "anthropic", "bandit", "playground"].includes(this.config.provider || "")) {
+      } else if (["openai", "azure-openai", "anthropic", "bandit", "playground"].includes(
+        this.config.provider || ""
+      )) {
         if (lastUserMessageIndex !== -1) {
           const currentMessage = messages[lastUserMessageIndex];
           const contentArray = [
@@ -2460,7 +2605,9 @@ var GatewayProvider = class {
       stream: request.stream,
       hasImages: !!(request.images && request.images.length > 0),
       imageCount: request.images?.length || 0,
-      imageStrategy: this.config.provider === "ollama" ? "message-level-array" : ["openai", "azure-openai", "anthropic", "playground"].includes(this.config.provider || "") ? "structured-content" : "top-level-fallback",
+      imageStrategy: this.config.provider === "ollama" ? "message-level-array" : ["openai", "azure-openai", "anthropic", "playground"].includes(
+        this.config.provider || ""
+      ) ? "structured-content" : "top-level-fallback",
       finalMessages: messages.map((m) => ({
         role: m.role,
         hasImages: Array.isArray(m.images) && m.images.length > 0,
@@ -2471,12 +2618,14 @@ var GatewayProvider = class {
       map12((response) => {
         const choice = response.choices?.[0];
         const toolCalls = choice?.message?.tool_calls ?? choice?.delta?.tool_calls;
+        const contextSources = readContextSourcesField(response);
         return {
           message: {
             content: choice?.message?.content ?? choice?.delta?.content ?? "",
             role: "assistant",
             tool_calls: toolCalls
           },
+          ...contextSources ? { contextSources } : {},
           done: choice?.finish_reason === "stop" || choice?.finish_reason === "length" || choice?.finish_reason === "tool_calls"
         };
       })
@@ -2503,26 +2652,32 @@ var GatewayProvider = class {
     );
   }
   listModels() {
-    debugLogger.debug("Gateway provider listing models", { provider: this.config.provider });
+    debugLogger.debug("Gateway provider listing models", {
+      provider: this.config.provider
+    });
     if (this.config.provider) {
       return this.gatewayService.listModelsByProvider(this.config.provider).pipe(
-        map12((models) => models.map((model) => ({
-          name: model.id || model.name,
-          size: model.size,
-          details: model.details,
-          digest: model.digest,
-          modified_at: model.modified_at
-        })))
+        map12(
+          (models) => models.map((model) => ({
+            name: model.id || model.name,
+            size: model.size,
+            details: model.details,
+            digest: model.digest,
+            modified_at: model.modified_at
+          }))
+        )
       );
     } else {
       return this.gatewayService.listModels().pipe(
-        map12((models) => models.map((model) => ({
-          name: model.id || model.name,
-          size: model.size,
-          details: model.details,
-          digest: model.digest,
-          modified_at: model.modified_at
-        })))
+        map12(
+          (models) => models.map((model) => ({
+            name: model.id || model.name,
+            size: model.size,
+            details: model.details,
+            digest: model.digest,
+            modified_at: model.modified_at
+          }))
+        )
       );
     }
   }
@@ -2663,9 +2818,7 @@ var PlaygroundProvider = class {
   }
   chat(request) {
     const lastUserMessage = [...request.messages].reverse().find((m) => m.role === "user")?.content ?? "";
-    const script = PLAYGROUND_SCRIPTS.find(
-      (scenario) => scenario.match(lastUserMessage)
-    );
+    const script = PLAYGROUND_SCRIPTS.find((scenario) => scenario.match(lastUserMessage));
     const responseText = script?.response ?? FALLBACK_RESPONSE;
     const chunks = this.splitIntoChunks(responseText);
     return new Observable8((observer) => {
@@ -2774,9 +2927,11 @@ var XAIProvider = class {
   }
   listModels() {
     const url = `${this.baseUrl}/models`;
-    return from6(fetch(url, {
-      headers: this.getHeaders()
-    })).pipe(
+    return from6(
+      fetch(url, {
+        headers: this.getHeaders()
+      })
+    ).pipe(
       switchMap5((response) => {
         if (!response.ok) {
           return throwError5(() => new Error(`Failed to list models: ${response.status}`));
@@ -2892,7 +3047,10 @@ var XAIProvider = class {
                     });
                   }
                 } catch (err) {
-                  debugLogger.error("Error parsing xAI stream data:", { data, error: err });
+                  debugLogger.error("Error parsing xAI stream data:", {
+                    data,
+                    error: err
+                  });
                 }
               }
             }
@@ -2904,14 +3062,16 @@ var XAIProvider = class {
     });
   }
   nonStreamChatRequest(url, payload) {
-    return from6(fetch(url, {
-      method: "POST",
-      headers: {
-        ...this.getHeaders(),
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(payload)
-    })).pipe(
+    return from6(
+      fetch(url, {
+        method: "POST",
+        headers: {
+          ...this.getHeaders(),
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(payload)
+      })
+    ).pipe(
       switchMap5((response) => {
         if (!response.ok) {
           return throwError5(() => new Error(`xAI request failed: ${response.status}`));
@@ -3026,8 +3186,13 @@ var BanditAIProvider = class {
     return from7(fetch(url, { headers: this.getHeaders() })).pipe(
       switchMap6((response) => {
         if (!response.ok) {
-          debugLogger.error("BanditAI listModels failed", { status: response.status, url });
-          return throwError6(() => new Error(`Failed to list Bandit models: ${response.status}`));
+          debugLogger.error("BanditAI listModels failed", {
+            status: response.status,
+            url
+          });
+          return throwError6(
+            () => new Error(`Failed to list Bandit models: ${response.status}`)
+          );
         }
         return from7(response.json());
       }),
@@ -3098,7 +3263,10 @@ var BanditAIProvider = class {
         const read = () => {
           reader?.read().then(({ done, value }) => {
             if (done) {
-              observer.next({ message: { content: "", role: "assistant" }, done: true });
+              observer.next({
+                message: { content: "", role: "assistant" },
+                done: true
+              });
               observer.complete();
               return;
             }
@@ -3112,18 +3280,29 @@ var BanditAIProvider = class {
               }
               const data = line.slice(6).trim();
               if (data === "[DONE]") {
-                observer.next({ message: { content: "", role: "assistant" }, done: true });
+                observer.next({
+                  message: { content: "", role: "assistant" },
+                  done: true
+                });
                 observer.complete();
                 return;
               }
               try {
                 const parsed = JSON.parse(data);
                 const content = parsed.choices?.[0]?.delta?.content ?? "";
-                if (content) {
-                  observer.next({ message: { content, role: "assistant" }, done: false });
+                const contextSources = readContextSourcesField(parsed);
+                if (content || contextSources) {
+                  observer.next({
+                    message: { content, role: "assistant" },
+                    ...contextSources ? { contextSources } : {},
+                    done: false
+                  });
                 }
               } catch (error) {
-                debugLogger.error("BanditAI stream chunk parse failure", { data, error });
+                debugLogger.error("BanditAI stream chunk parse failure", {
+                  data,
+                  error
+                });
               }
             }
             read();
@@ -3134,17 +3313,21 @@ var BanditAIProvider = class {
     });
   }
   nonStreamChatRequest(url, payload) {
-    return from7(fetch(url, {
-      method: "POST",
-      headers: {
-        ...this.getHeaders(),
-        "Content-Type": "application/json"
-      },
-      body: JSON.stringify(payload)
-    })).pipe(
+    return from7(
+      fetch(url, {
+        method: "POST",
+        headers: {
+          ...this.getHeaders(),
+          "Content-Type": "application/json"
+        },
+        body: JSON.stringify(payload)
+      })
+    ).pipe(
       switchMap6((response) => {
         if (!response.ok) {
-          return throwError6(() => new Error(`BanditAI request failed: ${response.status}`));
+          return throwError6(
+            () => new Error(`BanditAI request failed: ${response.status}`)
+          );
         }
         return from7(response.json());
       }),
@@ -3258,8 +3441,9 @@ export {
   TOKEN_KEY2 as TOKEN_KEY,
   AUTH_TOKEN_CHANGED_EVENT,
   authenticationService,
+  readContextSourcesField,
   GatewayService,
   AIProviderFactory,
   useAIProviderStore
 };
-//# sourceMappingURL=chunk-GRPTCHFN.mjs.map
+//# sourceMappingURL=chunk-5KC42IIR.mjs.map