import {Step, Workflow} from "@mastra/core/workflows";
import {z} from "zod";

const oneStep = new Step({
  id: "one",
  inputSchema: z.object({
    query: z.string()
  }),
  outputSchema: z.object({
    answer: z.string(),
  }),
  async execute() {
    console.log("one");
    return { answer: "one" };
  }
})


const twoStep = new Step({
  id: "queryAll",
  outputSchema: z.string(),
  async execute() {
    console.log("two")
    return "end";
  }
})

const templateWorkflow = new Workflow({
  name: "poetWorkflow",
  triggerSchema: z.object({
    query: z.string()
  })
})
  .step(oneStep)
  .then(twoStep, {
    when: async () => {
      console.log("false")
      return false;
    }
  })
  .commit()

export { templateWorkflow };


// execution
const { start } = templateWorkflow.createRun();
const res = await start({
  triggerData: { query: "你好" }
});
console.log("d", res.results);

one // ⬅️ oneStep
// ⚠️ start twoStep, but when field always is false
// 🤔 i dont know why need to exec three times? i just wanna to exec once
false
false
false
d {
  one: { status: 'success', output: { answer: 'one' } },
  queryAll: { status: 'failed', error: 'Step:queryAll condition check failed' }
}

one // ⬅️ oneStep
// 💡 twoStep
false
d {
  one: { status: 'success', output: { answer: 'one' } },
  queryAll: { status: 'failed', error: 'Step:queryAll condition check failed' }
}