85 lines
2.9 KiB
Python
85 lines
2.9 KiB
Python
"""按 taskId 存取的「大屏绘制智能体」独立页面会话存储 API。
|
||
|
||
独立大屏页（前端 ``DashboardAgentPage``）按外部 ``task_id`` 存取：聊天记录
|
||
（``transcript``）+ 最新结构化数据（``report_json``）。**按 task 共享、不按 user
|
||
分权**（open read + 写放开），镜像 ``roundtable_drafts`` 的 by-task 语义。
|
||
|
||
Routes (prefix ``/api/dashboard-sessions``):
|
||
GET /by-task/{task_id} 取该 task 的会话（无则 404）
|
||
PUT /by-task/{task_id} upsert（body: {transcript?, report_json?}）
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from typing import Any
|
||
|
||
from fastapi import APIRouter, HTTPException, Request
|
||
from pydantic import BaseModel
|
||
|
||
from deerflow.runtime.user_context import get_effective_user_id
|
||
|
||
router = APIRouter(prefix="/api/dashboard-sessions", tags=["dashboard-sessions"])
|
||
|
||
|
||
class DashboardSessionResponse(BaseModel):
|
||
task_id: str
|
||
transcript: Any | None = None
|
||
report_json: str | None = None
|
||
created_at: str | None = None
|
||
updated_at: str | None = None
|
||
|
||
|
||
class DashboardSessionUpsertRequest(BaseModel):
|
||
# 只传需要更新的字段：聊天记录 / 结构化数据，二者皆可单独更新。
|
||
transcript: Any | None = None
|
||
report_json: str | None = None
|
||
|
||
|
||
def _current_user_id(request: Request) -> str:
|
||
user = getattr(request.state, "user", None)
|
||
if user is not None:
|
||
return str(user.id)
|
||
return get_effective_user_id()
|
||
|
||
|
||
def _get_store(request: Request):
|
||
store = getattr(request.app.state, "dashboard_session_store", None)
|
||
if store is None:
|
||
raise HTTPException(status_code=503, detail="Dashboard session store not available")
|
||
return store
|
||
|
||
|
||
def _project(row: dict[str, Any]) -> DashboardSessionResponse:
|
||
return DashboardSessionResponse(
|
||
task_id=row.get("task_id", ""),
|
||
transcript=row.get("transcript"),
|
||
report_json=row.get("report_json"),
|
||
created_at=row.get("created_at"),
|
||
updated_at=row.get("updated_at"),
|
||
)
|
||
|
||
|
||
@router.get("/by-task/{task_id}", response_model=DashboardSessionResponse)
|
||
async def get_dashboard_session(request: Request, task_id: str) -> DashboardSessionResponse:
|
||
"""取该 task 的大屏会话（**不按 user 分权**）；无则 404。"""
|
||
store = _get_store(request)
|
||
row = await store.get_by_task(task_id)
|
||
if row is None:
|
||
raise HTTPException(status_code=404, detail="No dashboard session for this task")
|
||
return _project(row)
|
||
|
||
|
||
@router.put("/by-task/{task_id}", response_model=DashboardSessionResponse)
|
||
async def upsert_dashboard_session(
|
||
request: Request, task_id: str, body: DashboardSessionUpsertRequest
|
||
) -> DashboardSessionResponse:
|
||
"""按 task upsert 大屏会话（**不按 user 分权**），返回最新快照。"""
|
||
store = _get_store(request)
|
||
row = await store.upsert_by_task(
|
||
task_id,
|
||
transcript=body.transcript,
|
||
report_json=body.report_json,
|
||
user_id=_current_user_id(request),
|
||
)
|
||
return _project(row)
|