@burtson-labs/bandit-engine 2.0.167 → 2.0.168

package/dist/management/management.mjs

@@ -1,12 +1,12 @@
 import {
   management_default
-} from "../chunk-PK3LFBGC.mjs";
-import "../chunk-5BZWJQ5O.mjs";
+} from "../chunk-4SXXEGEK.mjs";
+import "../chunk-JD6C64CT.mjs";
 import "../chunk-CABOLDNR.mjs";
 import "../chunk-E5WMDLE3.mjs";
-import "../chunk-N4BQ6HKL.mjs";
-import "../chunk-KDIL5WUL.mjs";
-import "../chunk-46OGYEME.mjs";
+import "../chunk-R3NOM2DG.mjs";
+import "../chunk-62IYSU3I.mjs";
+import "../chunk-KAXUI5H7.mjs";
 import "../chunk-SZJB5FLZ.mjs";
 import "../chunk-GRPTCHFN.mjs";
 import "../chunk-HNXPDIWC.mjs";

package/dist/modals/chat-modal/chat-modal.js

@@ -5316,7 +5316,7 @@ init_packageSettingsStore();
 
 // src/util.ts
 init_debugLogger();
-var toTitleCase = (str) => str.toLowerCase().split(" ").map((w) => w.charAt(0).toUpperCase() + w.slice(1)).join(" ");
+var toTitleCase = (str2) => str2.toLowerCase().split(" ").map((w) => w.charAt(0).toUpperCase() + w.slice(1)).join(" ");
 
 // src/services/prompts/conversationStarters.ts
 init_debugLogger();
@@ -5772,6 +5772,95 @@ init_debugLogger();
 
 // src/services/vectorDatabase/vectorDatabaseService.ts
 init_debugLogger();
+
+// src/services/vectorDatabase/knowledgeFiles.ts
+var KNOWLEDGE_PAGE_LIMIT_MAX = 200;
+var MEDIA_MIME_PREFIXES = ["image/", "video/", "audio/"];
+var MEDIA_EXTENSIONS = /* @__PURE__ */ new Set([
+  "png",
+  "jpg",
+  "jpeg",
+  "gif",
+  "webp",
+  "bmp",
+  "tif",
+  "tiff",
+  "heic",
+  "heif",
+  "svg",
+  "ico",
+  "avif",
+  "mp4",
+  "mov",
+  "m4v",
+  "webm",
+  "avi",
+  "mkv",
+  "mp3",
+  "wav",
+  "m4a",
+  "aac",
+  "ogg",
+  "flac"
+]);
+var TEMP_NAME = /^(~WRD\d*|~\$|~\w+\.tmp$)|\.tmp$/i;
+var str = (value) => typeof value === "string" ? value : "";
+var recordName = (record) => str(record.filename) || str(record.fileName) || str(record.name);
+var recordMime = (record) => (str(record.mimeType) || str(record.contentType) || str(record.type)).toLowerCase();
+var isLikelyKnowledgeFile = (record) => {
+  const name = recordName(record);
+  if (TEMP_NAME.test(name)) return false;
+  const ext = name.includes(".") ? name.split(".").pop().toLowerCase() : "";
+  if (MEDIA_EXTENSIONS.has(ext)) return false;
+  const mime = recordMime(record);
+  if (MEDIA_MIME_PREFIXES.some((prefix) => mime.startsWith(prefix))) return false;
+  return true;
+};
+var toRecords = (value) => Array.isArray(value) ? value.filter((item) => !!item && typeof item === "object" && !Array.isArray(item)) : [];
+var num = (value, fallback) => typeof value === "number" && Number.isFinite(value) ? value : fallback;
+var parseKnowledgeFileList = (body, request) => {
+  if (body && typeof body === "object" && !Array.isArray(body) && "total" in body) {
+    const envelope = body;
+    const files2 = toRecords(envelope.files);
+    const total = num(envelope.total, files2.length);
+    const skip = num(envelope.skip, request.skip);
+    const scope = envelope.scope === "all" ? "all" : "knowledge";
+    return {
+      files: files2,
+      total,
+      hasMore: typeof envelope.hasMore === "boolean" ? envelope.hasMore : skip + files2.length < total,
+      scope,
+      serverScoped: true,
+      knowledgeFilterUnavailable: envelope.knowledgeFilterUnavailable === true,
+      hiddenCount: 0
+    };
+  }
+  const raw = body && typeof body === "object" && !Array.isArray(body) ? toRecords(body.files) : toRecords(body);
+  const hasMore = raw.length >= request.limit;
+  if (request.scope === "all") {
+    return {
+      files: raw,
+      total: request.skip + raw.length,
+      hasMore,
+      scope: "all",
+      serverScoped: false,
+      knowledgeFilterUnavailable: false,
+      hiddenCount: 0
+    };
+  }
+  const files = raw.filter(isLikelyKnowledgeFile);
+  return {
+    files,
+    total: files.length,
+    hasMore,
+    scope: "knowledge",
+    serverScoped: false,
+    knowledgeFilterUnavailable: false,
+    hiddenCount: raw.length - files.length
+  };
+};
+
+// src/services/vectorDatabase/vectorDatabaseService.ts
 var VectorDatabaseService = class {
   baseUrl;
   fileStorageApiUrl;
@@ -6531,36 +6620,44 @@ var VectorDatabaseService = class {
     }
   }
   /**
-   * Get user's files - tries both Gateway API and file storage API
+   * One page of the Knowledge file list from the gateway (see ./knowledgeFiles).
+   *
+   * No fallback to the file storage API here: its listing is the whole Drive (every product's
+   * files), which is exactly what Knowledge must not show. A failure surfaces to the caller.
    */
