prompt_chaining_workflow.h

#pragma once
 
#include <agents-cpp/workflow.h>
#include <functional>
#include <vector>
 
namespace agents {
namespace workflows {

class PromptChainingWorkflow : public Workflow {
public:
    struct Step {
        String name;

        String prompt_template;

        std::function<bool(const JsonObject&)> validator;

        std::function<JsonObject(const JsonObject&)> transformer;

        bool use_tools = false;

        Step(
            const String& name,
            const String& prompt_template,
            std::function<bool(const JsonObject&)> validator = nullptr,
            std::function<JsonObject(const JsonObject&)> transformer = nullptr
        ) : name(name), prompt_template(prompt_template),
            validator(validator), transformer(transformer) {}
    };

    PromptChainingWorkflow(std::shared_ptr<AgentContext> context);

    void addStep(const Step& step);

    void addStep(
        const String& name,
        const String& prompt_template,
        std::function<bool(const JsonObject&)> validator = nullptr,
        std::function<JsonObject(const JsonObject&)> transformer = nullptr
    );

    JsonObject run(const String& input) override;

    JsonObject run();
 
private:
    std::vector<Step> steps_;

    JsonObject step_outputs_;

    JsonObject run(const JsonObject& input);
};
 
} // namespace workflows
} // namespace agents