ai写作

frontend-web/src/open-canvas/api/ai-writing.ts

@@ -0,0 +1,195 @@
+import { getBackendBaseURL } from '@/core/config';
+import { apiFetch } from '@/core/api';
+
+import type {
+  AIWritingRequest,
+  ProgressEvent,
+  ProgressEventType,
+  UserInterventionPayload,
+} from '../types/ai-writing';
+
+const BASE = () => `${getBackendBaseURL()}/api/ai-writing`;
+
+/** 启动写作流程，返回 session_id */
+export async function startWritingSession(req: AIWritingRequest): Promise<string> {
+  const resp = await apiFetch(`${BASE()}/start`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({
+      user_intent: req.userIntent,
+      article_type: req.articleType,
+      target_audience: req.targetAudience,
+      word_count_target: req.wordCountTarget,
+      model_name: req.modelName ?? '',
+    }),
+  });
+  if (!resp.ok) throw new Error(`启动失败: ${resp.status}`);
+  const data = await resp.json();
+  return data.session_id;
+}
+
+const SSE_EVENT_TYPES: ProgressEventType[] = [
+  'connected', 'step_start', 'step_done', 'materials_ready', 'outline_ready',
+  'draft_chunk', 'draft_ready', 'review_ready', 'await_user', 'resumed', 'done', 'error',
+];
+
+// ── 后端返回 snake_case，前端类型是 camelCase，统一在此转换 ──────────────────
+
+function normalizeMaterial(m: Record<string, unknown>): Record<string, unknown> {
+  return {
+    ...m,
+    relevanceScore: m.relevance_score ?? m.relevanceScore ?? 0,
+    publishedAt:    m.published_at   ?? m.publishedAt   ?? '',
+    sourceType:     m.source_type    ?? m.sourceType     ?? 'internet',
+    recUuid:        m.recUuid        ?? m.rec_uuid       ?? '',
+  };
+}
+
+function normalizeOutlineSection(s: Record<string, unknown>): Record<string, unknown> {
+  return {
+    ...s,
+    sectionTitle: s.section_title ?? s.sectionTitle ?? '',
+    keyPoints:    s.key_points    ?? s.keyPoints    ?? [],
+    materialIds:  s.material_ids  ?? s.materialIds  ?? [],
+  };
+}
+
+function normalizeOutline(o: unknown): unknown {
+  if (!o || typeof o !== 'object') return o;
+  const obj = o as Record<string, unknown>;
+  const sections = Array.isArray(obj.sections)
+    ? obj.sections.map((s: unknown) => normalizeOutlineSection(s as Record<string, unknown>))
+    : [];
+  return { ...obj, sections };
+}
+
+function normalizeReviewIssue(i: Record<string, unknown>): Record<string, unknown> {
+  return { ...i };
+}
+
+function normalizeReviewResult(r: unknown): unknown {
+  if (!r || typeof r !== 'object') return r;
+  const obj = r as Record<string, unknown>;
+  return {
+    ...obj,
+    factScore:      obj.fact_score      ?? obj.factScore      ?? 0,
+    logicScore:     obj.logic_score     ?? obj.logicScore     ?? 0,
+    languageScore:  obj.language_score  ?? obj.languageScore  ?? 0,
+    overallScore:   obj.overall_score   ?? obj.overallScore   ?? 0,
+    passThreshold:  obj.pass_threshold  ?? obj.passThreshold  ?? 75,
+    revisionNotes: Array.isArray(obj.revision_notes)
+      ? obj.revision_notes.map((x: unknown) => normalizeReviewIssue(x as Record<string, unknown>))
+      : Array.isArray(obj.revisionNotes)
+        ? obj.revisionNotes
+        : [],
+  };
+}
+
+function normalizeEvent(parsed: Record<string, unknown>): Partial<ProgressEvent> {
+  const out: Record<string, unknown> = { ...parsed };
+
+  // 通用字段
+  if ('agent_name'  in out) { out.agentName  = out.agent_name;  }
+  if ('pause_point' in out) { out.pausePoint = out.pause_point; }
+
+  // materials 数组（materials_ready 和 await_user 里都可能有）
+  if (Array.isArray(out.materials)) {
+    out.materials = out.materials.map((m: unknown) =>
+      normalizeMaterial(m as Record<string, unknown>)
+    );
+  }
+
+  // outline
+  if (out.outline) out.outline = normalizeOutline(out.outline);
+
+  // draft_ready: 后端发 "title"/"markdown"，前端期望 draftTitle/draftMarkdown
+  if ('markdown' in out && !out.draftMarkdown) out.draftMarkdown = out.markdown;
+  if ('title'    in out && !out.draftTitle)    out.draftTitle    = out.title;
+  if ('revision_count' in out) out.revisionCount = out.revision_count;
+
+  // review_ready
+  if (out.review_result)  out.reviewResult  = normalizeReviewResult(out.review_result);
+  if (out.reviewResult && !out.review_result) out.reviewResult = normalizeReviewResult(out.reviewResult);
+
+  return out as Partial<ProgressEvent>;
+}
+
+/** 监听 SSE 进度流，使用 fetch + ReadableStream 以支持认证头 */
+export function subscribeToProgress(
+  sessionId: string,
+  onEvent: (event: ProgressEvent) => void,
+  signal?: AbortSignal,
+): () => void {
+  let stopped = false;
+
+  apiFetch(`${BASE()}/${sessionId}/stream`, { signal })
+    .then(async (response) => {
+      if (!response.ok || !response.body) return;
+
+      const reader = response.body.getReader();
+      const decoder = new TextDecoder();
+      let buffer = '';
+      let currentType = '';
+      let currentData = '';
+
+      while (!stopped) {
+        const { done, value } = await reader.read();
+        if (done) break;
+
+        buffer += decoder.decode(value, { stream: true });
+        const lines = buffer.split('\n');
+        buffer = lines.pop() ?? '';
+
+        for (const line of lines) {
+          if (line.startsWith('event: ')) {
+            currentType = line.slice(7).trim();
+          } else if (line.startsWith('data: ')) {
+            currentData = line.slice(6).trim();
+          } else if (line === '' && currentType && currentData) {
+            if (SSE_EVENT_TYPES.includes(currentType as ProgressEventType)) {
+              try {
+                const parsed = JSON.parse(currentData);
+                const normalized = normalizeEvent(parsed);
+                onEvent({
+                  type: currentType as ProgressEventType,
+                  sessionId,
+                  message: '',
+                  ...normalized,
+                } as ProgressEvent);
+              } catch {
+                // malformed JSON — skip
+              }
+            }
+            currentType = '';
+            currentData = '';
+          }
+        }
+      }
+
+      reader.releaseLock();
+    })
+    .catch(() => {});
+
+  return () => { stopped = true; };
+}
+
+/** 提交用户干预决策，恢复暂停的流程 */
+export async function resumeWriting(
+  sessionId: string,
+  payload: UserInterventionPayload,
+): Promise<void> {
+  const resp = await apiFetch(`${BASE()}/${sessionId}/resume`, {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({
+      action: payload.action,
+      approved_material_ids: payload.approvedMaterialIds ?? [],
+      extra_keywords: payload.extraKeywords ?? [],
+      edited_outline: payload.editedOutline ?? null,
+      outline_feedback: payload.outlineFeedback ?? '',
+      user_revision_notes: payload.userRevisionNotes ?? '',
+      override_verdict: payload.overrideVerdict ?? false,
+    }),
+  });
+  if (!resp.ok) throw new Error(`提交干预失败: ${resp.status}`);
+}