tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 5 Packages Projects Releases Wiki Activity

fix: use the pre-fetched mime type (#2417)

Browse Source
This commit is contained in:
Yingjie He
2025-05-01 14:42:13 -07:00
committed by GitHub
parent cf1a2db78a
commit 76cb77ebbb
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose-mcp/src/google_drive/mod.rs
+3 -5
View File
@@ -1153,6 +1153,7 @@ impl GoogleDriveRouter {
async fn get_google_file( async fn get_google_file(
&self, &self,
uri: &str, uri: &str,
mime_type: &str,
include_images: bool, include_images: bool,
) -> Result<Vec<Content>, ToolError> { ) -> Result<Vec<Content>, ToolError> {
let result = self let result = self
@@ -1171,10 +1172,6 @@ impl GoogleDriveRouter {
uri, e uri, e
))), ))),
Ok(r) => { Ok(r) => {
let file = r.1;
let mime_type = file
.mime_type
.unwrap_or("application/octet-stream".to_string());
if mime_type.starts_with("text/") || mime_type == "application/json" { if mime_type.starts_with("text/") || mime_type == "application/json" {
if let Ok(body) = r.0.into_body().collect().await { if let Ok(body) = r.0.into_body().collect().await {
if let Ok(response) = String::from_utf8(body.to_bytes().to_vec()) { if let Ok(response) = String::from_utf8(body.to_bytes().to_vec()) {
@@ -1277,7 +1274,8 @@ impl GoogleDriveRouter {
self.export_google_file(&drive_uri, &mime_type, include_images) self.export_google_file(&drive_uri, &mime_type, include_images)
.await .await
} else { } else {
self.get_google_file(&drive_uri, include_images).await self.get_google_file(&drive_uri, &mime_type, include_images)
.await
} }
} }