Merge pull request #378 from MODSetter/dev

feat: Moved searchconnectors association from user to searchspace

surfsense_backend/alembic/versions/23_associate_connectors_with_search_spaces.py

@@ -0,0 +1,159 @@
+"""Associate SearchSourceConnector with SearchSpace instead of User
+
+Revision ID: '23'
+Revises: '22'
+Create Date: 2025-01-10 12:00:00.000000
+
+"""
+
+from collections.abc import Sequence
+
+import sqlalchemy as sa
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "23"
+down_revision: str | None = "22"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """
+    Add search_space_id to SearchSourceConnector and update unique constraint.
+
+    Changes:
+    1. Add search_space_id column (nullable initially)
+    2. Populate search_space_id with user's first search space
+    3. Make search_space_id NOT NULL
+    4. Add foreign key constraint
+    5. Drop old unique constraint (user_id, connector_type)
+    6. Add new unique constraint (search_space_id, user_id, connector_type)
+    """
+
+    from sqlalchemy import inspect
+
+    conn = op.get_bind()
+    inspector = inspect(conn)
+
+    # Get existing columns
+    columns = [col["name"] for col in inspector.get_columns("search_source_connectors")]
+
+    # Step 1: Add search_space_id column as nullable first (if it doesn't exist)
+    if "search_space_id" not in columns:
+        op.add_column(
+            "search_source_connectors",
+            sa.Column("search_space_id", sa.Integer(), nullable=True),
+        )
+
+    # Step 2: Populate search_space_id with each user's first search space
+    # This ensures existing connectors are assigned to a valid search space
+    op.execute(
+        """
+        UPDATE search_source_connectors ssc
+        SET search_space_id = (
+            SELECT id 
+            FROM searchspaces ss 
+            WHERE ss.user_id = ssc.user_id 
+            ORDER BY ss.created_at ASC 
+            LIMIT 1
+        )
+        WHERE search_space_id IS NULL
+        """
+    )
+
+    # Step 3: Make search_space_id NOT NULL
+    op.alter_column(
+        "search_source_connectors",
+        "search_space_id",
+        nullable=False,
+    )
+
+    # Step 4: Add foreign key constraint (if it doesn't exist)
+    foreign_keys = [
+        fk["name"] for fk in inspector.get_foreign_keys("search_source_connectors")
+    ]
+    if "fk_search_source_connectors_search_space_id" not in foreign_keys:
+        op.create_foreign_key(
+            "fk_search_source_connectors_search_space_id",
+            "search_source_connectors",
+            "searchspaces",
+            ["search_space_id"],
+            ["id"],
+            ondelete="CASCADE",
+        )
+
+    # Step 5: Drop the old unique constraint (user_id, connector_type) if it exists
+    unique_constraints = [
+        uc["name"]
+        for uc in inspector.get_unique_constraints("search_source_connectors")
+    ]
+    if "uq_user_connector_type" in unique_constraints:
+        op.drop_constraint(
+            "uq_user_connector_type",
+            "search_source_connectors",
+            type_="unique",
+        )
+
+    # Step 6: Create new unique constraint (search_space_id, user_id, connector_type) if it doesn't exist
+    if "uq_searchspace_user_connector_type" not in unique_constraints:
+        op.create_unique_constraint(
+            "uq_searchspace_user_connector_type",
+            "search_source_connectors",
+            ["search_space_id", "user_id", "connector_type"],
+        )
+
+
+def downgrade() -> None:
+    """
+    Revert SearchSourceConnector association back to User only.
+
+    WARNING: This downgrade may result in data loss if multiple connectors
+    of the same type exist for a user across different search spaces.
+    """
+
+    from sqlalchemy import inspect
+
+    conn = op.get_bind()
+    inspector = inspect(conn)
+
+    # Get existing constraints and columns
+    unique_constraints = [
+        uc["name"]
+        for uc in inspector.get_unique_constraints("search_source_connectors")
+    ]
+    foreign_keys = [
+        fk["name"] for fk in inspector.get_foreign_keys("search_source_connectors")
+    ]
+    columns = [col["name"] for col in inspector.get_columns("search_source_connectors")]
+
+    # Step 1: Drop the new unique constraint if it exists
+    if "uq_searchspace_user_connector_type" in unique_constraints:
+        op.drop_constraint(
+            "uq_searchspace_user_connector_type",
+            "search_source_connectors",
+            type_="unique",
+        )
+
+    # Step 2: Recreate the old unique constraint (user_id, connector_type) if it doesn't exist
+    # NOTE: This will fail if there are duplicate (user_id, connector_type) combinations
+    # Manual cleanup may be required before downgrading
+    if "uq_user_connector_type" not in unique_constraints:
+        op.create_unique_constraint(
+            "uq_user_connector_type",
+            "search_source_connectors",
+            ["user_id", "connector_type"],
+        )
+
+    # Step 3: Drop the foreign key constraint if it exists
+    if "fk_search_source_connectors_search_space_id" in foreign_keys:
+        op.drop_constraint(
+            "fk_search_source_connectors_search_space_id",
+            "search_source_connectors",
+            type_="foreignkey",
+        )
+
+    # Step 4: Drop the search_space_id column if it exists
+    if "search_space_id" in columns:
+        op.drop_column("search_source_connectors", "search_space_id")

surfsense_backend/alembic/versions/24_fix_null_chat_types.py

@@ -0,0 +1,39 @@
+"""Fix NULL chat types by setting them to QNA
+
+Revision ID: 24
+Revises: 23
+Create Date: 2025-01-10 14:00:00.000000
+
+"""
+
+from collections.abc import Sequence
+
+from alembic import op
+
+# revision identifiers, used by Alembic.
+revision: str = "24"
+down_revision: str | None = "23"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
+
+
+def upgrade() -> None:
+    """
+    Fix any chats with NULL type values by setting them to QNA.
+    This handles edge cases from previous migrations where type values were not properly migrated.
+    """
+    # Update any NULL type values to QNA (the default chat type)
+    op.execute(
+        """
+        UPDATE chats
+        SET type = 'QNA'
+        WHERE type IS NULL
+        """
+    )
+
+
+def downgrade() -> None:
+    """
+    No downgrade necessary - we can't restore NULL values as we don't know which ones were NULL.
+    """
+    pass

