actor_agent.h

#pragma once
 
#include <agents-cpp/agent.h>
#include <agents-cpp/agent_context.h>
#include <agents-cpp/coroutine_utils.h>
#include <map>
#include <memory>
#include <mutex>
#include <optional>
#include <queue>
#include <thread>
#include <vector>
 
namespace agents {

class ActorAgent : public Agent {
public:
    ActorAgent(std::shared_ptr<AgentContext> context);

    virtual ~ActorAgent();

    void init() override;

    Task<JsonObject> run(const String& task) override;

    void stop() override;

    void provideFeedback(const String& feedback) override;

    void setSystemPrompt(const String& system_prompt);

    String getSystemPrompt() const;

    Task<String> waitForFeedback(const String& message, const JsonObject& context) override;

protected:
    String system_prompt_;

    std::vector<Message> conversation_;

    int run_interval_ms_ = 100;

    std::promise<String> feedback_promise_;

    std::mutex queue_mutex_;

    std::condition_variable queue_cv_;

    std::queue<String> inbound_tasks_;

    std::atomic<bool> stop_flag_{false};

    std::thread worker_thread_;

    virtual void onToolUsed(const String& tool_name, const JsonObject& params, const ToolResult& result);

    virtual void onResponse(const String& response);

    virtual void onError(const String& error);

    void startWorker();

    void stopWorker();

    virtual String createAgentPrompt() const;

    Task<ToolResult> executeTool(const String& tool_name, const JsonObject& params);

    Task<String> processMessage(const String& message);
};
 
} // namespace agents