agent.h

#pragma once
 
#include <agents-cpp/agent_context.h>
#include <agents-cpp/coroutine_utils.h>
#include <agents-cpp/types.h>
#include <functional>
#include <memory>
 
namespace agents {

class Agent {
public:
    enum class State {
        READY,
        RUNNING,
        WAITING,
        COMPLETED,
        FAILED,
        STOPPED
    };

    struct Options {
        int max_iterations = 10;

        int max_consecutive_errors = 3;

        bool human_feedback_enabled = true;

        std::function<bool(const String&, const JsonObject&)> human_in_the_loop;
    };

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

    virtual ~Agent() = default;

    virtual void init() = 0;

    virtual Task<JsonObject> run(const String& task) = 0;

    void runAsync(const String& task, std::function<void(const JsonObject&)> callback);

    virtual void stop();

    std::shared_ptr<AgentContext> getContext() const;

    State getState() const;

    void setOptions(const Options& options);

    const Options& getOptions() const;

    void setStatusCallback(std::function<void(const String&)> callback);

    virtual void provideFeedback(const String& feedback);

    virtual Task<String> waitForFeedback(
        const String& message,
        const JsonObject& context
    ) {
        // Placeholder implementation, to be overridden
        co_return "";
    }

protected:
    std::shared_ptr<AgentContext> context_;

    State state_ = State::READY;

    Options options_;

    std::function<void(const String&)> status_callback_;

    void setState(State state);

    void logStatus(const String& status);
};
 
} // namespace agents