-  async getUserFiles(skip = 0, limit = 50) {
+  async listKnowledgeFiles(skip = 0, limit = 50, scope = "knowledge") {
     if (!this.isAvailable()) {
-      return [];
+      return parseKnowledgeFileList([], { skip, limit, scope });
+    }
+    const safeSkip = Math.max(0, Math.floor(skip));
+    const safeLimit = Math.min(KNOWLEDGE_PAGE_LIMIT_MAX, Math.max(1, Math.floor(limit)));
+    const query = new URLSearchParams({
+      skip: String(safeSkip),
+      limit: String(safeLimit),
+      scope
+    });
+    debugLogger.debug("Fetching knowledge files from Gateway API", { skip: safeSkip, limit: safeLimit, scope });
+    const response = await fetch(`${this.baseUrl}/embedding/files?${query.toString()}`, {
+      method: "GET",
+      headers: this.getHeaders()
+    });
+    if (!response.ok) {
+      throw new Error(`Couldn't list documents (gateway answered ${response.status})`);
     }
+    const body = await response.json();
+    return parseKnowledgeFileList(body, { skip: safeSkip, limit: safeLimit, scope });
+  }
+  /**
+   * Get the user's Knowledge files (first page, knowledge scope). Kept for existing callers;
+   * prefer listKnowledgeFiles for totals and paging.
+   */
+  async getUserFiles(skip = 0, limit = 50) {
     try {
-      debugLogger.debug("Fetching user files from Gateway API", { skip, limit });
-      const response = await fetch(`${this.baseUrl}/embedding/files?skip=${skip}&limit=${limit}`, {
-        method: "GET",
-        headers: this.getHeaders()
-      });
-      if (!response.ok) {
-        debugLogger.warn("Gateway API files endpoint failed, falling back to file storage API", {
-          status: response.status
-        });
-        return await this.getUserFilesFromFileStorage(skip, limit);
-      }
-      const result = await response.json();
-      debugLogger.debug("Gateway API files response", result);
-      if (Array.isArray(result)) {
-        return this.toRecordArray(result);
-      }
-      if (result && Array.isArray(result.files)) {
-        return this.toRecordArray(result.files);
-      }
-      return [];
+      const page = await this.listKnowledgeFiles(skip, limit, "knowledge");
+      return page.files;
     } catch (error) {
-      debugLogger.error("Gateway API failed while fetching user files, trying file storage API", { error });
-      return await this.getUserFilesFromFileStorage(skip, limit);
+      debugLogger.error("Failed to fetch knowledge files from Gateway API", { error });
+      return [];
     }
   }
   /**
@@ -9340,6 +9437,7 @@ var knowledgeDocToVectorDocument = (doc) => ({
   isTeamContent: doc.isTeamContent ?? false,
   contentSource: doc.contentSource ?? (doc.isTeamContent ? "team" : "user")
 });
+var optionalString = (value) => typeof value === "string" && value.length > 0 ? value : void 0;
 var rawFileRecordToVectorDocument = (record) => {
   if (!record || typeof record !== "object") {
     return null;
@@ -9354,10 +9452,12 @@ var rawFileRecordToVectorDocument = (record) => {
   const mimeType = typeof record.mimeType === "string" ? record.mimeType : "application/octet-stream";
   const uploadedBy = typeof record.uploadedBy === "string" ? record.uploadedBy : "unknown";
   const uploadedAt = typeof record.uploadedAt === "string" ? record.uploadedAt : (/* @__PURE__ */ new Date()).toISOString();
