@teamkeel/functions-runtime 0.451.2 → 0.452.1

package/dist/index.js

@@ -4060,6 +4060,23 @@ var defaultOpts = {
   retries: 4,
   timeout: 6e4
 };
+async function insertNewStep(db, runId, name, stage) {
+  await db.transaction().execute(async (trx) => {
+    await trx.selectFrom("keel.flow_run").select("id").where("id", "=", runId).forUpdate().executeTakeFirst();
+    const existing = await trx.selectFrom("keel.flow_step").select("id").where("run_id", "=", runId).where("name", "=", name).where("status", "=", "NEW" /* NEW */).executeTakeFirst();
+    if (existing) {
+      return;
+    }
+    await trx.insertInto("keel.flow_step").values({
+      run_id: runId,
+      name,
+      stage,
+      status: "NEW" /* NEW */,
+      type: "FUNCTION" /* FUNCTION */
+    }).execute();
+  });
+}
+__name(insertNewStep, "insertNewStep");
 function createFlowContext(runId, data, action, callback, element, spanId, ctx) {
   const usedNames = /* @__PURE__ */ new Set();
   return {
@@ -4077,12 +4094,15 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
     step: /* @__PURE__ */ __name(async (name, optionsOrFn, fn) => {
       return withSpan(`Step - ${name}`, async (span) => {
         return withUserSpan(span, async () => {
+          span.setAttribute("step.type", "FUNCTION" /* FUNCTION */);
+          span.setAttribute("step.name", name);
           const options = typeof optionsOrFn === "function" ? {} : optionsOrFn;
           const actualFn = typeof optionsOrFn === "function" ? optionsOrFn : fn;
           options.retries = options.retries ?? defaultOpts.retries;
           options.timeout = options.timeout ?? defaultOpts.timeout;
           const db = useDatabase();
           if (usedNames.has(name)) {
+            span.setAttribute("step.status", "FAILED" /* FAILED */);
             await db.insertInto("keel.flow_step").values({
               run_id: runId,
               name,
@@ -4119,6 +4139,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
           }
           if (completedSteps.length === 1) {
             span.setAttribute(KEEL_INTERNAL_ATTR, KEEL_INTERNAL_CHILDREN);
+            span.setAttribute("step.status", "COMPLETED" /* COMPLETED */);
             return completedSteps[0].value;
           }
           if (newSteps.length === 1) {
@@ -4140,18 +4161,14 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
                 error: e instanceof Error ? e.message : "An error occurred"
               }).where("id", "=", newSteps[0].id).returningAll().executeTakeFirst();
               if (failedSteps.length >= options.retries || e instanceof NonRetriableError) {
+                span.setAttribute("step.status", "FAILED" /* FAILED */);
                 if (options.onFailure) {
                   await options.onFailure();
                 }
                 throw new ExhuastedRetriesDisrupt();
               }
-              await db.insertInto("keel.flow_step").values({
-                run_id: runId,
-                name,
-                stage: options.stage,
-                status: "NEW" /* NEW */,
-                type: "FUNCTION" /* FUNCTION */
-              }).returningAll().executeTakeFirst();
+              await insertNewStep(db, runId, name, options.stage);
+              span.setAttribute("step.status", "NEW" /* NEW */);
               throw new StepCreatedDisrupt(
                 options.retryPolicy ? new Date(
                   Date.now() + options.retryPolicy(failedSteps.length + 1)
@@ -4164,16 +4181,12 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
               spanId,
               endTime: /* @__PURE__ */ new Date()
             }).where("id", "=", newSteps[0].id).returningAll().executeTakeFirst();
+            span.setAttribute("step.status", "COMPLETED" /* COMPLETED */);
             return result;
           }
-          await db.insertInto("keel.flow_step").values({
-            run_id: runId,
-            name,
-            stage: options.stage,
-            status: "NEW" /* NEW */,
-            type: "FUNCTION" /* FUNCTION */
-          }).returningAll().executeTakeFirst();
+          await insertNewStep(db, runId, name, options.stage);
           span.setAttribute(KEEL_INTERNAL_ATTR, KEEL_INTERNAL_CHILDREN);
+          span.setAttribute("step.status", "NEW" /* NEW */);
           throw new StepCreatedDisrupt();
         });
       });
@@ -4182,9 +4195,12 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
       page: /* @__PURE__ */ __name((async (name, options) => {
         return withSpan(`Page - ${name}`, async (span) => {
           return withUserSpan(span, async () => {
+            span.setAttribute("step.type", "UI" /* UI */);
+            span.setAttribute("step.name", name);
             const db = useDatabase();
             const isCallback = element && callback;
             if (usedNames.has(name)) {
+              span.setAttribute("step.status", "FAILED" /* FAILED */);
               await db.insertInto("keel.flow_step").values({
                 run_id: runId,
                 name,
@@ -4201,6 +4217,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
             let step = await db.selectFrom("keel.flow_step").where("run_id", "=", runId).where("name", "=", name).selectAll().executeTakeFirst();
             if (step && step.status === "COMPLETED" /* COMPLETED */) {
               span.setAttribute(KEEL_INTERNAL_ATTR, KEEL_INTERNAL_CHILDREN);
+              span.setAttribute("step.status", "COMPLETED" /* COMPLETED */);
               const parsedData2 = transformRichDataTypes(step.value);
               if (step.action) {
                 return { data: parsedData2, action: step.action };
@@ -4217,6 +4234,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
                 startTime: /* @__PURE__ */ new Date()
               }).returningAll().executeTakeFirst();
               span.setAttribute("rendered", true);
+              span.setAttribute("step.status", "PENDING" /* PENDING */);
               throw new UIRenderDisrupt(
                 step?.id,
                 (await page(options, null, null)).page
@@ -4224,6 +4242,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
             }
             if (isCallback) {
               span.setAttribute("callback", callback);
+              span.setAttribute("step.status", "PENDING" /* PENDING */);
               try {
                 const response = await callbackFn(
                   options.content,
@@ -4243,6 +4262,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
               }
             }
             if (!data) {
+              span.setAttribute("step.status", "PENDING" /* PENDING */);
               throw new UIRenderDisrupt(
                 step?.id,
                 (await page(options, null, null)).page
@@ -4255,8 +4275,10 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
               }
             } catch (e) {
               if (e instanceof UIRenderDisrupt) {
+                span.setAttribute("step.status", "PENDING" /* PENDING */);
                 throw e;
               }
+              span.setAttribute("step.status", "FAILED" /* FAILED */);
               await db.updateTable("keel.flow_step").set({
                 status: "FAILED" /* FAILED */,
                 spanId,
@@ -4272,6 +4294,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
               spanId,
               endTime: /* @__PURE__ */ new Date()
             }).where("id", "=", step.id).returningAll().executeTakeFirst();
+            span.setAttribute("step.status", "COMPLETED" /* COMPLETED */);
             const parsedData = transformRichDataTypes(data);
             if (action) {
               return { data: parsedData, action };