@teamkeel/functions-runtime 0.451.1 → 0.452.0

package/dist/index.js

@@ -4077,12 +4077,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 +4122,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,6 +4144,7 @@ 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();
                 }
@@ -4152,6 +4157,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
                 status: "NEW" /* NEW */,
                 type: "FUNCTION" /* FUNCTION */
               }).returningAll().executeTakeFirst();
+              span.setAttribute("step.status", "NEW" /* NEW */);
               throw new StepCreatedDisrupt(
                 options.retryPolicy ? new Date(
                   Date.now() + options.retryPolicy(failedSteps.length + 1)
@@ -4164,6 +4170,7 @@ 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({
@@ -4174,6 +4181,7 @@ function createFlowContext(runId, data, action, callback, element, spanId, ctx)
             type: "FUNCTION" /* FUNCTION */
           }).returningAll().executeTakeFirst();
           span.setAttribute(KEEL_INTERNAL_ATTR, KEEL_INTERNAL_CHILDREN);
+          span.setAttribute("step.status", "NEW" /* NEW */);
           throw new StepCreatedDisrupt();
         });
       });
@@ -4182,9 +4190,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 +4212,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 +4229,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 +4237,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 +4257,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 +4270,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 +4289,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 };