-  const isUserContent = typeof record.isUserContent === "boolean" ? record.isUserContent : true;
-  const isTeamContent = typeof record.isTeamContent === "boolean" ? record.isTeamContent : false;
-  const contentSource = record.contentSource === "team" ? "team" : "user";
+  const teamSid = optionalString(record.teamSid);
+  const isTeamContent = typeof record.isTeamContent === "boolean" ? record.isTeamContent : !!teamSid;
+  const isUserContent = typeof record.isUserContent === "boolean" ? record.isUserContent : !isTeamContent;
+  const contentSource = record.contentSource === "team" || record.contentSource === void 0 && isTeamContent ? "team" : "user";
   const score = typeof record.score === "number" ? record.score : void 0;
+  const size = typeof record.size === "number" && Number.isFinite(record.size) && record.size >= 0 ? record.size : void 0;
   return {
     id,
     filename,
@@ -9368,7 +9468,12 @@ var rawFileRecordToVectorDocument = (record) => {
     isUserContent,
     isTeamContent,
     contentSource,
-    score
+    score,
+    size,
+    bucket: optionalString(record.bucket),
+    teamSid,
+    userEmail: optionalString(record.userEmail),
+    preview
   };
 };
 var useVectorStore = () => {
@@ -9920,22 +10025,32 @@ var useVectorStore = () => {
   (0, import_react6.useEffect)(() => {
     initializeVectorService();
   }, [initializeVectorService]);
-  const getUserDocuments = (0, import_react6.useCallback)(async (skip = 0, limit = 50) => {
+  const getKnowledgeDocumentsPage = (0, import_react6.useCallback)(async (skip = 0, limit = 50, scope = "knowledge") => {
     if (!status.isEnabled || status.migrationRequired) {
-      return [];
+      return {
+        documents: [],
+        total: 0,
+        hasMore: false,
+        scope,
+        serverScoped: false,
+        knowledgeFilterUnavailable: false,
+        hiddenCount: 0
+      };
     }
+    const { files, ...meta } = await vectorDatabaseService.listKnowledgeFiles(skip, limit, scope);
+    const documents = files.map(rawFileRecordToVectorDocument).filter((doc) => doc !== null);
+    debugLogger.info("Retrieved knowledge documents", { count: documents.length, total: meta.total, scope: meta.scope });
+    return { documents, ...meta };
+  }, [status.isEnabled, status.migrationRequired]);
+  const getUserDocuments = (0, import_react6.useCallback)(async (skip = 0, limit = 50) => {
     try {
-      const records = await vectorDatabaseService.getUserFiles(skip, limit);
-      const documents = records.map(rawFileRecordToVectorDocument).filter((doc) => doc !== null);
-      debugLogger.info("Retrieved user documents from vector database", {
-        count: documents.length
-      });
-      return documents;
+      const page = await getKnowledgeDocumentsPage(skip, limit, "knowledge");
+      return page.documents;
     } catch (error) {
       debugLogger.error("Failed to get user documents", { error });
       return [];
     }
-  }, [status.isEnabled, status.migrationRequired]);
+  }, [getKnowledgeDocumentsPage]);
   const deleteDocument = (0, import_react6.useCallback)(async (fileId) => {
     if (!status.isEnabled || status.migrationRequired) {
       return {
@@ -10032,6 +10147,7 @@ var useVectorStore = () => {
     uploadDocument,
     searchDocuments,
     getUserDocuments,
+    getKnowledgeDocumentsPage,
     deleteDocument,
     downloadVectorFile,
     getFilePreview,
@@ -10362,7 +10478,8 @@ var DocumentCard = ({
   const metaLine = [
     sourceLabel,
     fileInfo.language ?? fileInfo.category,
-    formatFileSize((doc.content?.length || 0) * 2)
+    // Real file size when the listing reports one; otherwise estimate from the text content.
+    formatFileSize(typeof doc.size === "number" && doc.size > 0 ? doc.size : (doc.content?.length || 0) * 2)
   ].join(" \xB7 ");
   if (documentState === "unavailable" && !allowErrorStates) {
     return null;
@@ -19074,8 +19191,8 @@ var ChatModal = ({
       }
     }
   };
-  const toTitleCase2 = (str) => {
-    return str.replace(/\w\S*/g, (txt) => txt.charAt(0).toUpperCase() + txt.substr(1).toLowerCase());
+  const toTitleCase2 = (str2) => {
+    return str2.replace(/\w\S*/g, (txt) => txt.charAt(0).toUpperCase() + txt.substr(1).toLowerCase());
   };
   (0, import_react27.useEffect)(() => {
     if (!SYSTEM_PROMPT) {