tokenizer.hpp

Go to the documentation of this file.

#pragma once
 
// SPDX-License-Identifier: Apache-2.0
// Copyright 2026 Lloyal Labs
 
#include "common.hpp"
#include <cstdint>
#include <llama/llama.h>
#include <string>
#include <vector>
 
namespace lloyal::tokenizer {
 
// ===== TOKENIZATION (TEXT → TOKENS) =====
 
inline std::vector<llama_token> tokenize(const llama_vocab *vocab,
                                         const std::string &text,
                                         bool add_special, bool parse_special) {
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::tokenize] ERROR: vocab is null");
    return {};
  }
 
  // Two-pass algorithm for safety:
  // Pass 1: Determine required buffer size (negative return = size needed)
  const int n_tokens =
      -llama_tokenize(vocab, text.c_str(), static_cast<int32_t>(text.length()),
                      nullptr, // null buffer to get size
                      0, add_special, parse_special);
 
  if (n_tokens <= 0) {
    LLOYAL_LOG_DEBUG("[tokenizer::tokenize] WARNING: Empty tokenization result "
                     "for text: '%.50s...'",
                     text.c_str());
    return {};
  }
 
  // Pass 2: Actual tokenization
  std::vector<llama_token> tokens(n_tokens);
  const int n_tokenized =
      llama_tokenize(vocab, text.c_str(), static_cast<int32_t>(text.length()),
                     tokens.data(), n_tokens, add_special, parse_special);
 
  if (n_tokenized != n_tokens) {
    LLOYAL_LOG_DEBUG("[tokenizer::tokenize] ERROR: Token count mismatch "
                     "(expected %d, got %d)",
                     n_tokens, n_tokenized);
    return {};
  }
 
  LLOYAL_LOG_DEBUG("[tokenizer::tokenize] Tokenized %zu bytes → %d tokens",
                   text.length(), n_tokens);
  return tokens;
}
 
// ===== DETOKENIZATION (TOKENS → TEXT) =====
 
inline std::string detokenize(const llama_vocab *vocab, llama_token token,
                              bool special) {
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::detokenize] ERROR: vocab is null");
    return "";
  }
 
  // Two-pass algorithm (vendored from llama.cpp/common/common.cpp)
  std::string piece(64, '\0');
 
  const int n_chars =
      llama_token_to_piece(vocab, token, &piece[0], piece.size(), 0, special);
 
  if (n_chars < 0) {
    // Buffer too small, resize and retry
    piece.resize(-n_chars);
    int check =
        llama_token_to_piece(vocab, token, &piece[0], piece.size(), 0, special);
    if (check != -n_chars) {
      LLOYAL_LOG_DEBUG(
          "[tokenizer::detokenize] ERROR: Inconsistent sizing for token %d",
          token);
      return "";
    }
  } else {
    piece.resize(n_chars);
  }
 
  return piece;
}
 
inline std::string detokenize_batch(const llama_vocab *vocab,
                                    const llama_token *tokens, int32_t n_tokens,
                                    bool remove_special, bool unparse_special) {
  if (!vocab || !tokens || n_tokens <= 0) {
    LLOYAL_LOG_DEBUG("[tokenizer::detokenize_batch] ERROR: Invalid parameters "
                     "(vocab=%p, tokens=%p, n_tokens=%d)",
                     vocab, tokens, n_tokens);
    return "";
  }
 
  // Two-pass algorithm for safety:
  // Pass 1: Determine required buffer size (negative return = size needed)
  int32_t required_size = llama_detokenize(vocab, tokens, n_tokens,
                                           nullptr, // null buffer to get size
                                           0, remove_special, unparse_special);
 
  if (required_size < 0) {
    // Negative return means we need abs(required_size) bytes
    required_size = -required_size;
  }
 
  if (required_size == 0) {
    LLOYAL_LOG_DEBUG("[tokenizer::detokenize_batch] WARNING: Empty "
                     "detokenization result for %d tokens",
                     n_tokens);
    return "";
  }
 
  // Pass 2: Actual detokenization
  std::vector<char> buffer(required_size + 1); // +1 for null terminator
  int32_t written =
      llama_detokenize(vocab, tokens, n_tokens, buffer.data(), required_size,
                       remove_special, unparse_special);
 
  if (written < 0) {
    LLOYAL_LOG_DEBUG("[tokenizer::detokenize_batch] ERROR: Detokenization "
                     "failed (needed %d bytes, got %d)",
                     required_size, written);
    return "";
  }
 
  std::string result(buffer.data(), written);
  LLOYAL_LOG_DEBUG(
      "[tokenizer::detokenize_batch] Detokenized %d tokens → %zu bytes",
      n_tokens, result.size());
  return result;
}
 
