qwen36_adapter.cpp

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#include "qwen36_adapter.h"
 
#include <entropic/types/logging.h>
 
#include <nlohmann/json.hpp>
 
#include <regex>
#include <sstream>
 
namespace entropic {
 
namespace {
auto logger = entropic::log::get("inference.adapter.qwen36");
 
constexpr const char* TOOL_RESULT_SUFFIX =
    "Continue. Batch multiple tool calls in one response when possible.";
 
std::string generate_uuid() {
    static std::atomic<uint64_t> counter{0};
    return "tc-" + std::to_string(counter.fetch_add(1, std::memory_order_relaxed));
}
 
constexpr const char* VISION_INSTRUCTION =
    "\n\nYou can see and analyze images. When the user shares an "
    "image, describe what you observe before responding to their "
    "question.";
 
} // anonymous namespace
 
// ── Tool call parsing ──────────────────────────────────────
 
ParseResult Qwen36Adapter::parse_tool_calls(const std::string& content) const {
    ParseResult result;
 
    auto calls = parse_xml_function_calls(content);
    if (calls.empty()) {
        calls = parse_tagged_tool_calls(content);
    }
 
    result.tool_calls = std::move(calls);
    result.cleaned_content = clean_content(content);
    return result;
}
 
std::vector<ToolCall> Qwen36Adapter::parse_xml_function_calls(
    const std::string& content) const
{
    std::vector<ToolCall> calls;
    std::regex func_pattern(R"(<function=([^>]+)>([\s\S]*?)</function>)");
 
    auto begin = std::sregex_iterator(content.begin(), content.end(), func_pattern);
    auto end = std::sregex_iterator();
 
    for (auto it = begin; it != end; ++it) {
        std::string func_name = (*it)[1].str();
        std::string func_body = (*it)[2].str();
 
        auto ns = func_name.find_first_not_of(" \t");
        auto ne = func_name.find_last_not_of(" \t");
        if (ns != std::string::npos) {
            func_name = func_name.substr(ns, ne - ns + 1);
        }
 
        ToolCall tc;
        tc.id = generate_uuid();
        tc.name = func_name;
        tc.arguments = extract_xml_parameters(func_body);
        calls.push_back(std::move(tc));
 
        logger->info("Parsed XML tool call: {}", func_name);
    }
    return calls;
}
 
std::unordered_map<std::string, std::string> Qwen36Adapter::extract_xml_parameters(
    const std::string& func_body) const
{
    std::string body = func_body;
    auto nested = body.find("<function=");
    if (nested != std::string::npos) {
        logger->warn("Truncating function body at nested <function= tag");
        body = body.substr(0, nested);
    }
 
    std::unordered_map<std::string, std::string> arguments;
    std::regex param_pattern(R"(<parameter=([^>]+)>([\s\S]*?)</parameter>)");
 
    auto begin = std::sregex_iterator(body.begin(), body.end(), param_pattern);
    auto end = std::sregex_iterator();
 
    for (auto it = begin; it != end; ++it) {
        std::string key = (*it)[1].str();
        std::string value = (*it)[2].str();
 
        auto ks = key.find_first_not_of(" \t\n\r");
        auto ke = key.find_last_not_of(" \t\n\r");
        if (ks != std::string::npos) key = key.substr(ks, ke - ks + 1);
 
        auto vs = value.find_first_not_of(" \t\n\r");
        auto ve = value.find_last_not_of(" \t\n\r");
        if (vs != std::string::npos) value = value.substr(vs, ve - vs + 1);
 
        if (key.empty() || value.empty()) {
            logger->warn("Skipping empty XML parameter: key='{}' value='{}'", key, value);
            continue;
        }
        arguments[key] = value;
    }
    return arguments;
}
 
// ── Tool result formatting ─────────────────────────────────
 
Message Qwen36Adapter::format_tool_result(
    const ToolCall& /*tool_call*/,
    const std::string& result) const
{
    Message msg;
    msg.role = "user";
    msg.content = "<tool_response>\n" + result +
                  "\n</tool_response>\n\n" + TOOL_RESULT_SUFFIX;
    return msg;
}
 
// ── Tool definition formatting ─────────────────────────────
 
std::string Qwen36Adapter::format_tools(
    const std::vector<std::string>& tool_jsons) const
{
    nlohmann::json tool_defs = nlohmann::json::array();
    for (const auto& json_str : tool_jsons) {
        try {
            auto j = nlohmann::json::parse(json_str);
            tool_defs.push_back({
                {"type", "function"},
                {"function", {
                    {"name", j.value("name", "unknown")},
                    {"description", j.value("description", "")},
                    {"parameters", j.value("inputSchema",
                        nlohmann::json::object())}
                }}
            });
        } catch (...) {
            continue;
        }
    }
 
    std::ostringstream out;
    out << "# Tools\n\n"
        << "You may call one or more functions to assist with the user query.\n"
        << "Put your final answer OUTSIDE of any tool calls.\n\n"
        << "Here are the available tools:\n"
        << "<tools>\n"
        << tool_defs.dump(2) << "\n"
        << "</tools>\n\n"
        << "For each function call, return within <tool_call></tool_call> XML tags:\n"
        << "<tool_call>\n"
        << "<function=example_function>\n"
        << "<parameter=param_name>value</parameter>\n"
        << "</function>\n"
        << "</tool_call>";
    return out.str();
}
 
// ── Content cleaning ───────────────────────────────────────
 
std::string Qwen36Adapter::clean_content(const std::string& content) const {
    std::string cleaned = std::regex_replace(content,
        std::regex(R"(<tool_call>\s*[\s\S]*?\s*</tool_call>)"), "");
    cleaned = std::regex_replace(cleaned,
        std::regex(R"(<function=[^>]+>[\s\S]*?</function>)"), "");
    cleaned = strip_think_blocks(cleaned);
    return cleaned;
}
 
// ── Vision / multimodal ────────────────────────────────────
 
std::string Qwen36Adapter::format_system_with_vision(
    const std::string& base_system,
    bool has_vision) const {
    if (!has_vision) {
        return base_system;
    }
    return base_system + VISION_INSTRUCTION;
}
 
std::string Qwen36Adapter::format_content_parts(
    const std::vector<ContentPart>& parts) const {
    return ChatAdapter::format_content_parts(parts);
}
 
} // namespace entropic