XciD HF Staff commited on
Commit
f257f85
·
verified ·
1 Parent(s): 92ec493

Update src/main.rs

Browse files
Files changed (1) hide show
  1. src/main.rs +5 -9
src/main.rs CHANGED
@@ -66,8 +66,12 @@ impl HfFile {
66
  .unwrap();
67
 
68
  if !response.status().is_success() {
69
- return Err(McpError::internal_error("Failed to list files", None));
 
 
70
  }
 
 
71
  let body = response.json::<serde_json::Value>().await.unwrap();
72
  let siblings = body.get("siblings").unwrap().as_array().unwrap();
73
  let mut contents = Vec::new();
@@ -100,14 +104,6 @@ impl HfFile {
100
  .await
101
  .unwrap();
102
 
103
- if !response.status().is_success() {
104
- println!("Failed to get file content: {}", response.status());
105
- println!(
106
- "Failed to get file content: {}",
107
- response.text().await.unwrap()
108
- );
109
- return Err(McpError::internal_error("Failed to get file content", None));
110
- }
111
  Ok(CallToolResult::success(vec![Content::text(
112
  response.text().await.unwrap(),
113
  )]))
 
66
  .unwrap();
67
 
68
  if !response.status().is_success() {
69
+ return Ok(CallToolResult::success(vec![Content::text(
70
+ response.text().await.unwrap(),
71
+ )]));
72
  }
73
+
74
+ println!("Successfully listed files in HF Hub {repo_type}/{repository}");
75
  let body = response.json::<serde_json::Value>().await.unwrap();
76
  let siblings = body.get("siblings").unwrap().as_array().unwrap();
77
  let mut contents = Vec::new();
 
104
  .await
105
  .unwrap();
106
 
 
 
 
 
 
 
 
 
107
  Ok(CallToolResult::success(vec![Content::text(
108
  response.text().await.unwrap(),
109
  )]))