feat: hide hf_api_token from debug

Author: OlivierDehaene

Cargo.lock

@@ -73,4 +73,4 @@ mod tests {
         assert!(!compute_cap_matching(90, 86));
         assert!(!compute_cap_matching(90, 89));
     }
-}
+}

backends/candle/src/compute_cap.rs

@@ -44,6 +44,7 @@ tracing-opentelemetry = "0.21.0"
 tracing-subscriber = { version = "0.3.16", features = ["json", "env-filter"] }
 utoipa = { version = "4.0.0", features = ["axum_extras"] }
 utoipa-swagger-ui = { version = "4.0.0", features = ["axum"] }
+veil = "0.1.6"
 
 [build-dependencies]
 vergen = { version = "8.0.0", features = ["build", "git", "gitcl"] }
@@ -56,4 +57,4 @@ python = ["text-embeddings-backend/python"]
 candle = ["text-embeddings-backend/candle"]
 candle-cuda = ["candle", "text-embeddings-backend/flash-attn"]
 candle-cuda-turing = ["candle", "text-embeddings-backend/flash-attn-v1"]
-static-linking = ["text-embeddings-backend/static-linking"]
+static-linking = ["text-embeddings-backend/static-linking"]

router/Cargo.toml

@@ -24,9 +24,10 @@ use tower_http::cors::AllowOrigin;
 use tracing_subscriber::layer::SubscriberExt;
 use tracing_subscriber::util::SubscriberInitExt;
 use tracing_subscriber::{EnvFilter, Layer};
+use veil::Redact;
 
 /// App Configuration
-#[derive(Parser, Debug)]
+#[derive(Parser, Redact)]
 #[clap(author, version, about, long_about = None)]
 struct Args {
     /// The name of the model to load.
@@ -35,6 +36,7 @@ struct Args {
     /// Or it can be a local directory containing the necessary files
     /// as saved by `save_pretrained(...)` methods of transformers
     #[clap(default_value = "thenlper/gte-base", long, env)]
+    #[redact(partial)]
     model_id: String,
 
     /// The actual revision of the model if you're referring to a model
@@ -91,6 +93,7 @@ struct Args {
 
     /// Your HuggingFace hub token
     #[clap(long, env)]
+    #[redact(partial)]
     hf_api_token: Option<String>,
 
     /// The IP address to listen on
@@ -212,7 +215,7 @@ async fn main() -> Result<()> {
     tokenizer.with_padding(None);
 
     // Position IDs offset. Used for Roberta.
-    let position_offset = if &config.model_type == "xlm-roberta"  {
+    let position_offset = if &config.model_type == "xlm-roberta" {
         config.pad_token_id + 1
     } else {
         0