@burtson-labs/bandit-engine 2.0.40 → 2.0.42

package/dist/{chunk-QJYPWWA5.mjs → chunk-54ZQ3FSN.mjs}

@@ -1163,61 +1163,15 @@ var GatewayService = class {
    */
   chat(request) {
     const endpoint = request.provider === "ollama" ? `/api/${request.provider}/chat` : request.provider ? `/api/${request.provider}/chat/completions` : "/api/chat/completions";
-    const url = `${this._baseUrl}${endpoint}`;
+    const fallbackEndpoint = request.provider === "bandit" ? "/completions" : null;
     const normalizedModel = request.provider === "bandit" ? (() => {
       const trimmed = (request.model ?? "").replace(/^bandit:/, "").trim();
       return trimmed !== "" ? trimmed : "bandit-core-1";
     })() : request.model;
-    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
-    });
     const requestBody = { ...request, model: normalizedModel, stream: request.stream !== false };
     return new Observable6((observer) => {
       const controller = new AbortController();
-      const task = fetch(url, {
-        method: "POST",
-        headers: this._getHeaders(),
-        body: JSON.stringify(requestBody),
-        signal: controller.signal
-      });
-      task.then(async (response) => {
-        debugLogger.debug(`Gateway chat response status: ${response.status} for provider: ${request.provider || "default"}`);
-        if (!response.ok) {
-          let errorText = "";
-          let errorData = null;
-          try {
-            errorText = await response.text();
-            debugLogger.error("GatewayService chat error response body", {
-              status: response.status,
-              statusText: response.statusText,
-              url: response.url,
-              body: errorText
-            });
-          } catch (readError) {
-            debugLogger.error("GatewayService chat failed to read error response body", { error: readError });
-            errorText = `Request failed with status ${response.status}`;
-          }
-          try {
-            errorData = JSON.parse(errorText);
-            debugLogger.error("GatewayService chat parsed error payload", errorData);
-          } catch (parseError) {
-            debugLogger.error("GatewayService chat error payload was not valid JSON");
-            errorData = { message: errorText };
-          }
-          const error = this._createHttpError(
-            `POST ${url} failed: ${response.status} ${response.statusText ?? ""}`,
-            {
-              status: response.status,
-              statusText: response.statusText ?? "",
-              data: errorData,
-              url
-            }
-          );
-          throw error;
-        }
+      const handleStreamingResponse = async (response) => {
         const reader = response.body?.getReader();
         const decoder = new TextDecoder();
         let buffer = "";
@@ -1292,14 +1246,75 @@ var GatewayService = class {
           }).catch((err) => observer.error(err));
         };
         read();
-      }).catch((err) => {
-        debugLogger.error("GatewayService chat fetch error", {
-          error: err,
-          url,
-          provider: request.provider
+      };
+      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
         });
-        observer.error(err);
-      });
+        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"}`);
+          if (response.status === 404 && allowFallback && fallbackEndpoint) {
+            debugLogger.warn("GatewayService chat endpoint returned 404, attempting fallback route", {
+              provider: request.provider,
+              attemptedEndpoint: targetEndpoint,
+              fallbackEndpoint
+            });
+            sendRequest(fallbackEndpoint, false);
+            return;
+          }
+          if (!response.ok) {
+            let errorText = "";
+            let errorData = null;
+            try {
+              errorText = await response.text();
+              debugLogger.error("GatewayService chat error response body", {
+                status: response.status,
+                statusText: response.statusText,
+                url: response.url,
+                body: errorText
+              });
+            } catch (readError) {
+              debugLogger.error("GatewayService chat failed to read error response body", { error: readError });
+              errorText = `Request failed with status ${response.status}`;
+            }
+            try {
+              errorData = JSON.parse(errorText);
+              debugLogger.error("GatewayService chat parsed error payload", errorData);
+            } catch (parseError) {
+              debugLogger.error("GatewayService chat error payload was not valid JSON");
+              errorData = { message: errorText };
+            }
+            const error = this._createHttpError(
+              `POST ${url} failed: ${response.status} ${response.statusText ?? ""}`,
+              {
+                status: response.status,
+                statusText: response.statusText ?? "",
+                data: errorData,
+                url
+              }
+            );
+            throw error;
+          }
+          await handleStreamingResponse(response);
+        }).catch((err) => {
+          debugLogger.error("GatewayService chat fetch error", {
+            error: err,
+            url,
+            provider: request.provider
+          });
+          observer.error(err);
+        });
+      };
+      sendRequest(endpoint, true);
       return () => {
         try {
           controller.abort();
@@ -1432,18 +1447,46 @@ var GatewayService = class {
     );
   }
   _getHeaders() {
-    const token = this._tokenFactory();
+    const rawToken = this._tokenFactory();
     const headers = {
       "Content-Type": "application/json"
     };
-    if (token && token.trim() !== "") {
-      headers["Authorization"] = `Bearer ${token}`;
-      debugLogger.debug("Authorization header set with token");
-    } else {
+    if (!rawToken) {
       debugLogger.warn("GatewayService: No token found, skipping Authorization header");
+      return headers;
     }
+    const token = rawToken.trim();
+    if (token === "") {
+      debugLogger.warn("GatewayService: Token factory returned empty string");
+      return headers;
+    }
+    if (/^(Bearer|ApiKey)\s+/i.test(token)) {
+      headers["Authorization"] = token;
+      debugLogger.debug("GatewayService: Authorization header set with explicit scheme");
+      return headers;
+    }
+    if (this._isLikelyBanditApiKey(token)) {
+      headers["Authorization"] = `ApiKey ${token}`;
+      headers["X-Burtson-Api-Key"] = token;
+      debugLogger.debug("GatewayService: Authorization header set using API key");
+      return headers;
+    }
+    if (this._isLikelyJwt(token)) {
+      headers["Authorization"] = `Bearer ${token}`;
+      debugLogger.debug("GatewayService: Authorization header set using bearer token");
+      return headers;
+    }
+    headers["Authorization"] = `Bearer ${token}`;
+    debugLogger.debug("GatewayService: Authorization header defaulted to bearer scheme");
     return headers;
   }
+  _isLikelyJwt(token) {
+    const segments = token.split(".");
+    return segments.length === 3 && segments.every((segment) => segment.length > 0);
+  }
+  _isLikelyBanditApiKey(value) {
+    return /^bai_[a-z0-9]{10,}$/i.test(value);
+  }
   /**
    * Submit feedback to the gateway API
    */
@@ -2904,4 +2947,4 @@ export {
   AIProviderFactory,
   useAIProviderStore
 };
-//# sourceMappingURL=chunk-QJYPWWA5.mjs.map
+//# sourceMappingURL=chunk-54ZQ3FSN.mjs.map