@axiom-lattice/gateway 8.1.0 → 8.2.0

package/dist/embed/embed-page.js

@@ -1,4 +1,4 @@
-import{b as h}from"./embed-page-chunk-YM76GTXK.js";import"./embed-page-chunk-SQSVDSG6.js";import"./embed-page-chunk-NNQHJZ5Q.js";import"./embed-page-chunk-ONXC5S22.js";import"./embed-page-chunk-6ZHR5GWS.js";import"./embed-page-chunk-F7ZUYG42.js";import"./embed-page-chunk-YCTDKWKW.js";import{a as E}from"./embed-page-chunk-QV7QAJ4N.js";import"./embed-page-chunk-VW3NF5NT.js";import"./embed-page-chunk-34RWKVD7.js";import"./embed-page-chunk-B2KNZ3EW.js";import{a as b}from"./embed-page-chunk-F7TA3LRH.js";import"./embed-page-chunk-R5QL57JJ.js";import"./embed-page-chunk-QOWM3LMZ.js";import"./embed-page-chunk-EUOOO2LS.js";import"./embed-page-chunk-5JDQ456X.js";import{b as f}from"./embed-page-chunk-6IIMW7OG.js";import{c as z,f as l}from"./embed-page-chunk-QBTAJ2YI.js";var x=z(m=>{"use strict";var y=b();m.createRoot=y.createRoot,m.hydrateRoot=y.hydrateRoot;var te});var T=l(x()),M=l(b());var w="axiom-web-app-embed";var _=["axiom:hello","axiom:identity","axiom:send","axiom:config"];function R(e,t){return{source:w,version:1,type:e,payload:t}}var B=6,H=40,Q=500;function q(e){if(typeof e!="object"||e===null||Array.isArray(e))return!1;let t=e,n=Object.keys(t);return n.length!==2||!n.includes("label")||!n.includes("text")?!1:typeof t.label=="string"&&t.label.length>=1&&t.label.length<=H&&typeof t.text=="string"&&t.text.length>=1&&t.text.length<=Q}function G(e){if(!Array.isArray(e))return[];let t=[];for(let n of e){if(t.length>=B)break;q(n)&&t.push({label:n.label,text:n.text})}return t}function C(e){if(typeof e!="object"||e===null)return null;let t=e,n={};if(typeof t.welcome=="string"&&t.welcome.length>0&&(n.welcome=t.welcome),Array.isArray(t.quicks)){let a=G(t.quicks);(a.length>0||t.quicks.length===0)&&(n.quicks=a)}return n.welcome===void 0&&n.quicks===void 0?null:n}function A(e,t){if(typeof e!="object"||e===null)return null;let n=e;return n.source!==w||n.version!==1||typeof n.type!="string"||!t.includes(n.type)?null:{type:n.type,payload:n.payload}}var r=l(E()),S=/^[A-Za-z0-9._:@-]{1,128}$/,v="axiom-embed-root",k="axiom-embed-style",P=32768,j=`
+import{b as h}from"./embed-page-chunk-HZZ5HNTF.js";import"./embed-page-chunk-CFFDP526.js";import"./embed-page-chunk-4XKCMM5C.js";import"./embed-page-chunk-PMOGSY6W.js";import"./embed-page-chunk-BSRASK7V.js";import"./embed-page-chunk-F7ZUYG42.js";import"./embed-page-chunk-ASEIOK7X.js";import{a as E}from"./embed-page-chunk-QV7QAJ4N.js";import"./embed-page-chunk-VW3NF5NT.js";import"./embed-page-chunk-34RWKVD7.js";import"./embed-page-chunk-B2KNZ3EW.js";import{a as b}from"./embed-page-chunk-F7TA3LRH.js";import"./embed-page-chunk-R5QL57JJ.js";import"./embed-page-chunk-QOWM3LMZ.js";import"./embed-page-chunk-EUOOO2LS.js";import"./embed-page-chunk-5JDQ456X.js";import{b as f}from"./embed-page-chunk-6IIMW7OG.js";import{c as z,f as l}from"./embed-page-chunk-QBTAJ2YI.js";var x=z(m=>{"use strict";var y=b();m.createRoot=y.createRoot,m.hydrateRoot=y.hydrateRoot;var te});var T=l(x()),M=l(b());var w="axiom-web-app-embed";var _=["axiom:hello","axiom:identity","axiom:send","axiom:config"];function R(e,t){return{source:w,version:1,type:e,payload:t}}var B=6,H=40,Q=500;function q(e){if(typeof e!="object"||e===null||Array.isArray(e))return!1;let t=e,n=Object.keys(t);return n.length!==2||!n.includes("label")||!n.includes("text")?!1:typeof t.label=="string"&&t.label.length>=1&&t.label.length<=H&&typeof t.text=="string"&&t.text.length>=1&&t.text.length<=Q}function G(e){if(!Array.isArray(e))return[];let t=[];for(let n of e){if(t.length>=B)break;q(n)&&t.push({label:n.label,text:n.text})}return t}function C(e){if(typeof e!="object"||e===null)return null;let t=e,n={};if(typeof t.welcome=="string"&&t.welcome.length>0&&(n.welcome=t.welcome),Array.isArray(t.quicks)){let a=G(t.quicks);(a.length>0||t.quicks.length===0)&&(n.quicks=a)}return n.welcome===void 0&&n.quicks===void 0?null:n}function A(e,t){if(typeof e!="object"||e===null)return null;let n=e;return n.source!==w||n.version!==1||typeof n.type!="string"||!t.includes(n.type)?null:{type:n.type,payload:n.payload}}var r=l(E()),S=/^[A-Za-z0-9._:@-]{1,128}$/,v="axiom-embed-root",k="axiom-embed-style",P=32768,j=`
 html, body, #${v} { height: 100%; margin: 0; }
 .axiom-embed-fill { height: 100% !important; min-height: 0 !important; }
 .axiom-embed-card { display: flex; align-items: center; justify-content: center; height: 100%; margin: 0; font-family: Inter, -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif; font-size: 14px; color: #64748b; padding: 24px; box-sizing: border-box; text-align: center; }

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 {
@@ -11696,6 +11744,29 @@ async function listAllRuns(ledger, tenantId, threadId) {
     if (page.length < ACTIVE_RUN_PAGE_SIZE) return runs;
   }
 }
+var INPUT_PREVIEW_MAX = 200;
+function contentToPreview(content) {
+  if (typeof content === "string") return content;
+  if (!Array.isArray(content)) return "";
+  return content.map((part) => {
+    if (typeof part === "string") return part;
+    return isRecord(part) && typeof part.text === "string" ? part.text : "";
+  }).join("");
+}
+function inputPreview(input) {
+  if (!isRecord(input)) return null;
+  if (typeof input.message === "string") {
+    return input.message.trim().slice(0, INPUT_PREVIEW_MAX) || null;
+  }
+  const messages = Array.isArray(input.messages) ? input.messages : [];
+  for (const message of messages) {
+    if (!isRecord(message)) continue;
+    if (message.role !== void 0 && message.role !== "human") continue;
+    const text = contentToPreview(message.content).trim();
+    if (text) return text.slice(0, INPUT_PREVIEW_MAX);
+  }
+  return null;
+}
 function toSafeRun(run2) {
   return {
     runId: run2.runId,
@@ -11707,7 +11778,9 @@ function toSafeRun(run2) {
     createdAt: run2.createdAt,
     updatedAt: run2.updatedAt,
     startedAt: run2.startedAt,
-    finishedAt: run2.finishedAt
+    finishedAt: run2.finishedAt,
+    messageId: run2.inputMessageIds[0] ?? null,
+    inputPreview: inputPreview(run2.input)
   };
 }
 function isTerminal(run2) {
@@ -11837,6 +11910,35 @@ function createRunLifecycleHandlers(deps = {}) {
   const pollIntervalMs = deps.pollIntervalMs ?? DEFAULT_POLL_INTERVAL_MS;
   const now = deps.now ?? Date.now;
   const sleep = deps.sleep ?? ((ms) => new Promise((resolve4) => setTimeout(resolve4, ms)));
+  const withRunCreateLock = async (request, reply, tenantId, assistantId, threadId, operation) => {
+    const key = runCreateScopeKey(tenantId, assistantId, threadId);
+    return serializeRunCreate(key, async () => {
+      const leaseStore = resolveLeaseStore();
+      if (!leaseStore) return operation();
+      try {
+        return await (0, import_core9.withLease)(
+          leaseStore,
+          key,
+          (0, import_core9.getReplicaId)(),
+          operation,
+          RUN_CREATE_LEASE
+        );
+      } catch (error) {
+        if (error instanceof import_core9.LeaseAcquisitionTimeoutError) {
+          request.log.warn(
+            { err: error, leaseKey: key },
+            "Run create lease acquisition timed out"
+          );
+          return unavailable(
+            reply,
+            RUN_CREATE_LOCK_TIMEOUT,
+            "Could not serialize run creation for this thread; retry shortly"
+          );
+        }
+        throw error;
+      }
+    });
+  };
   const waitForTerminal = async (tenantId, params, initial) => {
     let current4 = initial;
     const deadline = now() + waitTimeoutMs;
@@ -12066,35 +12168,6 @@ function createRunLifecycleHandlers(deps = {}) {
       const outcome2 = await createPending();
       return outcome2.ok ? respondCreated(outcome2.run) : outcome2.response;
     };
-    const withRunCreateLock = async (operation) => {
-      const key = runCreateScopeKey(tenantId, assistantId, threadId);
-      return serializeRunCreate(key, async () => {
-        const leaseStore = resolveLeaseStore();
-        if (!leaseStore) return operation();
-        try {
-          return await (0, import_core9.withLease)(
-            leaseStore,
-            key,
-            (0, import_core9.getReplicaId)(),
-            operation,
-            RUN_CREATE_LEASE
-          );
-        } catch (error) {
-          if (error instanceof import_core9.LeaseAcquisitionTimeoutError) {
-            request.log.warn(
-              { err: error, leaseKey: key },
-              "Run create lease acquisition timed out"
-            );
-            return unavailable(
-              reply,
-              RUN_CREATE_LOCK_TIMEOUT,
-              "Could not serialize run creation for this thread; retry shortly"
-            );
-          }
-          throw error;
-        }
-      });
-    };
     const scopedRunApiRuns = (runs) => runs.filter(
       (run2) => run2.tenantId === tenantId && run2.threadId === threadId && run2.assistantId === assistantId && run2.source === "run_api"
     );
@@ -12114,7 +12187,7 @@ function createRunLifecycleHandlers(deps = {}) {
       }
     };
     if (strategy === "reject") {
-      return withRunCreateLock(async () => {
+      return withRunCreateLock(request, reply, tenantId, assistantId, threadId, async () => {
         const read = await readThreadRuns();
         if (!read.ok) return read.response;
         const active2 = scopedRunApiRuns(read.runs).filter(isActiveRun);
@@ -12130,7 +12203,7 @@ function createRunLifecycleHandlers(deps = {}) {
       });
     }
     if (strategy === "interrupt") {
-      return withRunCreateLock(async () => {
+      return withRunCreateLock(request, reply, tenantId, assistantId, threadId, async () => {
         const read = await readThreadRuns();
         if (!read.ok) return read.response;
         const scoped = scopedRunApiRuns(read.runs);
@@ -12193,8 +12266,8 @@ function createRunLifecycleHandlers(deps = {}) {
         return createAndRespond();
       });
     }
-    if (strategy !== "rollback") return withRunCreateLock(createAndRespond);
-    return withRunCreateLock(async () => {
+    if (strategy !== "rollback") return withRunCreateLock(request, reply, tenantId, assistantId, threadId, createAndRespond);
+    return withRunCreateLock(request, reply, tenantId, assistantId, threadId, async () => {
       const active2 = (await listAllRuns(ledger, tenantId, threadId)).filter(
         (run2) => run2.tenantId === tenantId && run2.threadId === threadId && run2.assistantId === assistantId && run2.source !== "legacy" && (run2.status === "pending" || run2.status === "running")
       );
@@ -12329,6 +12402,76 @@ function createRunLifecycleHandlers(deps = {}) {
       return createAndRespond();
     });
   };
+  const cancelActive = async (request, reply) => {
+    if (!ledger) {
+      return unavailable(reply, RUN_LEDGER_UNAVAILABLE, "Run ledger is not available");
+    }
+    const tenantId = getTenantId4(request);
+    const { assistantId, threadId } = request.params;
+    if (!await ownsThread(tenantId, assistantId, threadId)) {
+      return notFound2(reply, THREAD_NOT_FOUND2, "Thread not found");
+    }
+    const router = resolveRouter();
+    return withRunCreateLock(request, reply, tenantId, assistantId, threadId, async () => {
+      let runs;
+      try {
+        runs = await listAllRuns(ledger, tenantId, threadId);
+      } catch (error) {
+        request.log.error({ err: error }, "Active run read failed");
+        return unavailable(reply, RUN_CREATE_UNAVAILABLE2, "Run queue is not available");
+      }
+      const active2 = runs.filter(
+        (run2) => run2.tenantId === tenantId && run2.assistantId === assistantId && run2.threadId === threadId && run2.source === "run_api" && isActiveRun(run2)
+      );
+      const cancelled = [];
+      const requested = [];
+      const failed = [];
+      for (const run2 of active2) {
+        try {
+          if (run2.status === "pending") {
+            const status2 = await ledger.requestCancel(tenantId, run2.runId);
+            if (status2 === "cancelled") {
+              const latest = await ledger.getRun(tenantId, run2.runId);
+              if (latest) cancelled.push(toSafeRun(latest));
+            } else if (status2 === "requested") {
+              requested.push(toSafeRun(run2));
+            }
+            continue;
+          }
+          if (router) {
+            const status2 = await router.requestCancelById({
+              tenantId,
+              assistantId,
+              threadId,
+              runId: run2.runId,
+              reason: "abort"
+            });
+            if (status2 === "requested" || status2 === "cancelled") {
+              requested.push(toSafeRun(run2));
+            }
+            continue;
+          }
+          const status = await ledger.requestCancel(tenantId, run2.runId);
+          if (status === "requested" || status === "cancelled") {
+            requested.push(toSafeRun(run2));
+          }
+        } catch (error) {
+          failed.push({
+            runId: run2.runId,
+            code: "RUN_CANCEL_FAILED",
+            message: error instanceof Error ? error.message : String(error)
+          });
+        }
+      }
+      return reply.status(200).send({
+        success: true,
+        accepted: failed.length === 0,
+        cancelled,
+        requested,
+        failed
+      });
+    });
+  };
   const list = async (request, reply) => {
     if (!ledger) {
       return unavailable(reply, RUN_LEDGER_UNAVAILABLE, "Run ledger is not available");
@@ -12338,12 +12481,14 @@ function createRunLifecycleHandlers(deps = {}) {
     if (!await ownsThread(tenantId, assistantId, threadId)) {
       return notFound2(reply, THREAD_NOT_FOUND2, "Thread not found");
     }
-    const runs = (await ledger.listRuns(tenantId, threadId)).filter(
-      (run2) => run2.tenantId === tenantId && run2.threadId === threadId && run2.assistantId === assistantId
+    const activeOnly = request.query.active === "true" || request.query.active === "1";
+    const source = request.query.source;
+    const scoped = (activeOnly ? await listAllRuns(ledger, tenantId, threadId) : await ledger.listRuns(tenantId, threadId)).filter(
+      (run2) => run2.tenantId === tenantId && run2.threadId === threadId && run2.assistantId === assistantId && (!activeOnly || isActiveRun(run2)) && (source === void 0 || run2.source === source)
     );
     return reply.status(200).send({
       success: true,
-      data: { runs: runs.map(toSafeRun) }
+      data: { runs: scoped.map(toSafeRun) }
     });
   };
   const get = async (request, reply) => {
@@ -12699,7 +12844,7 @@ function createRunLifecycleHandlers(deps = {}) {
       run: toSafeRun(outcome2.run)
     });
   };
-  return { create, list, get, join: join5, cancel };
+  return { create, list, get, join: join5, cancel, cancelActive };
 }
 function registerRunLifecycleRoutes(app2, deps = {}) {
   const handlers = createRunLifecycleHandlers(deps);
@@ -12711,6 +12856,10 @@ function registerRunLifecycleRoutes(app2, deps = {}) {
     "/api/assistants/:assistantId/threads/:threadId/runs",
     handlers.list
   );
+  app2.post(
+    "/api/assistants/:assistantId/threads/:threadId/runs/cancel-active",
+    handlers.cancelActive
+  );
   app2.get(
     "/api/assistants/:assistantId/threads/:threadId/runs/:runId",
     handlers.get
@@ -29517,6 +29666,15 @@ function registerAgentWebAppRuntimeRoutes(app2, deps) {
           return normalizeError2(reply, error);
         }
       });
+      routes.post(`${prefix}/assistants/:assistantId/threads/:threadId/runs/cancel-active`, async (request, reply) => {
+        try {
+          const params = mirrorThreadParams.parse(request.params);
+          const { proxy } = await authorizeRunMirror(request, params);
+          return await runLifecycle.cancelActive(proxy, reply);
+        } catch (error) {
+          return normalizeError2(reply, error);
+        }
+      });
       routes.post(`${prefix}/assistants/:assistantId/threads/:threadId/runs/:runId/cancel`, async (request, reply) => {
         try {
           const params = mirrorRunParams.parse(request.params);