feat(workspaces): return llm_setup on workspace create

This commit is contained in:
Anish Sarkar 2026-07-12 20:29:31 +05:30
parent ff676519ea
commit 561bc76a9a
2 changed files with 11 additions and 1 deletions

View file

@ -14,6 +14,7 @@ from app.db import (
get_async_session,
get_default_roles_config,
)
from app.routes.model_connections_routes import compute_llm_setup_status
from app.schemas import (
WorkspaceApiAccessUpdate,
WorkspaceCreate,
@ -93,7 +94,12 @@ async def create_workspace(
await session.commit()
await session.refresh(db_workspace)
return db_workspace
response = WorkspaceRead.model_validate(db_workspace)
response.llm_setup = await compute_llm_setup_status(
session, auth, db_workspace.id
)
return response
except HTTPException:
raise
except Exception as e:

View file

@ -4,6 +4,7 @@ from datetime import datetime
from pydantic import BaseModel, ConfigDict
from .base import IDModel, TimestampModel
from .model_connections import LlmSetupStatusRead
class WorkspaceBase(BaseModel):
@ -35,6 +36,9 @@ class WorkspaceRead(WorkspaceBase, IDModel, TimestampModel):
api_access_enabled: bool = False
qna_custom_instructions: str | None = None
shared_memory_md: str | None = None
# Populated only by create_workspace so the client can route straight to
# onboarding vs. new-chat on the first hop. Null everywhere else.
llm_setup: LlmSetupStatusRead | None = None
model_config = ConfigDict(from_attributes=True)