surfsense_backend/app/agents/researcher/nodes.py

@@ -1010,7 +1010,10 @@ async def fetch_relevant_documents(
                         source_object,
                         tavily_chunks,
                     ) = await connector_service.search_tavily(
-                        user_query=reformulated_query, user_id=user_id, top_k=top_k
+                        user_query=reformulated_query,
+                        user_id=user_id,
+                        search_space_id=search_space_id,
+                        top_k=top_k,
                     )
 
                     # Add to sources and raw documents
@@ -1037,6 +1040,7 @@ async def fetch_relevant_documents(
                     ) = await connector_service.search_linkup(
                         user_query=reformulated_query,
                         user_id=user_id,
+                        search_space_id=search_space_id,
                         mode=linkup_mode,
                     )
 

surfsense_backend/app/connectors/google_calendar_connector.py

@@ -31,15 +31,20 @@ class GoogleCalendarConnector:
         credentials: Credentials,
         session: AsyncSession,
         user_id: str,
+        connector_id: int | None = None,
     ):
         """
         Initialize the GoogleCalendarConnector class.
         Args:
             credentials: Google OAuth Credentials object
+            session: Database session for updating connector
+            user_id: User ID (kept for backward compatibility)
+            connector_id: Optional connector ID for direct updates
         """
         self._credentials = credentials
         self._session = session
         self._user_id = user_id
