fix: PR comment, use join()

huggingface · Feb 25, 2025 · 018545f · 018545f
1 parent 1424fa4
commit 018545f
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 10 deletions.
diff --git a/src/api/sync.rs b/src/api/sync.rs
@@ -317,11 +317,13 @@ impl ApiBuilder {
 
     fn build_headers(&self) -> HeaderMap {
         let mut headers = HeaderMap::new();
-        let mut user_agent = "".to_string();
-        self.user_agent.iter().for_each(|(key, value)| {
-            user_agent = format!("{user_agent}; {key}/{value}");
-        });
-        headers.insert(USER_AGENT, user_agent[2..].to_string());
+        let user_agent = self
+            .user_agent
+            .iter()
+            .map(|(key, value)| format!("{key}/{value}"))
+            .collect::<Vec<_>>()
+            .join("; ");
+        headers.insert(USER_AGENT, user_agent.to_string());
         if let Some(token) = &self.token {
             headers.insert(AUTHORIZATION, format!("Bearer {token}"));
         }

diff --git a/src/api/tokio.rs b/src/api/tokio.rs
@@ -342,11 +342,13 @@ impl ApiBuilder {
 
     fn build_headers(&self) -> Result<HeaderMap, ApiError> {
         let mut headers = HeaderMap::new();
-        let mut user_agent = "".to_string();
-        self.user_agent.clone().iter().for_each(|(key, value)| {
-            user_agent = format!("{user_agent}; {key}/{value}");
-        });
-        headers.insert(USER_AGENT, HeaderValue::from_str(&user_agent[2..])?);
+        let user_agent = self
+            .user_agent
+            .iter()
+            .map(|(key, value)| format!("{key}/{value}"))
+            .collect::<Vec<_>>()
+            .join("; ");
+        headers.insert(USER_AGENT, HeaderValue::from_str(&user_agent)?);
         if let Some(token) = &self.token {
             headers.insert(
                 AUTHORIZATION,