From 6c01bb46e668829e8b9a399cace905993ee90150 Mon Sep 17 00:00:00 2001 From: EricLBuehler Date: Thu, 18 Jul 2024 22:15:04 -0400 Subject: [PATCH] Trim token --- src/openai/pipelines/mod.rs | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/src/openai/pipelines/mod.rs b/src/openai/pipelines/mod.rs index 41cca9b..6b39475 100644 --- a/src/openai/pipelines/mod.rs +++ b/src/openai/pipelines/mod.rs @@ -73,14 +73,16 @@ pub(crate) fn get_token( hf_token_path: Option, ) -> Result { Ok(match (hf_token, hf_token_path) { - (Some(envvar), None) => try_api!(env::var(envvar)), - (None, Some(path)) => try_api!(fs::read_to_string(path)), + (Some(envvar), None) => try_api!(env::var(envvar)).trim().to_string(), + (None, Some(path)) => try_api!(fs::read_to_string(path)).trim().to_string(), (None, None) => try_api!(fs::read_to_string(format!( "{}/.cache/huggingface/token", dirs::home_dir() .ok_or(APIError::new_str("No home directory"))? .display() - ))), + ))) + .trim() + .to_string(), _ => { return Err(APIError::new_str( "Do not specify `hf_token` and `hf_token_path` at the same time.",