Remove stepDisplayName and stepIndex from step context structure and related logic

Author: Roman Snapko

packages/react-ui/src/app/features/ai/lib/enrich-context.ts

@@ -22,14 +22,10 @@ export const createAdditionalContext = (
     flowId: flowVersion.flowId,
     flowVersionId: flowVersion.id,
     runId,
-    currentStepId: stepData?.id ?? '',
-    currentStepDisplayName: stepData?.displayName ?? '',
     currentStepIndex: stepData?.stepIndex,
     steps: [
       {
         id: stepData?.id ?? '',
-        stepDisplayName: stepData?.displayName ?? '',
-        stepIndex: stepData?.stepIndex,
         variables: variables.length > 0 ? variables : undefined,
       },
     ],

packages/server/api/src/app/ai/chat/context-enrichment.service.ts

@@ -181,8 +181,6 @@ async function enrichSteps(
       } catch (error) {
         return {
           id: step.id,
-          stepDisplayName: step.stepDisplayName,
-          stepIndex: step.stepIndex,
           variables: step.variables?.map((variable) => ({
             name: variable.name,
             value: String(error),
@@ -203,8 +201,6 @@ async function enrichStep(
   if (!step.variables || step.variables.length === 0) {
     return {
       id: step.id,
-      stepDisplayName: step.stepDisplayName,
-      stepIndex: step.stepIndex,
       variables: step.variables,
     };
   }
@@ -217,8 +213,6 @@ async function enrichStep(
 
   return {
     id: step.id,
-    stepDisplayName: step.stepDisplayName,
-    stepIndex: step.stepIndex,
     variables: resolvedVariables,
   };
 }
@@ -256,11 +250,20 @@ async function resolveVariable(
   flowData: FlowData,
   projectId: string,
 ): Promise<{ name: string; value: unknown }> {
+  const workflowStep = flowHelper.getStepById(flowData.flow.version, step.id);
+
+  if (!workflowStep) {
+    return {
+      name: variable.name,
+      value: 'Failed to resolve variable: Step not found in flow.',
+    };
+  }
+
   const { result } = await engineRunner.executeVariable(flowData.engineToken, {
     variableExpression: variable.value,
     flowVersion: flowData.flow.version,
     projectId,
-    stepName: step.stepDisplayName,
+    stepName: workflowStep.name,
     stepTestOutputs: flowData.stepTestOutputs,
   });
 

packages/shared/src/lib/ai/chat/index.ts

@@ -43,8 +43,6 @@ export type VariableContext = Static<typeof VariableContext>;
 
 export const StepContext = Type.Object({
   id: Type.String(),
-  stepDisplayName: Type.String(),
-  stepIndex: Type.Optional(Type.Number()),
   variables: Type.Optional(Type.Array(VariableContext)),
 });
 export type StepContext = Static<typeof StepContext>;