From d673cf8514ae66fbc6f50b75b9edad49bca151f6 Mon Sep 17 00:00:00 2001
From: Jon Saad-Falcon <41205309+jonsaadfalcon@users.noreply.github.com>
Date: Thu, 26 Mar 2026 17:05:33 -0700
Subject: [PATCH] feat: add imap_host parameter to GmailIMAPConnector

---
 src/openjarvis/connectors/gmail_imap.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/openjarvis/connectors/gmail_imap.py b/src/openjarvis/connectors/gmail_imap.py
index 6b498446..bcd4fbab 100644
--- a/src/openjarvis/connectors/gmail_imap.py
+++ b/src/openjarvis/connectors/gmail_imap.py
@@ -83,6 +83,7 @@ class GmailIMAPConnector(BaseConnector):
     connector_id = "gmail_imap"
     display_name = "Gmail (IMAP)"
     auth_type = "oauth"  # Reuses credential storage pattern
+    _default_imap_host = "imap.gmail.com"
 
     def __init__(
         self,
@@ -90,11 +91,13 @@ class GmailIMAPConnector(BaseConnector):
         app_password: str = "",
         credentials_path: str = "",
         *,
+        imap_host: str = "",
         max_messages: int = 500,
     ) -> None:
         self._email = email_address
         self._password = app_password
         self._credentials_path = credentials_path or _DEFAULT_CREDENTIALS_PATH
+        self._imap_host = imap_host or self._default_imap_host
         self._max_messages = max_messages
         self._items_synced = 0
         self._items_total = 0
@@ -144,7 +147,7 @@ class GmailIMAPConnector(BaseConnector):
         if not em or not pw:
             return
 
-        imap = imaplib.IMAP4_SSL("imap.gmail.com")
+        imap = imaplib.IMAP4_SSL(self._imap_host)
         try:
             imap.login(em, pw)
         except imaplib.IMAP4.error as exc: