harald/wyoming-whisper-rs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Change token_to_str to return &str instead of String

Browse source
This commit is contained in:
0/0 2023-05-05 01:33:06 +00:00 • committed by GitHub
parent a7caf1b96a
commit d77108fca6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/whisper_ctx.rs
View file

@ -306,14 +306,14 @@ impl WhisperContext {
/// * token_id: ID of the token. /// * token_id: ID of the token.
/// ///
/// # Returns /// # Returns
/// Ok(String) on success, Err(WhisperError) on failure. /// Ok(&str) on success, Err(WhisperError) on failure.
/// ///
/// # C++ equivalent /// # C++ equivalent
/// `const char * whisper_token_to_str(struct whisper_context * ctx, whisper_token token)` /// `const char * whisper_token_to_str(struct whisper_context * ctx, whisper_token token)`
pub fn token_to_str(&self, token_id: WhisperToken) -> Result<String, WhisperError> { pub fn token_to_str(&self, token_id: WhisperToken) -> Result<&str, WhisperError> {
let c_str = self.token_to_cstr(token_id)?; let c_str = self.token_to_cstr(token_id)?;
let r_str = c_str.to_str()?; let r_str = c_str.to_str()?;
Ok(r_str.to_string()) Ok(r_str)
} }
/// Convert a token ID to a &CStr. /// Convert a token ID to a &CStr.