[model-gateway] fix tokenizer encode in golang bindings (#16482)

This commit is contained in:
realWeilai
2026-01-05 11:40:09 -08:00
committed by GitHub
parent 51541404f8
commit 23849eba7b
4 changed files with 14 additions and 4 deletions
@@ -17,6 +17,7 @@ uuid = { version = "1.10", features = ["v4", "serde"] }
once_cell = "1.21.3"
futures-util = "0.3"
tracing = "0.1"
libc = "0.2.179"
[dependencies.sgl-model-gateway]
path = "../.."
@@ -160,7 +160,7 @@ pub unsafe extern "C" fn sgl_client_chat_completion_stream(
};
// Tokenize
let token_ids = match tokenizer.encode(&processed_messages.text) {
let token_ids = match tokenizer.encode(&processed_messages.text, false) {
Ok(encoding) => encoding.token_ids().to_vec(),
Err(e) => {
set_error_message(error_out, &format!("Failed to tokenize: {}", e));
@@ -115,7 +115,7 @@ pub unsafe extern "C" fn sgl_preprocess_chat_request(
};
// Tokenize the processed text
let encoding = match tokenizer.encode(&processed_messages.text) {
let encoding = match tokenizer.encode(&processed_messages.text, false) {
Ok(enc) => enc,
Err(e) => {
set_error_message(error_out, &format!("Tokenization failed: {}", e));
@@ -267,7 +267,7 @@ pub unsafe extern "C" fn sgl_preprocess_chat_request_with_tokenizer(
};
// Tokenize the processed text
let encoding = match tokenizer.encode(&processed_messages.text) {
let encoding = match tokenizer.encode(&processed_messages.text, false) {
Ok(enc) => enc,
Err(e) => {
set_error_message(error_out, &format!("Tokenization failed: {}", e));
@@ -15,6 +15,11 @@ use smg::tokenizer::{
use super::error::{SglErrorCode, set_error_message, clear_error_message};
#[cfg(target_os = "macos")]
type BooleanT = libc::boolean_t;
#[cfg(not(target_os = "macos"))]
type BooleanT = libc::c_int;
/// Opaque handle for a tokenizer instance
#[repr(C)]
pub struct TokenizerHandle {
@@ -69,6 +74,7 @@ pub unsafe extern "C" fn sgl_tokenizer_create_from_file(
/// # Arguments
/// * `handle` - Tokenizer handle (must not be null)
/// * `text` - Input text (null-terminated C string)
/// * `add_special_tokens` - Whether to add special tokens
/// * `token_ids_out` - Pointer to receive array of token IDs (must be freed with sgl_free_token_ids)
/// * `token_count_out` - Pointer to receive token count
/// * `error_out` - Optional pointer to receive error message
@@ -82,6 +88,7 @@ pub unsafe extern "C" fn sgl_tokenizer_create_from_file(
pub unsafe extern "C" fn sgl_tokenizer_encode(
handle: *mut TokenizerHandle,
text: *const c_char,
add_special_tokens: BooleanT,
token_ids_out: *mut *mut u32,
token_count_out: *mut usize,
error_out: *mut *mut c_char,
@@ -99,8 +106,10 @@ pub unsafe extern "C" fn sgl_tokenizer_encode(
}
};
let add_special_tokens_bool = add_special_tokens != 0;
let tokenizer = &(*handle).tokenizer;
match tokenizer.encode(text_str) {
match tokenizer.encode(text_str, add_special_tokens_bool) {
Ok(encoding) => {
let token_ids = encoding.token_ids();
let count = token_ids.len();