init with react+axum+spacetimedb

src/services/puzzle-agent/index.ts

@@ -0,0 +1,8 @@
+export {
+  createPuzzleAgentSession,
+  executePuzzleAgentAction,
+  getPuzzleAgentSession,
+  puzzleAgentClient,
+  sendPuzzleAgentMessage,
+  streamPuzzleAgentMessage,
+} from './puzzleAgentClient';

src/services/puzzle-agent/puzzleAgentClient.ts

@@ -0,0 +1,92 @@
+import type {
+  PuzzleAgentActionRequest,
+  PuzzleAgentActionResponse,
+} from '../../../packages/shared/src/contracts/puzzleAgentActions';
+import type {
+  CreatePuzzleAgentSessionRequest,
+  CreatePuzzleAgentSessionResponse,
+  PuzzleAgentSessionSnapshot,
+  SendPuzzleAgentMessageRequest,
+} from '../../../packages/shared/src/contracts/puzzleAgentSession';
+import type { TextStreamOptions } from '../aiTypes';
+import { createCreationAgentClient } from '../creation-agent';
+
+const PUZZLE_AGENT_API_BASE = '/api/runtime/puzzle/agent/sessions';
+const puzzleAgentHttpClient = createCreationAgentClient<
+  CreatePuzzleAgentSessionRequest,
+  CreatePuzzleAgentSessionResponse,
+  CreatePuzzleAgentSessionResponse,
+  PuzzleAgentSessionSnapshot,
+  SendPuzzleAgentMessageRequest,
+  { session: PuzzleAgentSessionSnapshot },
+  PuzzleAgentActionRequest,
+  PuzzleAgentActionResponse
+>({
+  apiBase: PUZZLE_AGENT_API_BASE,
+  messages: {
+    createSession: '创建拼图共创会话失败',
+    getSession: '读取拼图共创会话失败',
+    sendMessage: '发送拼图共创消息失败',
+    streamIncomplete: '拼图共创消息流式结果不完整',
+    executeAction: '执行拼图共创操作失败',
+  },
+});
+
+/**
+ * 创建拼图 Agent 共创会话。
+ * 首版继续走 Axum facade，前端不直连 SpacetimeDB。
+ */
+export async function createPuzzleAgentSession(
+  payload: CreatePuzzleAgentSessionRequest = {},
+) {
+  return puzzleAgentHttpClient.createSession(payload);
+}
+
+/**
+ * 读取拼图 Agent 会话快照。
+ */
+export async function getPuzzleAgentSession(sessionId: string) {
+  return puzzleAgentHttpClient.getSession(sessionId);
+}
+
+/**
+ * 非流式发送拼图 Agent 消息。
+ * 当前 UI 主链使用 SSE，但保留普通接口便于后续降级。
+ */
+export async function sendPuzzleAgentMessage(
+  sessionId: string,
+  payload: SendPuzzleAgentMessageRequest,
+) {
+  return puzzleAgentHttpClient.sendMessage(sessionId, payload);
+}
+
+/**
+ * 流式发送拼图 Agent 消息。
+ * 后端当前会先回传一段 assistant 文本，再附上最新 session 快照。
+ */
+export async function streamPuzzleAgentMessage(
+  sessionId: string,
+  payload: SendPuzzleAgentMessageRequest,
+  options: TextStreamOptions = {},
+) {
+  return puzzleAgentHttpClient.streamMessage(sessionId, payload, options);
+}
+
+/**
+ * 执行拼图结果页相关操作。
+ * 后端会返回 operation 记录，前端再按需刷新 session 或 works/gallery。
+ */
+export async function executePuzzleAgentAction(
+  sessionId: string,
+  payload: PuzzleAgentActionRequest,
+) {
+  return puzzleAgentHttpClient.executeAction(sessionId, payload);
+}
+
+export const puzzleAgentClient = {
+  createSession: createPuzzleAgentSession,
+  getSession: getPuzzleAgentSession,
+  sendMessage: sendPuzzleAgentMessage,
+  streamMessage: streamPuzzleAgentMessage,
+  executeAction: executePuzzleAgentAction,
+};