Add Google Keep Integration

auth/scopes.py

@@ -43,6 +43,10 @@
 CHAT_SPACES_SCOPE = "https://www.googleapis.com/auth/chat.spaces"
 CHAT_SPACES_READONLY_SCOPE = "https://www.googleapis.com/auth/chat.spaces.readonly"
 
+# Google Keep API scopes
+KEEP_READONLY_SCOPE = "https://www.googleapis.com/auth/keep.readonly"
+KEEP_WRITE_SCOPE = "https://www.googleapis.com/auth/keep"
+
 # Google Sheets API scopes
 SHEETS_READONLY_SCOPE = "https://www.googleapis.com/auth/spreadsheets.readonly"
 SHEETS_WRITE_SCOPE = "https://www.googleapis.com/auth/spreadsheets"
@@ -96,6 +100,7 @@
     DOCS_WRITE_SCOPE: {DOCS_READONLY_SCOPE},
     SHEETS_WRITE_SCOPE: {SHEETS_READONLY_SCOPE},
     SLIDES_SCOPE: {SLIDES_READONLY_SCOPE},
+    KEEP_WRITE_SCOPE: {KEEP_READONLY_SCOPE},
     TASKS_SCOPE: {TASKS_READONLY_SCOPE},
     CONTACTS_SCOPE: {CONTACTS_READONLY_SCOPE},
     CHAT_WRITE_SCOPE: {CHAT_READONLY_SCOPE},
@@ -143,6 +148,8 @@ def has_required_scopes(available_scopes, required_scopes):
 
 DRIVE_SCOPES = [DRIVE_SCOPE, DRIVE_READONLY_SCOPE, DRIVE_FILE_SCOPE]
 
