Spaces:
Sleeping
Sleeping
Update src/main.rs
Browse files- 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
|
|
|
|
|
|
|
| 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 |
)]))
|