sea-dev 1.0.0

package/apps/api/src/routes/v2/chat/handlers.ts

@@ -0,0 +1,693 @@
+import { getApiConfig } from "@/lib/config";
+import {
+  chatStatic,
+  chatStream,
+  chatStreamAgent,
+  generateConversationSummary,
+  getChatModel,
+} from "@sea/core";
+import {
+  eraseConversation,
+  getAgentConversation,
+  getConversation,
+  getFormConversation,
+  getSubmissionConversation,
+  summarizeEraseConversation as persistSummary,
+} from "@sea/dal/conversation";
+import type { MemberHydrated } from "@sea/dal/member";
+import { nanoid } from "@sea/util/id";
+import type { UIMessage } from "ai";
+import type { FastifyReply } from "fastify";
+import type { ChatRequest } from "./schemas";
+
+// Helper function to postprocess parts from static chat:
+// 1. Merge tool calls with their results
+// 2. Group parts by steps
+function processPartsForResponse(parts: UIMessage["parts"]) {
+  const result: Array<{ type: string; [key: string]: unknown }> = [];
+  const toolCallMap = new Map<string, { type: string; [key: string]: unknown }>();
+
+  // Split parts into steps
+  const steps: Array<Array<{ type: string; [key: string]: unknown }>> = [[]];
+  for (const part of parts) {
+    if (part.type === "step-start") {
+      steps.push([]);
+    } else {
+      const currentStep = steps[steps.length - 1];
+      if (currentStep) {
+        currentStep.push(part);
+      }
+    }
+  }
+
+  // Process each step
+  for (const stepParts of steps) {
+    const stepResult: Array<{ type: string; [key: string]: unknown }> = [];
+
+    for (const part of stepParts) {
+      // Handle tool parts - merge call and result
+      if (part.type.startsWith("tool-")) {
+        const toolCallId = part.toolCallId as string;
+        const state = part.state as string;
+
+        if (state === "input-available") {
+          // Store tool call
+          toolCallMap.set(toolCallId, { ...part });
+        } else if (state === "output-available" || state === "output-error") {
+          // Merge with stored tool call
+          const toolCall = toolCallMap.get(toolCallId);
+          if (toolCall) {
+            stepResult.push({
+              type: part.type,
+              input: toolCall.input,
+              output: part.output,
+              errorText: part.errorText,
+              state,
+            });
+            toolCallMap.delete(toolCallId);
+          } else {
+            // No matching call, include as-is
+            stepResult.push(part);
+          }
+        }
+      } else {
+        // Non-tool parts - include as-is
+        stepResult.push(part);
+      }
+    }
+
+    // Add any remaining tool calls that didn't have results
+    for (const [, toolCall] of toolCallMap) {
+      stepResult.push(toolCall);
+    }
+    toolCallMap.clear();
+
+    // Merge all parts in the step into one part
+    if (stepResult.length > 0) {
+      // If step has only one part, use it as-is
+      if (stepResult.length === 1) {
+        result.push(stepResult[0]!);
+      } else {
+        // Extract text and tool parts separately
+        const textParts = stepResult.filter((p) => p.type === "text");
+        const toolParts = stepResult.filter((p) => p.type.startsWith("tool-"));
+
+        // Create merged text part if there are any text parts
+        if (textParts.length > 0) {
+          result.push({
+            type: "text",
+            text: textParts.map((p) => p.text).join(""),
+          });
+        }
+
+        // Add tool parts
+        result.push(...toolParts);
+      }
+    }
+  }
+
+  return result;
+}
+
+export async function chatFormStreamHandler(
+  params: { formId: string },
+  body: ChatRequest,
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const { message } = body;
+
+    const formConversation = await getFormConversation({ member, formId: params.formId });
+    if (!formConversation.ok) {
+      return reply.status(404).send({
+        description: `Form with ID ${params.formId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = formConversation.data;
+    const uiMessage = {
+      id: nanoid(),
+      role: "user" as const,
+      parts: [{ type: "text" as const, text: message }],
+    };
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-5.2",
+    });
+
+    const textStream = await chatStream(
+      model,
+      member,
+      conversationId,
+      uiMessage as UIMessage,
+      "form",
+      false,
+    );
+
+    return reply.send(textStream);
+  } catch (error) {
+    console.error("Error in chatFormStream:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function chatFormJsonHandler(
+  params: { formId: string },
+  body: ChatRequest,
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const { message } = body;
+
+    const formConversation = await getFormConversation({ member, formId: params.formId });
+    if (!formConversation.ok) {
+      return reply.status(404).send({
+        description: `Form with ID ${params.formId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = formConversation.data;
+    const uiMessage = {
+      id: nanoid(),
+      role: "user" as const,
+      parts: [{ type: "text" as const, text: message }],
+    };
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-5.2",
+    });
+
+    const result = await chatStatic(
+      model,
+      member,
+      conversationId,
+      uiMessage as UIMessage,
+      "form",
+      false,
+    );
+
+    // Get the updated conversation with all messages
+    const conversationResult = await getConversation({ member, conversationId });
+    if (!conversationResult.ok) {
+      return reply.status(500).send({ description: "Failed to retrieve conversation" });
+    }
+
+    const { messages } = conversationResult.data;
+    const processedMessages = (messages as UIMessage[]).map((m) => ({
+      id: m.id,
+      role: m.role,
+      parts: processPartsForResponse(m.parts),
+    }));
+
+    // Extract text from assistant response
+    const textParts =
+      (result.parts?.filter((p) => p.type === "text") as Array<{ text: string }> | undefined) ?? [];
+    const text = textParts.length > 0 ? textParts.map((p) => p.text).join("") : undefined;
+
+    return reply.send({
+      text,
+      messages: processedMessages,
+    });
+  } catch (error) {
+    console.error("Error in chatFormJson:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function getFormChatHistoryHandler(
+  params: { formId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const formConversation = await getFormConversation({ member, formId: params.formId });
+    if (!formConversation.ok) {
+      return reply.status(404).send({
+        description: `Form with ID ${params.formId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = formConversation.data;
+    const conversationResult = await getConversation({ member, conversationId });
+    if (!conversationResult.ok) {
+      return reply.status(404).send({ description: "Conversation not found" });
+    }
+
+    const { messages } = conversationResult.data;
+    const processedMessages = (messages as UIMessage[]).map((m) => ({
+      id: m.id,
+      role: m.role,
+      parts: processPartsForResponse(m.parts),
+    }));
+
+    return reply.send(processedMessages);
+  } catch (error) {
+    console.error("Error in getFormChatHistory:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function deleteFormChatHandler(
+  params: { formId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const formConversation = await getFormConversation({ member, formId: params.formId });
+    if (!formConversation.ok) {
+      return reply.status(404).send({
+        description: `Form with ID ${params.formId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = formConversation.data;
+    await eraseConversation({ member, conversationId });
+
+    return reply.send({ success: true });
+  } catch (error) {
+    console.error("Error in deleteFormChat:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function summarizeFormChatHandler(
+  params: { formId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const formConversation = await getFormConversation({ member, formId: params.formId });
+    if (!formConversation.ok) {
+      return reply.status(404).send({
+        description: `Form with ID ${params.formId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = formConversation.data;
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-4.1",
+    });
+
+    const summary = await generateConversationSummary(model, member, conversationId);
+    await persistSummary({ member, conversationId, summary });
+
+    return reply.send({ success: true });
+  } catch (error) {
+    console.error("Error in summarizeFormChat:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function chatSubmissionStreamHandler(
+  params: { submissionId: string },
+  body: ChatRequest,
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const { message } = body;
+
+    const submissionConversation = await getSubmissionConversation({
+      member,
+      submissionId: params.submissionId,
+    });
+    if (!submissionConversation.ok) {
+      return reply.status(404).send({
+        description: `Submission with ID ${params.submissionId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = submissionConversation.data;
+    const uiMessage = {
+      id: nanoid(),
+      role: "user" as const,
+      parts: [{ type: "text" as const, text: message }],
+    };
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-5.2",
+    });
+
+    const textStream = await chatStream(
+      model,
+      member,
+      conversationId,
+      uiMessage as UIMessage,
+      "submission",
+      false,
+    );
+
+    return reply.send(textStream);
+  } catch (error) {
+    console.error("Error in chatSubmissionStream:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function chatSubmissionJsonHandler(
+  params: { submissionId: string },
+  body: ChatRequest,
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const { message } = body;
+
+    const submissionConversation = await getSubmissionConversation({
+      member,
+      submissionId: params.submissionId,
+    });
+    if (!submissionConversation.ok) {
+      return reply.status(404).send({
+        description: `Submission with ID ${params.submissionId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = submissionConversation.data;
+    const uiMessage = {
+      id: nanoid(),
+      role: "user" as const,
+      parts: [{ type: "text" as const, text: message }],
+    };
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-5.2",
+    });
+
+    const result = await chatStatic(
+      model,
+      member,
+      conversationId,
+      uiMessage as UIMessage,
+      "submission",
+      false,
+    );
+
+    // Get the updated conversation with all messages
+    const conversationResult = await getConversation({ member, conversationId });
+    if (!conversationResult.ok) {
+      return reply.status(500).send({ description: "Failed to retrieve conversation" });
+    }
+
+    const { messages } = conversationResult.data;
+    const processedMessages = (messages as UIMessage[]).map((m) => ({
+      id: m.id,
+      role: m.role,
+      parts: processPartsForResponse(m.parts),
+    }));
+
+    // Extract text from assistant response
+    const textParts =
+      (result.parts?.filter((p) => p.type === "text") as Array<{ text: string }> | undefined) ?? [];
+    const text = textParts.length > 0 ? textParts.map((p) => p.text).join("") : undefined;
+
+    return reply.send({
+      text,
+      messages: processedMessages,
+    });
+  } catch (error) {
+    console.error("Error in chatSubmissionJson:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function getSubmissionChatHistoryHandler(
+  params: { submissionId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const submissionConversation = await getSubmissionConversation({
+      member,
+      submissionId: params.submissionId,
+    });
+    if (!submissionConversation.ok) {
+      return reply.status(404).send({
+        description: `Submission with ID ${params.submissionId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = submissionConversation.data;
+    const conversationResult = await getConversation({ member, conversationId });
+    if (!conversationResult.ok) {
+      return reply.status(404).send({ description: "Conversation not found" });
+    }
+
+    const { messages } = conversationResult.data;
+    const processedMessages = (messages as UIMessage[]).map((m) => ({
+      id: m.id,
+      role: m.role,
+      parts: processPartsForResponse(m.parts),
+    }));
+
+    return reply.send(processedMessages);
+  } catch (error) {
+    console.error("Error in getSubmissionChatHistory:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function deleteSubmissionChatHandler(
+  params: { submissionId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const submissionConversation = await getSubmissionConversation({
+      member,
+      submissionId: params.submissionId,
+    });
+    if (!submissionConversation.ok) {
+      return reply.status(404).send({
+        description: `Submission with ID ${params.submissionId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = submissionConversation.data;
+    await eraseConversation({ member, conversationId });
+
+    return reply.send({ success: true });
+  } catch (error) {
+    console.error("Error in deleteSubmissionChat:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function summarizeSubmissionChatHandler(
+  params: { submissionId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const submissionConversation = await getSubmissionConversation({
+      member,
+      submissionId: params.submissionId,
+    });
+    if (!submissionConversation.ok) {
+      return reply.status(404).send({
+        description: `Submission with ID ${params.submissionId} not found or you don't have access to it`,
+      });
+    }
+
+    const { conversationId } = submissionConversation.data;
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-4.1",
+    });
+
+    const summary = await generateConversationSummary(model, member, conversationId);
+    await persistSummary({ member, conversationId, summary });
+
+    return reply.send({ success: true });
+  } catch (error) {
+    console.error("Error in summarizeSubmissionChat:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function chatAgentStreamHandler(
+  params: { conversationId: string },
+  body: ChatRequest,
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const { message } = body;
+
+    const agentConversation = await getAgentConversation({
+      member,
+      conversationId: params.conversationId,
+    });
+    if (!agentConversation.ok) {
+      return reply.status(404).send({
+        description: `Agent conversation with ID ${params.conversationId} not found or you don't have access to it`,
+      });
+    }
+
+    const { id: conversationId } = agentConversation.data;
+    const uiMessage = {
+      id: nanoid(),
+      role: "user" as const,
+      parts: [{ type: "text" as const, text: message }],
+    };
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-5.2",
+    });
+
+    const textStream = await chatStreamAgent(
+      model,
+      member,
+      conversationId,
+      uiMessage as UIMessage,
+      "agent",
+      false,
+    );
+
+    return reply.send(textStream);
+  } catch (error) {
+    console.error("Error in chatAgentStream:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function getAgentChatHistoryHandler(
+  params: { conversationId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const agentConversation = await getAgentConversation({
+      member,
+      conversationId: params.conversationId,
+    });
+    if (!agentConversation.ok) {
+      return reply.status(404).send({
+        description: `Agent conversation with ID ${params.conversationId} not found or you don't have access to it`,
+      });
+    }
+
+    const { id: conversationId } = agentConversation.data;
+    const conversationResult = await getConversation({ member, conversationId });
+    if (!conversationResult.ok) {
+      return reply.status(404).send({ description: "Conversation not found" });
+    }
+
+    const { messages } = conversationResult.data;
+    const processedMessages = (messages as UIMessage[]).map((m) => ({
+      id: m.id,
+      role: m.role,
+      parts: processPartsForResponse(m.parts),
+    }));
+
+    return reply.send(processedMessages);
+  } catch (error) {
+    console.error("Error in getAgentChatHistory:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function deleteAgentChatHandler(
+  params: { conversationId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const agentConversation = await getAgentConversation({
+      member,
+      conversationId: params.conversationId,
+    });
+    if (!agentConversation.ok) {
+      return reply.status(404).send({
+        description: `Agent conversation with ID ${params.conversationId} not found or you don't have access to it`,
+      });
+    }
+
+    const { id: conversationId } = agentConversation.data;
+    await eraseConversation({ member, conversationId });
+
+    return reply.send({ success: true });
+  } catch (error) {
+    console.error("Error in deleteAgentChat:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}
+
+export async function summarizeAgentChatHandler(
+  params: { conversationId: string },
+  member: MemberHydrated,
+  reply: FastifyReply,
+) {
+  try {
+    const agentConversation = await getAgentConversation({
+      member,
+      conversationId: params.conversationId,
+    });
+    if (!agentConversation.ok) {
+      return reply.status(404).send({
+        description: `Agent conversation with ID ${params.conversationId} not found or you don't have access to it`,
+      });
+    }
+
+    const { id: conversationId } = agentConversation.data;
+
+    const config = getApiConfig();
+    const model = getChatModel({
+      aiGatewayApiKey: config.vercel.aiGatewayApiKey,
+      modelId: "openai/gpt-4.1",
+    });
+
+    const summary = await generateConversationSummary(model, member, conversationId);
+    await persistSummary({ member, conversationId, summary });
+
+    return reply.send({ success: true });
+  } catch (error) {
+    console.error("Error in summarizeAgentChat:", error);
+    return reply.status(500).send({
+      description: error instanceof Error ? error.message : "An unexpected error occurred",
+    });
+  }
+}