32 lines
1.6 KiB
Python
32 lines
1.6 KiB
Python
from fastapi import APIRouter, Depends
|
||
from app.core.auth import get_current_user
|
||
from app.core.response import create_api_response
|
||
from app.services.async_transcription_service import AsyncTranscriptionService
|
||
from app.services.async_meeting_service import async_meeting_service
|
||
|
||
router = APIRouter()
|
||
|
||
transcription_service = AsyncTranscriptionService()
|
||
|
||
@router.get("/tasks/transcription/{task_id}/status")
|
||
def get_transcription_task_status(task_id: str, current_user: dict = Depends(get_current_user)):
|
||
"""获取转录任务状态"""
|
||
try:
|
||
status_info = transcription_service.get_task_status(task_id)
|
||
if not status_info or status_info.get('status') == 'not_found':
|
||
return create_api_response(code="404", message="Transcription task not found")
|
||
return create_api_response(code="200", message="Task status retrieved", data=status_info)
|
||
except Exception as e:
|
||
return create_api_response(code="500", message=f"Failed to get task status: {str(e)}")
|
||
|
||
@router.get("/tasks/summaries/{task_id}/status")
|
||
def get_llm_task_status(task_id: str, current_user: dict = Depends(get_current_user)):
|
||
"""获取LLM总结任务状态（包括进度）"""
|
||
try:
|
||
status = async_meeting_service.get_task_status(task_id)
|
||
if status.get('status') == 'not_found':
|
||
return create_api_response(code="404", message="Task not found")
|
||
return create_api_response(code="200", message="Task status retrieved", data=status)
|
||
except Exception as e:
|
||
return create_api_response(code="500", message=f"Failed to get task status: {str(e)}")
|