+KEEP_SCOPES = [KEEP_WRITE_SCOPE, KEEP_READONLY_SCOPE]
+
 GMAIL_SCOPES = [
     GMAIL_READONLY_SCOPE,
     GMAIL_SEND_SCOPE,
@@ -194,6 +201,7 @@ def has_required_scopes(available_scopes, required_scopes):
     "sheets": SHEETS_SCOPES,
     "chat": CHAT_SCOPES,
     "forms": FORMS_SCOPES,
+    "keep": KEEP_SCOPES,
     "slides": SLIDES_SCOPES,
     "tasks": TASKS_SCOPES,
     "contacts": CONTACTS_SCOPES,
@@ -211,6 +219,7 @@ def has_required_scopes(available_scopes, required_scopes):
     "chat": [CHAT_READONLY_SCOPE, CHAT_SPACES_READONLY_SCOPE],
     "forms": [FORMS_BODY_READONLY_SCOPE, FORMS_RESPONSES_READONLY_SCOPE],
     "slides": [SLIDES_READONLY_SCOPE],
+    "keep": [KEEP_READONLY_SCOPE],
     "tasks": [TASKS_READONLY_SCOPE],
     "contacts": [CONTACTS_READONLY_SCOPE],
     "search": CUSTOM_SEARCH_SCOPES,

auth/service_decorator.py

@@ -40,6 +40,8 @@
     CHAT_WRITE_SCOPE,
     CHAT_SPACES_SCOPE,
     CHAT_SPACES_READONLY_SCOPE,
+    KEEP_WRITE_SCOPE,
+    KEEP_READONLY_SCOPE,
     FORMS_BODY_SCOPE,
     FORMS_BODY_READONLY_SCOPE,
     FORMS_RESPONSES_READONLY_SCOPE,
@@ -405,6 +407,7 @@ def _remove_user_email_arg_from_docstring(docstring: str) -> str:
     "drive": {"service": "drive", "version": "v3"},
     "calendar": {"service": "calendar", "version": "v3"},
     "docs": {"service": "docs", "version": "v1"},
+    "keep": {"service": "keep", "version": "v1"},
     "sheets": {"service": "sheets", "version": "v4"},
     "chat": {"service": "chat", "version": "v1"},
     "forms": {"service": "forms", "version": "v1"},
@@ -434,6 +437,9 @@ def _remove_user_email_arg_from_docstring(docstring: str) -> str:
     # Calendar scopes
     "calendar_read": CALENDAR_READONLY_SCOPE,
     "calendar_events": CALENDAR_EVENTS_SCOPE,
+    # Keep scopes
+    "keep_read": KEEP_READONLY_SCOPE,
+    "keep_write": KEEP_WRITE_SCOPE,
     # Sheets scopes
     "sheets_read": SHEETS_READONLY_SCOPE,
     "sheets_write": SHEETS_WRITE_SCOPE,

core/api_enablement.py

@@ -1,48 +1,32 @@
 import re
-from typing import Dict, Optional, Tuple
-
-
-API_ENABLEMENT_LINKS: Dict[str, str] = {
-    "calendar-json.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=calendar-json.googleapis.com",
-    "drive.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=drive.googleapis.com",
-    "gmail.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=gmail.googleapis.com",
-    "docs.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=docs.googleapis.com",
-    "sheets.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=sheets.googleapis.com",
-    "slides.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=slides.googleapis.com",
-    "forms.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=forms.googleapis.com",
-    "tasks.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=tasks.googleapis.com",
-    "chat.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=chat.googleapis.com",
-    "customsearch.googleapis.com": "https://console.cloud.google.com/flows/enableapi?apiid=customsearch.googleapis.com",
+from typing import Dict, List, Optional, Tuple
+
+
+SUPPORTED_APIS: Dict[str, Tuple[List[str], str]]= {
+    "calendar-json.googleapis.com": (["calendar"], "Google Calendar"),
+    "drive.googleapis.com": (["drive"], "Google Drive"),
+    "gmail.googleapis.com": (["gmail"], "Gmail"),
+    "keep.googleapis.com": (["keep"], "Google Keep"),
+    "docs.googleapis.com": (["docs"], "Google Docs"),
+    "sheets.googleapis.com": (["sheets"], "Google Sheets"),
+    "slides.googleapis.com": (["slides"], "Google Slides"),
+    "forms.googleapis.com": (["forms"], "Google Forms"),
+    "tasks.googleapis.com": (["tasks"], "Google Tasks"),
+    "chat.googleapis.com": (["chat"], "Google Chat"),
+    "customsearch.googleapis.com": (["customsearch", "search"], "Google Custom Search"),
 }
 
 
-SERVICE_NAME_TO_API: Dict[str, str] = {
-    "Google Calendar": "calendar-json.googleapis.com",
-    "Google Drive": "drive.googleapis.com",
-    "Gmail": "gmail.googleapis.com",
-    "Google Docs": "docs.googleapis.com",
-    "Google Sheets": "sheets.googleapis.com",
-    "Google Slides": "slides.googleapis.com",
-    "Google Forms": "forms.googleapis.com",
-    "Google Tasks": "tasks.googleapis.com",
-    "Google Chat": "chat.googleapis.com",
-    "Google Custom Search": "customsearch.googleapis.com",
-}
+API_ENABLEMENT_LINK_BASE = "https://console.cloud.google.com/flows/enableapi?apiid={api}"
 
 
-INTERNAL_SERVICE_TO_API: Dict[str, str] = {
-    "calendar": "calendar-json.googleapis.com",
-    "drive": "drive.googleapis.com",
-    "gmail": "gmail.googleapis.com",
-    "docs": "docs.googleapis.com",
-    "sheets": "sheets.googleapis.com",
-    "slides": "slides.googleapis.com",
-    "forms": "forms.googleapis.com",
-    "tasks": "tasks.googleapis.com",
-    "chat": "chat.googleapis.com",
-    "customsearch": "customsearch.googleapis.com",
-    "search": "customsearch.googleapis.com",
-}
+SERVICE_NAME_TO_API: Dict[str, str] = { name: api for api, (_, name) in SUPPORTED_APIS.items() }
+
+
+INTERNAL_SERVICE_TO_API: Dict[str, str] = {}
+for api, (internal_names, _) in SUPPORTED_APIS.items():
+    for internal in internal_names:
+        INTERNAL_SERVICE_TO_API[internal] = api
 
 
 def extract_api_info_from_error(
@@ -88,8 +72,8 @@ def get_api_enablement_message(
             # Check display names (e.g., "Google Calendar")
             api_service = SERVICE_NAME_TO_API.get(service_type)
 
-    if api_service and api_service in API_ENABLEMENT_LINKS:
-        enable_link = API_ENABLEMENT_LINKS[api_service]
+    if api_service and api_service in SUPPORTED_APIS:
+        enable_link = API_ENABLEMENT_LINK_BASE.format(api=api_service)
         service_display_name = next(
             (name for name, api in SERVICE_NAME_TO_API.items() if api == api_service),
             api_service,

core/log_formatter.py

@@ -63,6 +63,7 @@ def _get_ascii_prefix(self, logger_name: str, level_name: str) -> str:
             "gcalendar.calendar_tools": "[CALENDAR]",
             "gdrive.drive_tools": "[DRIVE]",
             "gmail.gmail_tools": "[GMAIL]",
+            "gkeep.keep_tools": "[KEEP]",
             "gdocs.docs_tools": "[DOCS]",
             "gsheets.sheets_tools": "[SHEETS]",
             "gchat.chat_tools": "[CHAT]",

core/tool_tiers.yaml

@@ -80,6 +80,26 @@ docs:
     - reply_to_document_comment
     - resolve_document_comment
 
+# TODO: Keep
+keep:
+  core:
+    - create_note
+    - delete_note
+    - list_notes
+    - get_note
+    - read_note
+  extended:
+    - download_attachment
+    - set_permissions
+  complete:
+    - create_note
+    - delete_note
+    - download_attachment
+    - get_note
+    - list_notes
+    - read_note
+    - set_permissions
+
 sheets:
   core:
     - create_spreadsheet

fastmcp_server.py

@@ -141,6 +141,7 @@ def format(self, record):
 # Import all tool modules to register their @server.tool() decorators
 import gmail.gmail_tools
 import gdrive.drive_tools
+import gkeep.keep_tools
 import gcalendar.calendar_tools
 import gdocs.docs_tools
 import gsheets.sheets_tools
@@ -161,6 +162,7 @@ def format(self, record):
     "docs",
     "sheets",
     "chat",
+    "keep",
     "forms",
     "slides",
     "tasks",

gkeep/__init__.py

@@ -0,0 +1 @@
+"""Google Keep MCP Integration"""