obot-platform · tybalex · Jun 23, 2025 · Jun 24, 2025 · Jun 24, 2025 · Jun 24, 2025
diff --git a/google/drive/.python-version b/google/drive/.python-version
@@ -0,0 +1 @@
+3.13
diff --git a/google/drive/README.md b/google/drive/README.md
@@ -0,0 +1,48 @@
+# Obot Google Drive MCP Server
+- Obot Google Drive mcp server, converted from the google-drive tool bundle.
+- supports streamable HTTP
+- tools of this mcp server expect `cred_token`(access_token of google oauth) as part of the tool input.
+
+## Installation & Running
+
+### Option 1: Using uvx (Recommended)
+install from local directory:
+```bash
+uvx --from . obot-google-drive-mcp
+```
+or stdio server:
+```bash
+uvx --from . obot-google-drive-mcp-stdio
+```
+
+### Option 2: Using uv (Development)
+Install dependencies:
+```bash
+uv pip install
+```
+
+Run the server:
+```bash
+uv run server.py
+```
+
+## Testing
+
+### Unit-test with pytest
+```
+uv run python -m pytest
+```
+
+### Integration Testing
+
+#### Get Your Access Token
+This MCP server assumes Obot will take care of the Oauth2.0 flow and supply an access token. To test locally or without Obot, you need to get an access token by yourself. I use [postman workspace](https://blog.postman.com/how-to-access-google-apis-using-oauth-in-postman/) to create and manage my tokens.
+
+#### Local Example Client
+```
+export GOOGLE_OAUTH_TOKEN=xxx
+```
+and then
+```
+uv run example_client.py
+```
diff --git a/google/drive/apis/helper.py b/google/drive/apis/helper.py
@@ -2,6 +2,7 @@
 from google.oauth2.credentials import Credentials
 from googleapiclient.discovery import build
 from googleapiclient.errors import HttpError
+from fastmcp.exceptions import ToolError
 import os
 import logging
 
@@ -32,39 +33,26 @@ def setup_logger(name, tool_name: str = "Google Drive Tool"):
 logger = setup_logger(__name__)
 
 
-def str_to_bool(value):
-    """Convert a string to a boolean."""
-    return str(value).lower() in ("true", "1", "yes")
+def get_client(cred_token: str, service_name: str = "drive", version: str = "v3"):
 
-
-def get_client(service_name: str = "drive", version: str = "v3"):
-    token = os.getenv("GOOGLE_OAUTH_TOKEN")
-    if token is None:
-        raise ValueError("GOOGLE_OAUTH_TOKEN environment variable is not set")
-
-    creds = Credentials(token=token)
+    creds = Credentials(token=cred_token)
     try:
         service = build(serviceName=service_name, version=version, credentials=creds)
         return service
     except HttpError as err:
-        print(err)
-        exit(1)
-
-
-def get_obot_user_timezone():
-    return os.getenv("OBOT_USER_TIMEZONE", "UTC").strip()
+        raise ToolError(f"HttpError retrieving google {service_name} client: {err}")
 
 
 def get_user_timezone(service):
     """Fetches the authenticated user's time zone from User's Google Calendar settings."""
     try:
         settings = service.settings().get(setting="timezone").execute()
         return settings.get(
-            "value", get_obot_user_timezone()
-        )  # Default to Obot's user timezone if not found
+            "value", "UTC"
+        )  # Default to UTC if not found
     except HttpError as err:
         if err.status_code == 403:
-            raise Exception(f"HttpError retrieving user timezone: {err}")
+            raise ToolError(f"HttpError retrieving user timezone: {err}")
         logger.error(f"HttpError retrieving user timezone: {err}")
         return "UTC"
     except Exception as e:

diff --git a/google/drive/apis/shared_drives.py b/google/drive/apis/shared_drives.py
@@ -1,6 +1,7 @@
 from typing import Optional, List
 from googleapiclient.discovery import Resource
 from googleapiclient.errors import HttpError
+from fastmcp.exceptions import ToolError
 
 
 def _generate_ids(service: Resource, count: int = 1, space: str = "drive") -> List[str]:
@@ -54,10 +55,7 @@ def list_drives(service: Resource) -> List[dict]:
         return drives
     except HttpError as error:
         error_details = error.error_details[0] if error.error_details else {}
-        print(
-            f"An error occurred. Error code: {error.resp.status}, Error message: {error_details}"
-        )
-        return []
+        raise ToolError(f"Error listing drives: {error_details}")
 
 
 def get_drive(service: Resource, drive_id: str) -> Optional[dict]:

diff --git a/google/drive/apis/workspace_file.py b/google/drive/apis/workspace_file.py
@@ -1,8 +1,8 @@
 import os
-import gptscript
+# import gptscript
 from pathlib import Path
 from apis.helper import setup_logger
-from gptscript.datasets import DatasetElement
+# from gptscript.datasets import DatasetElement
 import asyncio
 
 logger = setup_logger(__name__)

diff --git a/google/drive/commands/__init__.py b/google/drive/commands/__init__.py
diff --git a/google/drive/commands/copy_file.py b/google/drive/commands/copy_file.py
diff --git a/google/drive/commands/create_file.py b/google/drive/commands/create_file.py
diff --git a/google/drive/commands/create_folder.py b/google/drive/commands/create_folder.py
diff --git a/google/drive/commands/create_permission.py b/google/drive/commands/create_permission.py
diff --git a/google/drive/commands/create_shared_drive.py b/google/drive/commands/create_shared_drive.py
diff --git a/google/drive/commands/delete_file.py b/google/drive/commands/delete_file.py
diff --git a/google/drive/commands/delete_permission.py b/google/drive/commands/delete_permission.py
diff --git a/google/drive/commands/delete_shared_drive.py b/google/drive/commands/delete_shared_drive.py