apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-20 23:21:06 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(model-connections): stamp own-model readiness and derive setup stage

Browse source
This commit is contained in:
Anish Sarkar 2026-07-14 00:14:51 +05:30
parent 88d3d336bf
commit 2f048d87bc
1 changed files with 25 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

30
surfsense_backend/app/routes/model_connections_routes.py
View file

@ -1,4 +1,5 @@
import logging import logging
from datetime import UTC, datetime
from fastapi import APIRouter, Depends, HTTPException from fastapi import APIRouter, Depends, HTTPException
from sqlalchemy import select, update from sqlalchemy import select, update
@ -895,8 +896,12 @@ async def compute_llm_setup_status(
global_usable = _global_catalog_has_usable_chat() global_usable = _global_catalog_has_usable_chat()
if config.GLOBAL_LLM_CONFIG_FILE_EXISTS and global_usable: if config.GLOBAL_LLM_CONFIG_FILE_EXISTS and global_usable:
# Global readiness is never stamped: it is not this workspace's own setup.
return LlmSetupStatusRead( return LlmSetupStatusRead(
status="ready", source="global_config", can_configure=can_configure status="ready",
source="global_config",
can_configure=can_configure,
stage="ready",
) )
# Heal dangling role pins first: a chat_model_id pointing at a deleted or # Heal dangling role pins first: a chat_model_id pointing at a deleted or
@ -906,25 +911,40 @@ async def compute_llm_setup_status(
await session.commit() await session.commit()
await session.refresh(workspace) await session.refresh(workspace)
async def _stamp_own_setup() -> None:
# Record first own-model readiness once; the guard makes concurrent
# stamps idempotent. (Writing on this GET is precedented above.)
if workspace.llm_setup_completed_at is None:
workspace.llm_setup_completed_at = datetime.now(UTC)
await session.commit()
chat_model_id = workspace.chat_model_id or 0 chat_model_id = workspace.chat_model_id or 0
if chat_model_id != 0: if chat_model_id != 0:
# Survived _clear_invalid_roles => valid, enabled, chat-capable. # Survived _clear_invalid_roles => valid, enabled, chat-capable.
source = "global_config" if chat_model_id < 0 else "models" source = "global_config" if chat_model_id < 0 else "models"
if source == "models":
await _stamp_own_setup()
return LlmSetupStatusRead( return LlmSetupStatusRead(
status="ready", source=source, can_configure=can_configure status="ready", source=source, can_configure=can_configure, stage="ready"
) )
if global_usable: if global_usable:
return LlmSetupStatusRead( return LlmSetupStatusRead(
status="ready", source="global_config", can_configure=can_configure status="ready",
source="global_config",
can_configure=can_configure,
stage="ready",
) )
if await _workspace_has_enabled_chat_model(session, workspace_id): if await _workspace_has_enabled_chat_model(session, workspace_id):
await _stamp_own_setup()
return LlmSetupStatusRead( return LlmSetupStatusRead(
status="ready", source="models", can_configure=can_configure status="ready", source="models", can_configure=can_configure, stage="ready"
) )
# A set timestamp => self-configured before (recovery); NULL => never (first-run).
stage = "recovery" if workspace.llm_setup_completed_at else "initial_setup"
return LlmSetupStatusRead( return LlmSetupStatusRead(
status="needs_setup", source="none", can_configure=can_configure status="needs_setup", source="none", can_configure=can_configure, stage=stage
) )