+        self._connector_id = connector_id
         self.service = None
 
     async def _get_credentials(
@@ -84,6 +89,14 @@ class GoogleCalendarConnector:
                 self._credentials.refresh(Request())
                 # Update the connector config in DB
                 if self._session:
+                    # Use connector_id if available, otherwise fall back to user_id query
+                    if self._connector_id:
+                        result = await self._session.execute(
+                            select(SearchSourceConnector).filter(
+                                SearchSourceConnector.id == self._connector_id
+                            )
+                        )
+                    else:
                         result = await self._session.execute(
                             select(SearchSourceConnector).filter(
                                 SearchSourceConnector.user_id == self._user_id,
@@ -94,7 +107,7 @@ class GoogleCalendarConnector:
                     connector = result.scalars().first()
                     if connector is None:
                         raise RuntimeError(
-                            "GOOGLE_CALENDAR_CONNECTOR connector not found for current user; cannot persist refreshed token."
+                            "GOOGLE_CALENDAR_CONNECTOR connector not found; cannot persist refreshed token."
                         )
                     connector.config = json.loads(self._credentials.to_json())
                     flag_modified(connector, "config")

surfsense_backend/app/connectors/google_gmail_connector.py

@@ -30,15 +30,20 @@ class GoogleGmailConnector:
         credentials: Credentials,
         session: AsyncSession,
         user_id: str,
+        connector_id: int | None = None,
     ):
         """
         Initialize the GoogleGmailConnector class.
         Args:
             credentials: Google OAuth Credentials object
+            session: Database session for updating connector
+            user_id: User ID (kept for backward compatibility)
+            connector_id: Optional connector ID for direct updates
         """
         self._credentials = credentials
         self._session = session
         self._user_id = user_id
+        self._connector_id = connector_id
         self.service = None
 
     async def _get_credentials(
@@ -83,6 +88,14 @@ class GoogleGmailConnector:
                 self._credentials.refresh(Request())
                 # Update the connector config in DB
                 if self._session:
+                    # Use connector_id if available, otherwise fall back to user_id query
+                    if self._connector_id:
+                        result = await self._session.execute(
+                            select(SearchSourceConnector).filter(
+                                SearchSourceConnector.id == self._connector_id
+                            )
+                        )
+                    else:
                         result = await self._session.execute(
                             select(SearchSourceConnector).filter(
                                 SearchSourceConnector.user_id == self._user_id,
@@ -93,7 +106,7 @@ class GoogleGmailConnector:
                     connector = result.scalars().first()
                     if connector is None:
                         raise RuntimeError(
-                            "GMAIL connector not found for current user; cannot persist refreshed token."
+                            "GMAIL connector not found; cannot persist refreshed token."
                         )
                     connector.config = json.loads(self._credentials.to_json())
                     flag_modified(connector, "config")

surfsense_backend/app/db.py

@@ -234,12 +234,23 @@ class SearchSpace(BaseModel, TimestampMixin):
         order_by="Log.id",
         cascade="all, delete-orphan",
     )
+    search_source_connectors = relationship(
+        "SearchSourceConnector",
+        back_populates="search_space",
+        order_by="SearchSourceConnector.id",
+        cascade="all, delete-orphan",
+    )
 
 
 class SearchSourceConnector(BaseModel, TimestampMixin):
     __tablename__ = "search_source_connectors"
     __table_args__ = (
-        UniqueConstraint("user_id", "connector_type", name="uq_user_connector_type"),
+        UniqueConstraint(
+            "search_space_id",
+            "user_id",
+            "connector_type",
+            name="uq_searchspace_user_connector_type",
+        ),
     )
 
     name = Column(String(100), nullable=False, index=True)
@@ -248,10 +259,16 @@ class SearchSourceConnector(BaseModel, TimestampMixin):
     last_indexed_at = Column(TIMESTAMP(timezone=True), nullable=True)
     config = Column(JSON, nullable=False)
 
+    search_space_id = Column(
+        Integer, ForeignKey("searchspaces.id", ondelete="CASCADE"), nullable=False
+    )
+    search_space = relationship(
+        "SearchSpace", back_populates="search_source_connectors"
+    )
+
     user_id = Column(
         UUID(as_uuid=True), ForeignKey("user.id", ondelete="CASCADE"), nullable=False
     )
-    user = relationship("User", back_populates="search_source_connectors")
 
 
 class LLMConfig(BaseModel, TimestampMixin):
@@ -304,9 +321,6 @@ if config.AUTH_TYPE == "GOOGLE":
             "OAuthAccount", lazy="joined"
         )
         search_spaces = relationship("SearchSpace", back_populates="user")
-        search_source_connectors = relationship(
-            "SearchSourceConnector", back_populates="user"
-        )
         llm_configs = relationship(
             "LLMConfig",
             back_populates="user",
@@ -338,9 +352,6 @@ else:
 
     class User(SQLAlchemyBaseUserTableUUID, Base):
         search_spaces = relationship("SearchSpace", back_populates="user")
-        search_source_connectors = relationship(
-            "SearchSourceConnector", back_populates="user"
-        )
         llm_configs = relationship(
             "LLMConfig",
             back_populates="user",

surfsense_backend/app/routes/airtable_add_connector_route.py

@@ -217,9 +217,10 @@ async def airtable_callback(
             scope=token_json.get("scope"),
         )
 
-        # Check if connector already exists for this user
+        # Check if connector already exists for this search space and user
         existing_connector_result = await session.execute(
             select(SearchSourceConnector).filter(
+                SearchSourceConnector.search_space_id == space_id,
                 SearchSourceConnector.user_id == user_id,
                 SearchSourceConnector.connector_type
                 == SearchSourceConnectorType.AIRTABLE_CONNECTOR,
@@ -232,7 +233,9 @@ async def airtable_callback(
             existing_connector.config = credentials.to_dict()
             existing_connector.name = "Airtable Connector"
             existing_connector.is_indexable = True
-            logger.info(f"Updated existing Airtable connector for user {user_id}")
+            logger.info(
+                f"Updated existing Airtable connector for user {user_id} in space {space_id}"
+            )
         else:
             # Create new connector
             new_connector = SearchSourceConnector(
@@ -240,10 +243,13 @@ async def airtable_callback(
                 connector_type=SearchSourceConnectorType.AIRTABLE_CONNECTOR,
                 is_indexable=True,
                 config=credentials.to_dict(),
+                search_space_id=space_id,
                 user_id=user_id,
             )
             session.add(new_connector)
-            logger.info(f"Created new Airtable connector for user {user_id}")
+            logger.info(
+                f"Created new Airtable connector for user {user_id} in space {space_id}"
+            )
 
         try:
             await session.commit()

surfsense_backend/app/routes/google_calendar_add_connector_route.py

@@ -105,9 +105,10 @@ async def calendar_callback(
         creds_dict = json.loads(creds.to_json())
 
         try:
-            # Check if a connector with the same type already exists for this user
+            # Check if a connector with the same type already exists for this search space and user
             result = await session.execute(
                 select(SearchSourceConnector).filter(
+                    SearchSourceConnector.search_space_id == space_id,
                     SearchSourceConnector.user_id == user_id,
                     SearchSourceConnector.connector_type
                     == SearchSourceConnectorType.GOOGLE_CALENDAR_CONNECTOR,
@@ -117,12 +118,13 @@ async def calendar_callback(
             if existing_connector:
                 raise HTTPException(
                     status_code=409,
-                    detail="A GOOGLE_CALENDAR_CONNECTOR connector already exists. Each user can have only one connector of each type.",
+                    detail="A GOOGLE_CALENDAR_CONNECTOR connector already exists in this search space. Each search space can have only one connector of each type per user.",
                 )
             db_connector = SearchSourceConnector(
                 name="Google Calendar Connector",
                 connector_type=SearchSourceConnectorType.GOOGLE_CALENDAR_CONNECTOR,
                 config=creds_dict,
+                search_space_id=space_id,
                 user_id=user_id,
                 is_indexable=True,
             )

surfsense_backend/app/routes/google_gmail_add_connector_route.py

@@ -104,9 +104,10 @@ async def gmail_callback(
         creds_dict = json.loads(creds.to_json())
 
         try:
-            # Check if a connector with the same type already exists for this user
+            # Check if a connector with the same type already exists for this search space and user
             result = await session.execute(
                 select(SearchSourceConnector).filter(
+                    SearchSourceConnector.search_space_id == space_id,
                     SearchSourceConnector.user_id == user_id,
                     SearchSourceConnector.connector_type
                     == SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR,
@@ -116,12 +117,13 @@ async def gmail_callback(
             if existing_connector:
                 raise HTTPException(
                     status_code=409,
-                    detail="A GOOGLE_GMAIL_CONNECTOR connector already exists. Each user can have only one connector of each type.",
+                    detail="A GOOGLE_GMAIL_CONNECTOR connector already exists in this search space. Each search space can have only one connector of each type per user.",
                 )
             db_connector = SearchSourceConnector(
                 name="Google Gmail Connector",
                 connector_type=SearchSourceConnectorType.GOOGLE_GMAIL_CONNECTOR,
                 config=creds_dict,
+                search_space_id=space_id,
                 user_id=user_id,
                 is_indexable=True,
             )

surfsense_backend/app/routes/luma_add_connector_route.py

@@ -47,9 +47,10 @@ async def add_luma_connector(
         HTTPException: If connector already exists or validation fails
     """
     try:
-        # Check if a Luma connector already exists for this user
+        # Check if a Luma connector already exists for this search space and user
         result = await session.execute(
             select(SearchSourceConnector).filter(
+                SearchSourceConnector.search_space_id == request.space_id,
                 SearchSourceConnector.user_id == user.id,
                 SearchSourceConnector.connector_type
                 == SearchSourceConnectorType.LUMA_CONNECTOR,
@@ -64,7 +65,9 @@ async def add_luma_connector(
             await session.commit()
             await session.refresh(existing_connector)
 
-            logger.info(f"Updated existing Luma connector for user {user.id}")
+            logger.info(
+                f"Updated existing Luma connector for user {user.id} in space {request.space_id}"
+            )
 
             return {
                 "message": "Luma connector updated successfully",
@@ -77,6 +80,7 @@ async def add_luma_connector(
             name="Luma Event Connector",
             connector_type=SearchSourceConnectorType.LUMA_CONNECTOR,
             config={"api_key": request.api_key},
+            search_space_id=request.space_id,
             user_id=user.id,
             is_indexable=True,
         )
@@ -113,13 +117,15 @@ async def add_luma_connector(
 
 @router.delete("/connectors/luma")
 async def delete_luma_connector(
+    space_id: int,
     user: User = Depends(current_active_user),
     session: AsyncSession = Depends(get_async_session),
 ):
     """
-    Delete the Luma connector for the authenticated user.
+    Delete the Luma connector for the authenticated user in a specific search space.
 
     Args:
+        space_id: Search space ID
         user: Current authenticated user
         session: Database session
 
@@ -132,6 +138,7 @@ async def delete_luma_connector(
     try:
         result = await session.execute(
             select(SearchSourceConnector).filter(
+                SearchSourceConnector.search_space_id == space_id,
                 SearchSourceConnector.user_id == user.id,
                 SearchSourceConnector.connector_type
                 == SearchSourceConnectorType.LUMA_CONNECTOR,
@@ -165,13 +172,15 @@ async def delete_luma_connector(
 
 @router.get("/connectors/luma/test")
 async def test_luma_connector(
+    space_id: int,
     user: User = Depends(current_active_user),
     session: AsyncSession = Depends(get_async_session),
 ):
     """
-    Test the Luma connector for the authenticated user.
+    Test the Luma connector for the authenticated user in a specific search space.
 
     Args:
+        space_id: Search space ID
         user: Current authenticated user
         session: Database session
 
@@ -182,9 +191,10 @@ async def test_luma_connector(
         HTTPException: If connector doesn't exist or test fails
     """
     try:
-        # Get the Luma connector for this user
+        # Get the Luma connector for this search space and user
         result = await session.execute(
             select(SearchSourceConnector).filter(
+                SearchSourceConnector.search_space_id == space_id,
                 SearchSourceConnector.user_id == user.id,
                 SearchSourceConnector.connector_type
                 == SearchSourceConnectorType.LUMA_CONNECTOR,

surfsense_backend/app/routes/search_source_connectors_routes.py

@@ -1,13 +1,13 @@
 """
 SearchSourceConnector routes for CRUD operations:
 POST /search-source-connectors/ - Create a new connector
-GET /search-source-connectors/ - List all connectors for the current user
+GET /search-source-connectors/ - List all connectors for the current user (optionally filtered by search space)
 GET /search-source-connectors/{connector_id} - Get a specific connector
 PUT /search-source-connectors/{connector_id} - Update a specific connector
 DELETE /search-source-connectors/{connector_id} - Delete a specific connector
 POST /search-source-connectors/{connector_id}/index - Index content from a connector to a search space
 
-Note: Each user can have only one connector of each type (SERPER_API, TAVILY_API, SLACK_CONNECTOR, NOTION_CONNECTOR, GITHUB_CONNECTOR, LINEAR_CONNECTOR, DISCORD_CONNECTOR, LUMA_CONNECTOR).
+Note: Each search space can have only one connector of each type per user (based on search_space_id, user_id, and connector_type).
 """
 
 import logging
@@ -93,19 +93,26 @@ async def list_github_repositories(
 @router.post("/search-source-connectors/", response_model=SearchSourceConnectorRead)
 async def create_search_source_connector(
     connector: SearchSourceConnectorCreate,
+    search_space_id: int = Query(
+        ..., description="ID of the search space to associate the connector with"
+    ),
     session: AsyncSession = Depends(get_async_session),
     user: User = Depends(current_active_user),
 ):
     """
     Create a new search source connector.
 
-    Each user can have only one connector of each type (SERPER_API, TAVILY_API, SLACK_CONNECTOR, etc.).
+    Each search space can have only one connector of each type per user (based on search_space_id, user_id, and connector_type).
     The config must contain the appropriate keys for the connector type.
     """
     try:
-        # Check if a connector with the same type already exists for this user
+        # Check if the search space belongs to the user
+        await check_ownership(session, SearchSpace, search_space_id, user)
+
+        # Check if a connector with the same type already exists for this search space and user
         result = await session.execute(
             select(SearchSourceConnector).filter(
+                SearchSourceConnector.search_space_id == search_space_id,
                 SearchSourceConnector.user_id == user.id,
                 SearchSourceConnector.connector_type == connector.connector_type,
             )
@@ -114,9 +121,11 @@ async def create_search_source_connector(
         if existing_connector:
             raise HTTPException(
                 status_code=409,
-                detail=f"A connector with type {connector.connector_type} already exists. Each user can have only one connector of each type.",
+                detail=f"A connector with type {connector.connector_type} already exists in this search space. Each search space can have only one connector of each type per user.",
+            )
+        db_connector = SearchSourceConnector(
+            **connector.model_dump(), search_space_id=search_space_id, user_id=user.id
         )
-        db_connector = SearchSourceConnector(**connector.model_dump(), user_id=user.id)
         session.add(db_connector)
         await session.commit()
         await session.refresh(db_connector)
@@ -128,7 +137,7 @@ async def create_search_source_connector(
         await session.rollback()
         raise HTTPException(
             status_code=409,
-            detail=f"Integrity error: A connector with this type already exists. {e!s}",
+            detail=f"Integrity error: A connector with this type already exists in this search space. {e!s}",
         ) from e
     except HTTPException:
         await session.rollback()
@@ -152,13 +161,19 @@ async def read_search_source_connectors(
     session: AsyncSession = Depends(get_async_session),
     user: User = Depends(current_active_user),
 ):
-    """List all search source connectors for the current user."""
+    """List all search source connectors for the current user, optionally filtered by search space."""
     try:
         query = select(SearchSourceConnector).filter(
             SearchSourceConnector.user_id == user.id
         )
 
-        # No need to filter by search_space_id as connectors are user-owned, not search space specific
+        # Filter by search_space_id if provided
+        if search_space_id is not None:
+            # Verify the search space belongs to the user
+            await check_ownership(session, SearchSpace, search_space_id, user)
+            query = query.filter(
+                SearchSourceConnector.search_space_id == search_space_id
+            )
 
         result = await session.execute(query.offset(skip).limit(limit))
         return result.scalars().all()
@@ -255,6 +270,8 @@ async def update_search_source_connector(
         if key == "connector_type" and value != db_connector.connector_type:
             result = await session.execute(
                 select(SearchSourceConnector).filter(
+                    SearchSourceConnector.search_space_id
+                    == db_connector.search_space_id,
                     SearchSourceConnector.user_id == user.id,
                     SearchSourceConnector.connector_type == value,
                     SearchSourceConnector.id != connector_id,
@@ -264,7 +281,7 @@ async def update_search_source_connector(
             if existing_connector:
                 raise HTTPException(
                     status_code=409,
-                    detail=f"A connector with type {value} already exists. Each user can have only one connector of each type.",
+                    detail=f"A connector with type {value} already exists in this search space. Each search space can have only one connector of each type per user.",
                 )
 
         setattr(db_connector, key, value)

surfsense_backend/app/schemas/search_source_connector.py

@@ -224,6 +224,7 @@ class SearchSourceConnectorUpdate(BaseModel):
 
 
 class SearchSourceConnectorRead(SearchSourceConnectorBase, IDModel, TimestampModel):
+    search_space_id: int
     user_id: uuid.UUID
 
     model_config = ConfigDict(from_attributes=True)

surfsense_backend/app/services/connector_service.py

@@ -236,28 +236,37 @@ class ConnectorService:
         return transformed_results
 
     async def get_connector_by_type(
-        self, user_id: str, connector_type: SearchSourceConnectorType
+        self,
+        user_id: str,
+        connector_type: SearchSourceConnectorType,
+        search_space_id: int | None = None,
     ) -> SearchSourceConnector | None:
         """
-        Get a connector by type for a specific user
+        Get a connector by type for a specific user and optionally a search space
 
         Args:
             user_id: The user's ID
             connector_type: The connector type to retrieve
+            search_space_id: Optional search space ID to filter by
 
         Returns:
             Optional[SearchSourceConnector]: The connector if found, None otherwise
         """
-        result = await self.session.execute(
+        query = select(SearchSourceConnector).filter(
-            select(SearchSourceConnector).filter(
             SearchSourceConnector.user_id == user_id,
             SearchSourceConnector.connector_type == connector_type,
         )
+
+        if search_space_id is not None:
+            query = query.filter(
+                SearchSourceConnector.search_space_id == search_space_id
             )
+
+        result = await self.session.execute(query)
         return result.scalars().first()
 
     async def search_tavily(
-        self, user_query: str, user_id: str, top_k: int = 20
+        self, user_query: str, user_id: str, search_space_id: int, top_k: int = 20
     ) -> tuple:
         """
         Search using Tavily API and return both the source information and documents
@@ -265,6 +274,7 @@ class ConnectorService:
         Args:
             user_query: The user's query
             user_id: The user's ID
+            search_space_id: The search space ID
             top_k: Maximum number of results to return
 
         Returns:
@@ -272,7 +282,7 @@ class ConnectorService:
         """
         # Get Tavily connector configuration
         tavily_connector = await self.get_connector_by_type(
-            user_id, SearchSourceConnectorType.TAVILY_API
+            user_id, SearchSourceConnectorType.TAVILY_API, search_space_id
         )
 
         if not tavily_connector:
@@ -1637,7 +1647,11 @@ class ConnectorService:
         return result_object, clickup_chunks
 
     async def search_linkup(
-        self, user_query: str, user_id: str, mode: str = "standard"
+        self,
+        user_query: str,
+        user_id: str,
+        search_space_id: int,
+        mode: str = "standard",
     ) -> tuple:
         """
         Search using Linkup API and return both the source information and documents
@@ -1645,6 +1659,7 @@ class ConnectorService:
         Args:
             user_query: The user's query
             user_id: The user's ID
+            search_space_id: The search space ID
             mode: Search depth mode, can be "standard" or "deep"
 
         Returns:
@@ -1652,7 +1667,7 @@ class ConnectorService:
         """
         # Get Linkup connector configuration
         linkup_connector = await self.get_connector_by_type(
-            user_id, SearchSourceConnectorType.LINKUP_API
+            user_id, SearchSourceConnectorType.LINKUP_API, search_space_id
         )
 
         if not linkup_connector:

surfsense_backend/app/tasks/connector_indexers/google_calendar_indexer.py

@@ -113,7 +113,10 @@ async def index_google_calendar_events(
         )
 
         calendar_client = GoogleCalendarConnector(
-            credentials=credentials, session=session, user_id=user_id
+            credentials=credentials,
+            session=session,
+            user_id=user_id,
+            connector_id=connector_id,
         )
 
         # Calculate date range

surfsense_backend/app/tasks/connector_indexers/google_gmail_indexer.py

@@ -127,7 +127,9 @@ async def index_google_gmail_messages(
         )
 
         # Initialize Google gmail connector
-        gmail_connector = GoogleGmailConnector(credentials, session, user_id)
+        gmail_connector = GoogleGmailConnector(
+            credentials, session, user_id, connector_id
+        )
 
         # Fetch recent Google gmail messages
         logger.info(f"Fetching recent emails for connector {connector_id}")

surfsense_web/app/dashboard/[search_space_id]/connectors/(manage)/page.tsx

@@ -65,7 +65,7 @@ export default function ConnectorsPage() {
 	const today = new Date();
 
 	const { connectors, isLoading, error, deleteConnector, indexConnector } =
-		useSearchSourceConnectors();
+		useSearchSourceConnectors(false, parseInt(searchSpaceId));
 	const [connectorToDelete, setConnectorToDelete] = useState<number | null>(null);
 	const [indexingConnectorId, setIndexingConnectorId] = useState<number | null>(null);
 	const [datePickerOpen, setDatePickerOpen] = useState(false);
@@ -366,12 +366,7 @@ export default function ConnectorsPage() {
 										</Button>
 									</PopoverTrigger>
 									<PopoverContent className="w-auto p-0" align="start">
-										<Calendar
+										<Calendar mode="single" selected={endDate} onSelect={setEndDate} initialFocus />
-											mode="single"
-											selected={endDate}
-											onSelect={setEndDate}
-											initialFocus
-										/>
 									</PopoverContent>
 								</Popover>
 							</div>

surfsense_web/app/dashboard/[search_space_id]/connectors/[connector_id]/page.tsx

@@ -83,7 +83,7 @@ export default function EditConnectorPage() {
 	const searchSpaceId = params.search_space_id as string;
 	const connectorId = parseInt(params.connector_id as string, 10);
 
-	const { connectors, updateConnector } = useSearchSourceConnectors();
+	const { connectors, updateConnector } = useSearchSourceConnectors(false, parseInt(searchSpaceId));
 	const [connector, setConnector] = useState<SearchSourceConnector | null>(null);
 	const [isLoading, setIsLoading] = useState(true);
 	const [isSubmitting, setIsSubmitting] = useState(false);

surfsense_web/app/dashboard/[search_space_id]/connectors/add/airtable-connector/page.tsx

@@ -30,10 +30,10 @@ export default function AirtableConnectorPage() {
 	const [isConnecting, setIsConnecting] = useState(false);
 	const [doesConnectorExist, setDoesConnectorExist] = useState(false);
 
-	const { fetchConnectors } = useSearchSourceConnectors();
+	const { fetchConnectors } = useSearchSourceConnectors(true, parseInt(searchSpaceId));
 
 	useEffect(() => {
-		fetchConnectors().then((data) => {
+		fetchConnectors(parseInt(searchSpaceId)).then((data) => {
 			const connector = data.find(
 				(c: SearchSourceConnector) => c.connector_type === EnumConnectorName.AIRTABLE_CONNECTOR
 			);

surfsense_web/app/dashboard/[search_space_id]/connectors/add/clickup-connector/page.tsx

@@ -69,7 +69,7 @@ export default function ClickUpConnectorPage() {
 				last_indexed_at: null,
 			};
 
-			await createConnector(connectorData);
+			await createConnector(connectorData, parseInt(searchSpaceId));
 
 			toast.success("ClickUp connector created successfully!");
 			router.push(`/dashboard/${searchSpaceId}/connectors`);

surfsense_web/app/dashboard/[search_space_id]/connectors/add/confluence-connector/page.tsx

@@ -77,7 +77,8 @@ export default function ConfluenceConnectorPage() {
 	const onSubmit = async (values: ConfluenceConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.CONFLUENCE_CONNECTOR,
 					config: {
@@ -87,7 +88,9 @@ export default function ConfluenceConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Confluence connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/discord-connector/page.tsx

@@ -73,7 +73,8 @@ export default function DiscordConnectorPage() {
 	const onSubmit = async (values: DiscordConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.DISCORD_CONNECTOR,
 					config: {
@@ -81,7 +82,9 @@ export default function DiscordConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Discord connector created successfully!");
 			router.push(`/dashboard/${searchSpaceId}/connectors`);

surfsense_web/app/dashboard/[search_space_id]/connectors/add/github-connector/page.tsx

@@ -148,7 +148,8 @@ export default function GithubConnectorPage() {
 
 		setIsCreatingConnector(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: connectorName, // Use the stored name
 					connector_type: EnumConnectorName.GITHUB_CONNECTOR,
 					config: {
@@ -157,7 +158,9 @@ export default function GithubConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("GitHub connector created successfully!");
 			router.push(`/dashboard/${searchSpaceId}/connectors`);

surfsense_web/app/dashboard/[search_space_id]/connectors/add/google-calendar-connector/page.tsx

@@ -32,10 +32,10 @@ export default function GoogleCalendarConnectorPage() {
 	const [isConnecting, setIsConnecting] = useState(false);
 	const [doesConnectorExist, setDoesConnectorExist] = useState(false);
 
-	const { fetchConnectors } = useSearchSourceConnectors();
+	const { fetchConnectors } = useSearchSourceConnectors(true, parseInt(searchSpaceId));
 
 	useEffect(() => {
-		fetchConnectors().then((data) => {
+		fetchConnectors(parseInt(searchSpaceId)).then((data) => {
 			const connector = data.find(
 				(c: SearchSourceConnector) =>
 					c.connector_type === EnumConnectorName.GOOGLE_CALENDAR_CONNECTOR

surfsense_web/app/dashboard/[search_space_id]/connectors/add/google-gmail-connector/page.tsx

@@ -32,10 +32,10 @@ export default function GoogleGmailConnectorPage() {
 	const [isConnecting, setIsConnecting] = useState(false);
 	const [doesConnectorExist, setDoesConnectorExist] = useState(false);
 
-	const { fetchConnectors } = useSearchSourceConnectors();
+	const { fetchConnectors } = useSearchSourceConnectors(true, parseInt(searchSpaceId));
 
 	useEffect(() => {
-		fetchConnectors().then((data) => {
+		fetchConnectors(parseInt(searchSpaceId)).then((data) => {
 			const connector = data.find(
 				(c: SearchSourceConnector) => c.connector_type === EnumConnectorName.GOOGLE_GMAIL_CONNECTOR
 			);

surfsense_web/app/dashboard/[search_space_id]/connectors/add/jira-connector/page.tsx

@@ -90,7 +90,8 @@ export default function JiraConnectorPage() {
 	const onSubmit = async (values: JiraConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.JIRA_CONNECTOR,
 					config: {
@@ -100,7 +101,9 @@ export default function JiraConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Jira connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/linear-connector/page.tsx

@@ -77,7 +77,8 @@ export default function LinearConnectorPage() {
 	const onSubmit = async (values: LinearConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.LINEAR_CONNECTOR,
 					config: {
@@ -85,7 +86,9 @@ export default function LinearConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Linear connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/linkup-api/page.tsx

@@ -65,7 +65,8 @@ export default function LinkupApiPage() {
 	const onSubmit = async (values: LinkupApiFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.LINKUP_API,
 					config: {
@@ -73,7 +74,9 @@ export default function LinkupApiPage() {
 					},
 					is_indexable: false,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Linkup API connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/luma-connector/page.tsx

@@ -55,7 +55,10 @@ export default function LumaConnectorPage() {
 	const [isSubmitting, setIsSubmitting] = useState(false);
 	const [doesConnectorExist, setDoesConnectorExist] = useState(false);
 
-	const { fetchConnectors, createConnector } = useSearchSourceConnectors();
+	const { fetchConnectors, createConnector } = useSearchSourceConnectors(
+		true,
+		parseInt(searchSpaceId)
+	);
 
 	// Initialize the form
 	const form = useForm<LumaConnectorFormValues>({
@@ -67,7 +70,7 @@ export default function LumaConnectorPage() {
 	});
 
 	useEffect(() => {
-		fetchConnectors().then((data) => {
+		fetchConnectors(parseInt(searchSpaceId)).then((data) => {
 			const connector = data.find(
 				(c: SearchSourceConnector) => c.connector_type === EnumConnectorName.LUMA_CONNECTOR
 			);
@@ -81,7 +84,8 @@ export default function LumaConnectorPage() {
 	const onSubmit = async (values: LumaConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.LUMA_CONNECTOR,
 					config: {
@@ -89,7 +93,9 @@ export default function LumaConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Luma connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/notion-connector/page.tsx

@@ -72,7 +72,8 @@ export default function NotionConnectorPage() {
 	const onSubmit = async (values: NotionConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.NOTION_CONNECTOR,
 					config: {
@@ -80,7 +81,9 @@ export default function NotionConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Notion connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/serper-api/page.tsx

@@ -65,7 +65,8 @@ export default function SerperApiPage() {
 	const onSubmit = async (values: SerperApiFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.SERPER_API,
 					config: {
@@ -73,7 +74,9 @@ export default function SerperApiPage() {
 					},
 					is_indexable: false,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Serper API connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/slack-connector/page.tsx

@@ -72,7 +72,8 @@ export default function SlackConnectorPage() {
 	const onSubmit = async (values: SlackConnectorFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.SLACK_CONNECTOR,
 					config: {
@@ -80,7 +81,9 @@ export default function SlackConnectorPage() {
 					},
 					is_indexable: true,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Slack connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/connectors/add/tavily-api/page.tsx

@@ -65,7 +65,8 @@ export default function TavilyApiPage() {
 	const onSubmit = async (values: TavilyApiFormValues) => {
 		setIsSubmitting(true);
 		try {
-			await createConnector({
+			await createConnector(
+				{
 					name: values.name,
 					connector_type: EnumConnectorName.TAVILY_API,
 					config: {
@@ -73,7 +74,9 @@ export default function TavilyApiPage() {
 					},
 					is_indexable: false,
 					last_indexed_at: null,
-			});
+				},
+				parseInt(searchSpaceId)
+			);
 
 			toast.success("Tavily API connector created successfully!");
 

surfsense_web/app/dashboard/[search_space_id]/podcasts/podcasts-client.tsx

@@ -1,7 +1,6 @@
 "use client";
 
 import { format } from "date-fns";
-import { AnimatePresence, motion, type Variants } from "framer-motion";
 import {
 	Calendar,
 	MoreHorizontal,
@@ -16,6 +15,7 @@ import {
 	VolumeX,
 	X,
 } from "lucide-react";
+import { AnimatePresence, motion, type Variants } from "motion/react";
 import Image from "next/image";
 import { useEffect, useRef, useState } from "react";
 import { toast } from "sonner";

surfsense_web/components/chat/ChatInputGroup.tsx

@@ -124,19 +124,20 @@ const ConnectorSelector = React.memo(
 		onSelectionChange?: (connectorTypes: string[]) => void;
 		selectedConnectors?: string[];
 	}) => {
+		const { search_space_id } = useParams();
 		const [isOpen, setIsOpen] = useState(false);
 
 		const { connectorSourceItems, isLoading, isLoaded, fetchConnectors } =
-			useSearchSourceConnectors(true);
+			useSearchSourceConnectors(true, Number(search_space_id));
 
 		const handleOpenChange = useCallback(
 			(open: boolean) => {
 				setIsOpen(open);
 				if (open && !isLoaded) {
-					fetchConnectors();
+					fetchConnectors(Number(search_space_id));
 				}
 			},
-			[fetchConnectors, isLoaded]
+			[fetchConnectors, isLoaded, search_space_id]
 		);
 
 		const handleConnectorToggle = useCallback(

surfsense_web/hooks/use-connector-edit-page.ts

@@ -18,7 +18,11 @@ import {
 
 export function useConnectorEditPage(connectorId: number, searchSpaceId: string) {
 	const router = useRouter();
-	const { connectors, updateConnector, isLoading: connectorsLoading } = useSearchSourceConnectors();
+	const {
+		connectors,
+		updateConnector,
+		isLoading: connectorsLoading,
+	} = useSearchSourceConnectors(false, parseInt(searchSpaceId));
 
 	// State managed by the hook
 	const [connector, setConnector] = useState<SearchSourceConnector | null>(null);

surfsense_web/hooks/use-search-source-connectors.ts

@@ -7,6 +7,7 @@ export interface SearchSourceConnector {
 	is_indexable: boolean;
 	last_indexed_at: string | null;
 	config: Record<string, any>;
+	search_space_id: number;
 	user_id?: string;
 	created_at?: string;
 }
@@ -20,8 +21,10 @@ export interface ConnectorSourceItem {
 
 /**
  * Hook to fetch search source connectors from the API
+ * @param lazy - If true, connectors won't be fetched on mount
+ * @param searchSpaceId - Optional search space ID to filter connectors
  */
-export const useSearchSourceConnectors = (lazy: boolean = false) => {
+export const useSearchSourceConnectors = (lazy: boolean = false, searchSpaceId?: number) => {
 	const [connectors, setConnectors] = useState<SearchSourceConnector[]>([]);
 	const [isLoading, setIsLoading] = useState(!lazy); // Don't show loading initially for lazy mode
 	const [isLoaded, setIsLoaded] = useState(false); // Memoization flag
@@ -53,7 +56,8 @@ export const useSearchSourceConnectors = (lazy: boolean = false) => {
 		},
 	]);
 
-	const fetchConnectors = useCallback(async () => {
+	const fetchConnectors = useCallback(
+		async (spaceId?: number) => {
 			if (isLoaded && lazy) return; // Avoid redundant calls in lazy mode
 
 			try {
@@ -65,16 +69,21 @@ export const useSearchSourceConnectors = (lazy: boolean = false) => {
 					throw new Error("No authentication token found");
 				}
 
-			const response = await fetch(
+				// Build URL with optional search_space_id query parameter
-				`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-source-connectors/`,
+				const url = new URL(
-				{
+					`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-source-connectors/`
+				);
+				if (spaceId !== undefined) {
+					url.searchParams.append("search_space_id", spaceId.toString());
+				}
+
+				const response = await fetch(url.toString(), {
 					method: "GET",
 					headers: {
 						"Content-Type": "application/json",
 						Authorization: `Bearer ${token}`,
 					},
-				}
+				});
-			);
 
 				if (!response.ok) {
 					throw new Error(`Failed to fetch connectors: ${response.statusText}`);
@@ -94,19 +103,24 @@ export const useSearchSourceConnectors = (lazy: boolean = false) => {
 			} finally {
 				setIsLoading(false);
 			}
-	}, [isLoaded, lazy]);
+		},
+		[isLoaded, lazy]
+	);
 
 	useEffect(() => {
 		if (!lazy) {
-			fetchConnectors();
+			fetchConnectors(searchSpaceId);
 		}
-	}, [lazy, fetchConnectors]);
+	}, [lazy, fetchConnectors, searchSpaceId]);
 
 	// Function to refresh the connectors list
-	const refreshConnectors = useCallback(async () => {
+	const refreshConnectors = useCallback(
+		async (spaceId?: number) => {
 			setIsLoaded(false); // Reset memoization flag to allow refetch
-		await fetchConnectors();
+			await fetchConnectors(spaceId !== undefined ? spaceId : searchSpaceId);
-	}, [fetchConnectors]);
+		},
+		[fetchConnectors, searchSpaceId]
+	);
 
 	// Update connector source items when connectors change
 	const updateConnectorSourceItems = (currentConnectors: SearchSourceConnector[]) => {
@@ -151,9 +165,12 @@ export const useSearchSourceConnectors = (lazy: boolean = false) => {
 
 	/**
 	 * Create a new search source connector
+	 * @param connectorData - The connector data (excluding search_space_id)
+	 * @param spaceId - The search space ID to associate the connector with
 	 */
 	const createConnector = async (
-		connectorData: Omit<SearchSourceConnector, "id" | "user_id" | "created_at">
+		connectorData: Omit<SearchSourceConnector, "id" | "user_id" | "created_at" | "search_space_id">,
+		spaceId: number
 	) => {
 		try {
 			const token = localStorage.getItem("surfsense_bearer_token");
@@ -162,17 +179,20 @@ export const useSearchSourceConnectors = (lazy: boolean = false) => {
 				throw new Error("No authentication token found");
 			}
 
-			const response = await fetch(
+			// Add search_space_id as a query parameter
-				`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-source-connectors/`,
+			const url = new URL(
-				{
+				`${process.env.NEXT_PUBLIC_FASTAPI_BACKEND_URL}/api/v1/search-source-connectors/`
+			);
+			url.searchParams.append("search_space_id", spaceId.toString());
+
+			const response = await fetch(url.toString(), {
 				method: "POST",
 				headers: {
 					"Content-Type": "application/json",
 					Authorization: `Bearer ${token}`,
 				},
 				body: JSON.stringify(connectorData),
-				}
+			});
-			);
 
 			if (!response.ok) {
 				throw new Error(`Failed to create connector: ${response.statusText}`);
@@ -194,7 +214,9 @@ export const useSearchSourceConnectors = (lazy: boolean = false) => {
 	 */
 	const updateConnector = async (
 		connectorId: number,
-		connectorData: Partial<Omit<SearchSourceConnector, "id" | "user_id" | "created_at">>
+		connectorData: Partial<
+			Omit<SearchSourceConnector, "id" | "user_id" | "created_at" | "search_space_id">
+		>
 	) => {
 		try {
 			const token = localStorage.getItem("surfsense_bearer_token");