update search source connector / add gmail enum value

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

View file

@ -46,6 +46,7 @@ class DocumentType(str, Enum):
JIRA_CONNECTOR = "JIRA_CONNECTOR"
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
GOOGLE_GMAIL_CONNECTOR = "GOOGLE_GMAIL_CONNECTOR"
class SearchSourceConnectorType(str, Enum):
@ -60,6 +61,7 @@ class SearchSourceConnectorType(str, Enum):
JIRA_CONNECTOR = "JIRA_CONNECTOR"
CONFLUENCE_CONNECTOR = "CONFLUENCE_CONNECTOR"
CLICKUP_CONNECTOR = "CLICKUP_CONNECTOR"
GOOGLE_GMAIL_CONNECTOR = "GOOGLE_GMAIL_CONNECTOR"
class ChatType(str, Enum):

View file

@ -179,6 +179,14 @@ class SearchSourceConnectorBase(BaseModel):
if not config.get("CLICKUP_API_TOKEN"):
raise ValueError("CLICKUP_API_TOKEN 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