update search source connector / add gmail enum value

This commit is contained in:
CREDO23 2025-08-04 00:52:07 +02:00
parent 08f0a34ef5
commit a1bc9a1c60
2 changed files with 10 additions and 0 deletions

View file

@ -47,6 +47,7 @@ class DocumentType(str, Enum):
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR" CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR" CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
GOOGLE_CALENDAR_CONNECTOR = "GOOGLE_CALENDAR_CONNECTOR" GOOGLE_CALENDAR_CONNECTOR = "GOOGLE_CALENDAR_CONNECTOR"
GOOGLE_GMAIL_CONNECTOR = "GOOGLE_GMAIL_CONNECTOR"
class SearchSourceConnectorType(str, Enum): class SearchSourceConnectorType(str, Enum):
@ -62,6 +63,7 @@ class SearchSourceConnectorType(str, Enum):
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR" CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR" CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
GOOGLE_CALENDAR_CONNECTOR = "GOOGLE_CALENDAR_CONNECTOR" GOOGLE_CALENDAR_CONNECTOR = "GOOGLE_CALENDAR_CONNECTOR"
GOOGLE_GMAIL_CONNECTOR = "GOOGLE_GMAIL_CONNECTOR"
class ChatType(str, Enum): class ChatType(str, Enum):

View file

@ -188,6 +188,14 @@ class SearchSourceConnectorBase(BaseModel):
if key not in config or config[key] in (None, ""): if key not in config or config[key] in (None, ""):
raise ValueError(f"{key} is required and cannot be empty") raise ValueError(f"{key} is required and cannot be empty")
elif connector_type == SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR:
# Required fields for Gmail connector (same as Calendar - uses Google OAuth)
required_keys = list(GoogleAuthCredentialsBase.model_fields.keys())
for key in required_keys:
if key not in config or config[key] in (None, ""):
raise ValueError(f"{key} is required and cannot be empty")
return config return config