mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-06 22:12:12 +02:00
7fb0707933
Scoped codemod over surfsense_backend/app (excluding routes/, Wave E): renames search_space_id -> workspace_id, search_space -> workspace, SearchSpace -> Workspace across services, utils, tasks, agents, gateway, event_bus, notifications, podcasts, automations, observability params, and prompt .md files. Also flips the camelCase payload key searchSpaceId -> workspaceId (no backend reader; hard cutover). Preserved carve-outs (verbatim): Celery task names "delete_search_space_background" and "ai_sort_search_space" (wire names), and the OTel/metric key "search_space.id" (dashboards depend on it). Enum values 'SEARCH_SPACE' and SearchSourceConnector untouched.
66 lines
1.9 KiB
Python
66 lines
1.9 KiB
Python
"""External chat binding helpers."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
|
|
from sqlalchemy import select
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db import (
|
|
ChatVisibility,
|
|
ExternalChatBinding,
|
|
ExternalChatBindingState,
|
|
NewChatThread,
|
|
)
|
|
|
|
|
|
async def get_or_create_thread_for_binding(
|
|
session: AsyncSession,
|
|
binding: ExternalChatBinding,
|
|
) -> NewChatThread:
|
|
if binding.new_chat_thread_id is not None:
|
|
result = await session.execute(
|
|
select(NewChatThread).where(NewChatThread.id == binding.new_chat_thread_id)
|
|
)
|
|
thread = result.scalars().first()
|
|
if thread is not None and not thread.archived:
|
|
return thread
|
|
|
|
source = str((binding.external_metadata or {}).get("platform") or "").strip()
|
|
if not source:
|
|
kind = str((binding.external_metadata or {}).get("kind") or "")
|
|
source = "slack" if kind.startswith("slack_") else "telegram"
|
|
|
|
thread = NewChatThread(
|
|
title=f"{source.title()} chat",
|
|
workspace_id=binding.workspace_id,
|
|
created_by_id=binding.user_id,
|
|
visibility=ChatVisibility.PRIVATE,
|
|
source=source,
|
|
external_chat_binding_id=binding.id,
|
|
)
|
|
session.add(thread)
|
|
await session.flush()
|
|
binding.new_chat_thread_id = thread.id
|
|
return thread
|
|
|
|
|
|
def suspend_binding(binding: ExternalChatBinding, reason: str) -> None:
|
|
now = datetime.now(UTC)
|
|
binding.state = ExternalChatBindingState.SUSPENDED
|
|
binding.suspended_at = now
|
|
binding.suspended_reason = reason
|
|
|
|
|
|
def revoke_binding(binding: ExternalChatBinding) -> None:
|
|
now = datetime.now(UTC)
|
|
binding.state = ExternalChatBindingState.REVOKED
|
|
binding.revoked_at = now
|
|
binding.new_chat_thread_id = None
|
|
|
|
|
|
def resume_binding(binding: ExternalChatBinding) -> None:
|
|
binding.state = ExternalChatBindingState.BOUND
|
|
binding.suspended_at = None
|
|
binding.suspended_reason = None
|