agent_context.h

#pragma once
 
#include <agents-cpp/coroutine_utils.h>
#include <agents-cpp/llm_interface.h>
#include <agents-cpp/memory.h>
#include <agents-cpp/tool.h>
#include <agents-cpp/types.h>
#include <map>
#include <memory>
#include <vector>

namespace agents {

class AgentContext {
public:
    AgentContext();
    ~AgentContext() = default;

    void setLLM(std::shared_ptr<LLMInterface> llm);

    std::shared_ptr<LLMInterface> getLLM() const;

    void setSystemPrompt(const String& system_prompt);

    const String& getSystemPrompt() const;

    void registerTool(std::shared_ptr<Tool> tool);

    std::shared_ptr<Tool> getTool(const String& name) const;

    std::vector<std::shared_ptr<Tool>> getTools() const;

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

    std::shared_ptr<Memory> getMemory() const;

    void addMessage(const Message& message);

    std::vector<Message> getMessages() const;

    Task<LLMResponse> chat(const String& user_message);

    Task<LLMResponse> chatWithTools(const String& user_message);

    AsyncGenerator<String> streamChat(const String& user_message);
 
private:
    std::shared_ptr<LLMInterface> llm_;

    std::shared_ptr<Memory> memory_;

    std::map<String, std::shared_ptr<Tool>> tools_;

    String system_prompt_;
};
 
} // namespace agents