"""简单任务 API：将聊天任务与 WebSocket 解耦，支持后台运行与轮询。""" from __future__ import annotations from server.tasks import tasks_bp import mimetypes import json import time import threading import uuid import re from collections import deque from pathlib import Path from typing import Dict, Any, Optional, List from flask import Blueprint, request, jsonify from flask import current_app, session from server.auth_helpers import api_login_required, get_current_username from server.context import get_user_resources, ensure_conversation_loaded from server.chat_flow import run_chat_task_sync from server.state import stop_flags from server.utils_common import debug_log, log_conn_diag from utils.host_workspace_debug import write_host_workspace_debug from config import DATA_DIR, WORKSPACE_SKILLS_DIRNAME from modules.goal_state_manager import GoalStateManager, REASON_USER_CANCEL from server.tasks.models import TaskRecord SKILL_FRONTMATTER_RE = re.compile(r"^---\s*\n(?P.*?)\n---\s*\n?", re.S) SKILL_FIELD_RE = re.compile(r"^(?Pname|description)\s*:\s*(?P.*)$") def _conversation_title_for_task(rec: TaskRecord) -> Optional[str]: conv_id = (getattr(rec, "conversation_id", None) or "").strip() if not conv_id: return None if not conv_id.startswith("conv_"): conv_id = f"conv_{conv_id}" try: if rec.username == "host": path = Path(DATA_DIR).expanduser().resolve() / "conversations" / rec.workspace_id / f"{conv_id}.json" else: import server.state as state workspace = state.user_manager.ensure_user_workspace(rec.username, rec.workspace_id or "default") path = Path(workspace.data_dir).expanduser().resolve() / "conversations" / f"{conv_id}.json" if not path.exists(): return None data = json.loads(path.read_text(encoding="utf-8")) title = str(data.get("title") or "").strip() return title or None except Exception: return None def _workspace_label_for_task(rec: TaskRecord) -> Optional[str]: try: if rec.username == "host": from modules.host_workspace_manager import load_host_workspace_catalog catalog = load_host_workspace_catalog() for item in catalog.get("workspaces") or []: if item.get("workspace_id") == rec.workspace_id: return str(item.get("label") or rec.workspace_id) else: import server.state as state item = state.user_manager.list_user_workspaces(rec.username).get(rec.workspace_id) if item: return str(item.get("label") or rec.workspace_id) except Exception: pass return rec.workspace_id def _task_public_payload(rec: TaskRecord, *, include_title: bool = True) -> Dict[str, Any]: payload = { "task_id": rec.task_id, "username": rec.username, "workspace_id": rec.workspace_id, "workspace_label": _workspace_label_for_task(rec), "status": rec.status, "created_at": rec.created_at, "updated_at": rec.updated_at, "message": rec.message, "conversation_id": rec.conversation_id, "error": rec.error, "message_source": (rec.session_data or {}).get("message_source"), "goal_mode": bool((rec.session_data or {}).get("goal_mode")), "goal_progress": (rec.session_data or {}).get("goal_progress"), "task_type": getattr(rec, "task_type", "chat"), } if include_title: payload["conversation_title"] = _conversation_title_for_task(rec) return payload