Spaces:
Running
Running
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 |
)]))
|