// ===== VOCABULARY QUERIES =====
 
inline const llama_vocab *get_vocab(const llama_model *model) {
  if (!model) {
    LLOYAL_LOG_DEBUG("[tokenizer::get_vocab] ERROR: NULL model");
    return nullptr;
  }
 
  const llama_vocab *vocab = llama_model_get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG(
        "[tokenizer::get_vocab] ERROR: llama_model_get_vocab returned NULL");
  }
 
  return vocab;
}
 
inline bool is_eog(const llama_vocab *vocab, llama_token token) {
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::is_eog] ERROR: vocab is null");
    return false;
  }
 
  return llama_vocab_is_eog(vocab, token);
}
 
inline int32_t vocab_size(const llama_vocab *vocab) {
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::vocab_size] ERROR: vocab is null");
    return 0;
  }
 
  return llama_vocab_n_tokens(vocab);
}
 
// ===== MODEL-ACCEPTING CONVENIENCE OVERLOADS =====
//
// These overloads accept llama_model* and handle vocab extraction + metadata
// queries internally. They delegate to the vocab-accepting primitives above.
//
// Benefits:
// - Eliminate boilerplate (vocab extraction, add_bos queries) in calling code
// - Reduce code duplication across projects
// - Backwards compatible - existing code unchanged
 
inline std::vector<llama_token> tokenize(const llama_model *model,
                                         const std::string &text) {
  if (!model) {
    LLOYAL_LOG_DEBUG("[tokenizer::tokenize] ERROR: model is null");
    return {};
  }
 
  const llama_vocab *vocab = get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::tokenize] ERROR: get_vocab returned null");
    return {};
  }
 
  bool add_bos = llama_vocab_get_add_bos(vocab);
  return tokenize(vocab, text, add_bos, true);
}
 
inline std::string detokenize(const llama_model *model, llama_token token,
                              bool special = true) {
  if (!model) {
    LLOYAL_LOG_DEBUG("[tokenizer::detokenize] ERROR: model is null");
    return "";
  }
 
  const llama_vocab *vocab = get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::detokenize] ERROR: get_vocab returned null");
    return "";
  }
 
  return detokenize(vocab, token, special);
}
 
inline std::string detokenize_batch(const llama_model *model,
                                    const std::vector<llama_token> &tokens,
                                    bool remove_special = false,
                                    bool unparse_special = true) {
  if (!model) {
    LLOYAL_LOG_DEBUG(
        "[tokenizer::detokenize_batch] ERROR: model is null");
    return "";
  }
 
  const llama_vocab *vocab = get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG(
        "[tokenizer::detokenize_batch] ERROR: get_vocab returned null");
    return "";
  }
 
  return detokenize_batch(vocab, tokens.data(),
                          static_cast<int32_t>(tokens.size()), remove_special,
                          unparse_special);
}
 
inline std::string detokenize_batch(const llama_model *model,
                                    const llama_token *tokens, int32_t n_tokens,
                                    bool remove_special, bool unparse_special) {
  if (!model) {
    LLOYAL_LOG_DEBUG(
        "[tokenizer::detokenize_batch] ERROR: model is null");
    return "";
  }
 
  const llama_vocab *vocab = get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG(
        "[tokenizer::detokenize_batch] ERROR: get_vocab returned null");
    return "";
  }
 
  return detokenize_batch(vocab, tokens, n_tokens, remove_special,
                          unparse_special);
}
 
inline bool is_eog(const llama_model *model, llama_token token) {
  if (!model) {
    LLOYAL_LOG_DEBUG("[tokenizer::is_eog] ERROR: model is null");
    return false;
  }
 
  const llama_vocab *vocab = get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::is_eog] ERROR: get_vocab returned null");
    return false;
  }
 
  return is_eog(vocab, token);
}
 
inline int32_t vocab_size(const llama_model *model) {
  if (!model) {
    LLOYAL_LOG_DEBUG("[tokenizer::vocab_size] ERROR: model is null");
    return 0;
  }
 
  const llama_vocab *vocab = get_vocab(model);
  if (!vocab) {
    LLOYAL_LOG_DEBUG("[tokenizer::vocab_size] ERROR: get_vocab returned null");
    return 0;
  }
 
  return vocab_size(vocab);
}
 
} // namespace lloyal::tokenizer