types.h

#pragma once
 
#include <any>
#include <map>
#include <memory>
#include <nlohmann/json.hpp>
#include <optional>
#include <string>
#include <variant>
#include <vector>
 
namespace agents {

using String = std::string;

using StringMap = std::map<String, String>;

using JsonObject = nlohmann::json;

using JsonArray = nlohmann::json::array_t;

struct Parameter {
    String name;
    String description;
    String type;
    bool required;
    std::optional<JsonObject> default_value;
};

using ParameterMap = std::map<String, Parameter>;

struct LLMResponse {
    String content;
    std::vector<std::pair<String, JsonObject>> tool_calls;
    std::map<String, double> usage_metrics;
};

struct Message {
    enum class Role {
        SYSTEM,
        USER,
        ASSISTANT,
        TOOL
    };

    Role role;
    String content;
    std::optional<String> name;
    std::optional<String> tool_call_id;
    std::vector<std::pair<String, JsonObject>> tool_calls;
};

enum class MemoryType {
    SHORT_TERM,
    LONG_TERM,
    WORKING
};
 
} // namespace agents