orchestrator_workflow.h

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

class OrchestratorWorkflow : public Workflow {
public:
    struct Worker {
        String name;
        String description;
        String prompt_template;
        std::shared_ptr<LLMInterface> llm;
        std::function<JsonObject(const String&, const JsonObject&)> handler;

        Worker(
            const String& name,
            const String& description,
            const String& prompt_template,
            std::shared_ptr<LLMInterface> llm = nullptr,
            std::function<JsonObject(const String&, const JsonObject&)> handler = nullptr
        ) : name(name), description(description),
            prompt_template(prompt_template), llm(llm), handler(handler) {}
    };

    OrchestratorWorkflow(
        std::shared_ptr<AgentContext> context,
        const String& orchestrator_prompt_template = ""
    );

    void addWorker(const Worker& worker);

    void addWorker(
        const String& name,
        const String& description,
        const String& prompt_template,
        std::shared_ptr<LLMInterface> worker_llm = nullptr
    );

    void init();

    JsonObject run(const String& input) override;

    void setMaxIterations(int max_iterations);

    void setSynthesizer(std::function<JsonObject(const std::vector<JsonObject>&)> synthesizer);

    void setOrchestratorPrompt(const String& orchestrator_prompt_template);

    JsonObject getWorkersSchema() const;
 
private:
    String orchestrator_prompt_template_;

    std::vector<Worker> workers_;

    std::function<JsonObject(const std::vector<JsonObject>&)> synthesizer_;

    std::map<String, Worker> worker_map_;

    int max_iterations_ = 5;

    JsonObject defaultSynthesizer(const std::vector<JsonObject>& results);

    String createOrchestratorSystemPrompt() const;

    JsonObject executeWorker(
        const String& worker_name,
        const String& task,
        const JsonObject& context_data
    );
};
 
} // namespace workflows
} // namespace agents