adapter_base.h

Go to the documentation of this file.

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <entropic/types/message.h>
#include <entropic/types/tool_call.h>
 
#include <optional>
#include <string>
#include <string_view>
#include <unordered_set>
#include <vector>
 
namespace entropic {
 
struct ParseResult {
    std::string cleaned_content;       
    std::vector<ToolCall> tool_calls;  
};
 
class ChatAdapter {
public:
    ChatAdapter(std::string tier_name, std::string identity_prompt);
    virtual ~ChatAdapter() = default;
 
    /* ── System prompt assembly ──────────────────────────── */
 
    std::string format_system_prompt(
        const std::string& base_prompt,
        const std::vector<std::string>& tool_jsons) const;
 
    /* ── Tool call parsing (subclass overrides) ──────────── */
 
    virtual ParseResult parse_tool_calls(
        const std::string& content) const = 0;
 
    /* ── Tool result formatting ──────────────────────────── */
 
    virtual Message format_tool_result(
        const ToolCall& tool_call,
        const std::string& result) const;
 
    /* ── Tool formatting (subclass can override) ─────────── */
 
    virtual std::string format_tools(
        const std::vector<std::string>& tool_jsons) const;
 
    /* ── Response completeness ───────────────────────────── */
 
    bool is_response_complete(
        const std::string& content,
        const std::vector<ToolCall>& tool_calls) const;
 
    virtual std::string chat_format() const = 0;
 
protected:
    /* ── Shared parsing primitives ───────────────────────── */
 
    std::vector<ToolCall> parse_tagged_tool_calls(
        const std::string& content) const;
 
    std::vector<ToolCall> parse_bare_json_tool_calls(
        const std::string& content) const;
 
    std::string extract_thinking(const std::string& content) const;
 
    std::string strip_think_blocks(const std::string& content) const;
 
    std::optional<ToolCall> try_recover_json(
        const std::string& json_str) const;
 
    std::optional<ToolCall> parse_single_tool_call(
        const std::string& json_str) const;
 
    std::string tier_name_;          
    std::string identity_prompt_;    
    mutable std::unordered_set<std::string> tool_prefixes_; 
 
public:
    /* ── Vision / multimodal (v1.9.11) ────────────────────── */
 
    virtual std::string format_system_with_vision(
        const std::string& base_system,
        bool has_vision) const;
 
    virtual std::string format_content_parts(
        const std::vector<ContentPart>& parts) const;
};
 
} // namespace entropic