@axiom-lattice/gateway 8.1.0 → 8.1.1

package/dist/index.js

@@ -11421,6 +11421,52 @@ var import_core8 = require("@axiom-lattice/core");
 
 // src/services/threads/threadOwnership.ts
 var import_core7 = require("@axiom-lattice/core");
+var DELEGATED_THREAD_SEPARATOR = "____";
+var MAX_DELEGATED_THREAD_LINEAGE = 8;
+async function threadHasTenantRecord(tenantId, threadId) {
+  try {
+    const thread = await (0, import_core7.getStoreLattice)("default", "thread").store.getThreadById(
+      tenantId,
+      threadId
+    );
+    if (thread) return true;
+  } catch {
+  }
+  try {
+    const ledger = (0, import_core7.getStoreLattice)("default", "agentRun").store;
+    const runs = await ledger.listRuns(tenantId, threadId, { limit: 1 });
+    if (runs.length > 0) return true;
+  } catch {
+  }
+  try {
+    const tasks = (0, import_core7.getStoreLattice)("default", "task").store;
+    const matches = await tasks.list({ tenantId, threadId, limit: 1 });
+    return matches.length > 0;
+  } catch {
+    return false;
+  }
+}
+async function threadLineageRootedInTenant(tenantId, threadId) {
+  let candidate = threadId;
+  for (let depth = 0; depth < MAX_DELEGATED_THREAD_LINEAGE; depth += 1) {
+    if (await threadHasTenantRecord(tenantId, candidate)) return true;
+    const boundary = candidate.lastIndexOf(DELEGATED_THREAD_SEPARATOR);
+    if (boundary <= 0) return false;
+    candidate = candidate.slice(0, boundary);
+  }
+  return false;
+}
+async function delegatedThreadOwnedBy(tenantId, assistantId, threadId) {
+  const marker = `${DELEGATED_THREAD_SEPARATOR}${assistantId}_`;
+  let boundary = threadId.indexOf(marker);
+  while (boundary > 0) {
+    if (await threadLineageRootedInTenant(tenantId, threadId.slice(0, boundary))) {
+      return true;
+    }
+    boundary = threadId.indexOf(marker, boundary + DELEGATED_THREAD_SEPARATOR.length);
+  }
+  return false;
+}
 async function threadOwnedBy(tenantId, assistantId, threadId) {
   try {
     const thread = await (0, import_core7.getStoreLattice)("default", "thread").store.getThreadById(
@@ -11439,10 +11485,12 @@ async function threadOwnedBy(tenantId, assistantId, threadId) {
   try {
     const tasks = (0, import_core7.getStoreLattice)("default", "task").store;
     const matches = await tasks.list({ tenantId, threadId, limit: 1 });
-    return matches.some((task) => task.ownerType === "agent" && task.ownerId === assistantId);
+    if (matches.some((task) => task.ownerType === "agent" && task.ownerId === assistantId)) {
+      return true;
+    }
   } catch {
-    return false;
   }
+  return delegatedThreadOwnedBy(tenantId, assistantId, threadId);
 }
 async function threadIsNewForTenant(tenantId, threadId) {
   try {