aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2026-02-02 13:14:34 +01:00
Code Issues 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

8
crates/goose-mcp/src/google_drive/mod.rs
View File

@@ -1153,6 +1153,7 @@ impl GoogleDriveRouter {
async fn get_google_file(
&self,
uri: &str,
mime_type: &str,
include_images: bool,
) -> Result<Vec<Content>, ToolError> {
let result = self
@@ -1171,10 +1172,6 @@ impl GoogleDriveRouter {
uri, e
))),
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 let Ok(body) = r.0.into_body().collect().await {
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)
.await
} else {
self.get_google_file(&drive_uri, include_images).await
self.get_google_file(&drive_uri, &mime_type, include_images)
.await
}
}