messages_json.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#include <entropic/types/messages_json.h>
#include <entropic/types/content.h>
 
#include <nlohmann/json.hpp>
 
namespace entropic {
 
namespace {
 
ContentPart parse_content_part(const nlohmann::json& part) {
    ContentPart cp;
    auto type_str = part.value("type", "text");
    if (type_str == "image") {
        cp.type = ContentPartType::IMAGE;
        cp.image_path = part.value("path", "");
        cp.image_url = part.value("url", "");
    } else {
        cp.type = ContentPartType::TEXT;
        cp.text = part.value("text", "");
    }
    return cp;
}
 
Message parse_one_message(const nlohmann::json& m) {
    Message msg;
    msg.role = m.value("role", "user");
    if (m.contains("content") && m["content"].is_array()) {
        for (const auto& part : m["content"]) {
            msg.content_parts.push_back(parse_content_part(part));
        }
        msg.content = extract_text(msg.content_parts);
    } else {
        msg.content = m.value("content", "");
    }
    return msg;
}
 
} // namespace
 
std::vector<Message> parse_messages_json(const char* json_str) {
    std::vector<Message> messages;
    if (json_str == nullptr) { return messages; }
    auto arr = nlohmann::json::parse(json_str);
    if (!arr.is_array()) { return messages; }
    for (const auto& m : arr) {
        messages.push_back(parse_one_message(m));
    }
    return messages;
}
 
bool any_message_has_images(const std::vector<Message>& messages) {
    for (const auto& m : messages) {
        if (has_images(m.content_parts)) { return true; }
    }
    return false;
}
 
} // namespace entropic