memory.h

#pragma once
 
#include <agents-cpp/types.h>
#include <memory>
#include <optional>
#include <vector>
 
namespace agents {

class Memory {
public:
    virtual ~Memory() = default;

    virtual void add(const String& key, const JsonObject& value, MemoryType type = MemoryType::SHORT_TERM) = 0;

    virtual std::optional<JsonObject> get(const String& key, MemoryType type = MemoryType::SHORT_TERM) const = 0;

    virtual bool has(const String& key, MemoryType type = MemoryType::SHORT_TERM) const = 0;

    virtual void remove(const String& key, MemoryType type = MemoryType::SHORT_TERM) = 0;

    virtual void clear(MemoryType type = MemoryType::SHORT_TERM) = 0;

    virtual void addMessage(const Message& message) = 0;

    virtual std::vector<Message> getMessages() const = 0;

    virtual String getConversationSummary(int max_length = 0) const = 0;

    virtual std::vector<std::pair<JsonObject, float>> search(
        const String& query,
        MemoryType type = MemoryType::LONG_TERM,
        int max_results = 5
    ) const = 0;
};

std::shared_ptr<Memory> createMemory();
 
} // namespace agents