mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-04-27 17:56:25 +02:00
0fe94bfcf3
- Introduced Slack OAuth support with new environment variables for client ID, client secret, and redirect URI. - Implemented Slack connector routes for OAuth flow, including authorization and callback handling. - Updated configuration to support both new OAuth format and legacy token handling. - Enhanced the Slack indexer to decrypt tokens when necessary, ensuring compatibility with existing encrypted credentials. - Removed outdated Slack connector UI components and adjusted frontend logic to reflect the new integration.
72 lines
2.5 KiB
Python
72 lines
2.5 KiB
Python
import uuid
|
|
from datetime import datetime
|
|
from typing import Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, field_validator, model_validator
|
|
|
|
from app.db import SearchSourceConnectorType
|
|
from app.utils.validators import validate_connector_config
|
|
|
|
from .base import IDModel, TimestampModel
|
|
|
|
|
|
class SearchSourceConnectorBase(BaseModel):
|
|
name: str
|
|
connector_type: SearchSourceConnectorType
|
|
is_indexable: bool
|
|
last_indexed_at: datetime | None = None
|
|
config: dict[str, Any]
|
|
periodic_indexing_enabled: bool = False
|
|
indexing_frequency_minutes: int | None = None
|
|
next_scheduled_at: datetime | None = None
|
|
|
|
@field_validator("config")
|
|
@classmethod
|
|
def validate_config_for_connector_type(
|
|
cls, config: dict[str, Any], values: dict[str, Any]
|
|
) -> dict[str, Any]:
|
|
connector_type = values.data.get("connector_type")
|
|
return validate_connector_config(connector_type, config)
|
|
|
|
@model_validator(mode="after")
|
|
def validate_periodic_indexing(self):
|
|
"""Validate that periodic indexing configuration is consistent.
|
|
|
|
Supported frequencies: Any positive integer (in minutes).
|
|
Common values: 5, 15, 60 (1 hour), 360 (6 hours), 720 (12 hours), 1440 (daily), etc.
|
|
The schedule checker will handle any frequency >= 1 minute.
|
|
"""
|
|
if self.periodic_indexing_enabled:
|
|
if not self.is_indexable:
|
|
raise ValueError(
|
|
"periodic_indexing_enabled can only be True for indexable connectors"
|
|
)
|
|
if self.indexing_frequency_minutes is None:
|
|
raise ValueError(
|
|
"indexing_frequency_minutes is required when periodic_indexing_enabled is True"
|
|
)
|
|
if self.indexing_frequency_minutes <= 0:
|
|
raise ValueError("indexing_frequency_minutes must be greater than 0")
|
|
return self
|
|
|
|
|
|
class SearchSourceConnectorCreate(SearchSourceConnectorBase):
|
|
pass
|
|
|
|
|
|
class SearchSourceConnectorUpdate(BaseModel):
|
|
name: str | None = None
|
|
connector_type: SearchSourceConnectorType | None = None
|
|
is_indexable: bool | None = None
|
|
last_indexed_at: datetime | None = None
|
|
config: dict[str, Any] | None = None
|
|
periodic_indexing_enabled: bool | None = None
|
|
indexing_frequency_minutes: int | None = None
|
|
next_scheduled_at: datetime | None = None
|
|
|
|
|
|
class SearchSourceConnectorRead(SearchSourceConnectorBase, IDModel, TimestampModel):
|
|
search_space_id: int
|
|
user_id: uuid.UUID
|
|
|
|
model_config = ConfigDict(from_attributes=True)
|