@opengeni/documents 0.7.0-canary.0 → 0.8.2-canary.0

package/dist/index.d.ts

@@ -166,6 +166,7 @@ export type DocumentSearchInput = {
     limit?: number | undefined;
     mode?: DocumentSearchMode | undefined;
     sourceKinds?: KnowledgeSourceKind[] | undefined;
+    authorityKinds?: DocumentAuthorityKind[] | undefined;
     aclTags?: string[] | undefined;
     access?: DocumentAccessFilter | undefined;
 };

package/dist/index.js

@@ -844,7 +844,10 @@ async function getDocumentDefaultCollectionBackfillAudit(db, input) {
         operationsHasMore,
         operationsNextCursor: operationTail ? encodeDocumentMigrationAuditCursor(
           ["backfill-operations", input.accountId, input.runId],
-          { timestamp: operationTail.createdAt, id: operationTail.operationId }
+          {
+            timestamp: operationTail.createdAt,
+            id: operationTail.operationId
+          }
         ) : null,
         receiptsHasMore,
         receiptsNextCursor: receiptTail ? encodeDocumentMigrationAuditCursor(
@@ -931,9 +934,14 @@ function decodeDocumentMigrationAuditCursor(value, parts, requireTimestamp) {
     ) || (requireTimestamp ? typeof parsed.t !== "string" || !Number.isFinite(new Date(parsed.t).getTime()) : parsed.t !== null)) {
       throw new Error("cursor payload");
     }
-    return { timestamp: typeof parsed.t === "string" ? parsed.t : null, id: parsed.i };
+    return {
+      timestamp: typeof parsed.t === "string" ? parsed.t : null,
+      id: parsed.i
+    };
   } catch (error) {
-    throw new Error("invalid document migration audit cursor", { cause: error });
+    throw new Error("invalid document migration audit cursor", {
+      cause: error
+    });
   }
 }
 async function reclassifyDocumentAuthority(db, input) {
@@ -1035,7 +1043,9 @@ function decodeDocumentAuthorityReclassificationCursor(value, scope) {
     }
     return { createdAt: parsed.t, operationId: parsed.i };
   } catch (error) {
-    throw new Error("invalid document authority receipt cursor", { cause: error });
+    throw new Error("invalid document authority receipt cursor", {
+      cause: error
+    });
   }
 }
 async function addDocumentToBase(db, input) {
@@ -1576,7 +1586,11 @@ async function curateDroppedDocument(db, services, document, parsed, file) {
     return updated2 ?? document;
   }
   const bases = await listDocumentBases(db, document.workspaceId);
-  const candidates = bases.filter((base) => base.id !== document.baseId).map((base) => ({ id: base.id, name: base.name, description: base.description }));
+  const candidates = bases.filter((base) => base.id !== document.baseId).map((base) => ({
+    id: base.id,
+    name: base.name,
+    description: base.description
+  }));
   const input = {
     text: parsed.text.slice(0, DOCUMENT_CURATION_MAX_INPUT_CHARS),
     filename: file.filename,
@@ -1715,6 +1729,7 @@ async function searchEffectiveDocuments(db, input, services = createDocumentServ
       limit: input.limit,
       mode: input.mode,
       sourceKinds: input.sourceKinds,
+      authorityKinds: input.authorityKinds,
       aclTags: input.aclTags,
       // Construct the lower-level access filter here instead of spreading the
       // caller input, so an untyped/legacy access override is always ignored.
@@ -1744,6 +1759,7 @@ async function searchEffectiveKnowledge(db, input, services = createDocumentServ
       ...input.baseIds ? { baseIds: input.baseIds } : {},
       ...input.mode ? { mode: input.mode } : {},
       ...input.sourceKinds ? { sourceKinds: input.sourceKinds } : {},
+      ...input.authorityKinds ? { authorityKinds: input.authorityKinds } : {},
       ...input.aclTags ? { aclTags: input.aclTags } : {},
       access
     },
@@ -1782,6 +1798,7 @@ async function searchEffectiveKnowledge(db, input, services = createDocumentServ
           ranked.map((result) => result.chunkId)
         ),
         eq(schema.documents.status, "ready"),
+        ...input.authorityKinds?.length ? [inArray(schema.documents.authorityKind, input.authorityKinds)] : [],
         ...documentAccessConditions(input.workspaceId, access)
       )
     )
@@ -2182,7 +2199,10 @@ async function browseEffectiveKnowledge(db, input) {
     access,
     async (scopedDb) => {
       if (parent) {
-        const [authorizedParent] = await scopedDb.select({ id: schema.documents.id, indexSequence: schema.documents.indexSequence }).from(schema.documents).where(
+        const [authorizedParent] = await scopedDb.select({
+          id: schema.documents.id,
+          indexSequence: schema.documents.indexSequence
+        }).from(schema.documents).where(
           and(
             eq(schema.documents.accountId, input.accountId),
             eq(schema.documents.id, parent.id),
@@ -2191,7 +2211,10 @@ async function browseEffectiveKnowledge(db, input) {
           )
         ).limit(1);
         if (!authorizedParent) {
-          return selectKnowledgeBrowseRecords({ entries: [], hasMoreAfterEntries: false });
+          return selectKnowledgeBrowseRecords({
+            entries: [],
+            hasMoreAfterEntries: false
+          });
         }
         if (authorizedParent.indexSequence === null) {
           throw new Error("ready knowledge document is missing its index revision");
@@ -2279,7 +2302,11 @@ function encodeKnowledgeBrowseCursor(scope, position) {
   if (position < 0n) throw new Error("knowledge browse cursor position is invalid");
   const version = knowledgeBrowseCursorVersion(scope);
   return Buffer.from(
-    JSON.stringify({ v: version, s: knowledgeBrowseCursorScope(scope), q: position.toString() }),
+    JSON.stringify({
+      v: version,
+      s: knowledgeBrowseCursorScope(scope),
+      q: position.toString()
+    }),
     "utf8"
   ).toString("base64url");
 }
@@ -2326,7 +2353,10 @@ function parseKnowledgeRecordId(value) {
     value
   );
   if (!match) throw new Error("invalid knowledge record id");
-  return { kind: match[1], id: match[2].toLowerCase() };
+  return {
+    kind: match[1],
+    id: match[2].toLowerCase()
+  };
 }
 function knowledgeDocumentRecord(document, citation = null, firstChunkId = null) {
   if (!document.indexedAt) throw new Error(`Ready document is missing indexed_at: ${document.id}`);
@@ -2743,6 +2773,9 @@ function documentSearchConditions(input, embeddingModel) {
   if (input.sourceKinds?.length) {
     conditions.push(inArray(schema.documents.sourceKind, input.sourceKinds));
   }
+  if (input.authorityKinds?.length) {
+    conditions.push(inArray(schema.documents.authorityKind, input.authorityKinds));
+  }
   const aclTags = cleanStringArray(input.aclTags);
   if (aclTags.length > 0) {
     conditions.push(sql`${schema.documents.aclTags} @> ${JSON.stringify(aclTags)}::jsonb`);