Restructure agent chat messages with parts-based architecture (#14749)

Co-authored-by: Félix Malfait <felix@twenty.com>
2025-09-29 17:01:55 +05:30
parent 84cbd8e092
commit 2685f4a5b9
69 changed files with 1200 additions and 1539 deletions
@@ -0,0 +1,67 @@
+import {
+  type ReasoningUIPart,
+  type ToolUIPart,
+  type UIMessagePart,
+  type UITool,
+} from 'ai';
+import { type AgentChatMessagePart } from '~/generated/graphql';
+
+export const mapDBPartToUIMessagePart = (
+  part: AgentChatMessagePart,
+): UIMessagePart<never, Record<string, UITool>> => {
+  switch (part.type) {
+    case 'text':
+      return {
+        type: 'text',
+        text: part.textContent!,
+      };
+    case 'reasoning':
+      return {
+        type: 'reasoning',
+        text: part.reasoningContent!,
+        state: part.state as ReasoningUIPart['state'],
+      };
+    case 'file':
+      return {
+        type: 'file',
+        mediaType: part.fileMediaType!,
+        filename: part.fileFilename!,
+        url: part.fileUrl!,
+      };
+    case 'source-url':
+      return {
+        type: 'source-url',
+        sourceId: part.sourceUrlSourceId!,
+        url: part.sourceUrlUrl!,
+        title: part.sourceUrlTitle!,
+        providerMetadata: part.providerMetadata ?? undefined,
+      };
+    case 'source-document':
+      return {
+        type: 'source-document',
+        sourceId: part.sourceDocumentSourceId!,
+        mediaType: part.sourceDocumentMediaType!,
+        title: part.sourceDocumentTitle!,
+        filename: part.sourceDocumentFilename!,
+        providerMetadata: part.providerMetadata ?? undefined,
+      };
+    case 'step-start':
+      return {
+        type: 'step-start',
+      };
+    default:
+      {
+        if (part.type.includes('tool-') === true) {
+          return {
+            type: part.type as `tool-${string}`,
+            toolCallId: part.toolCallId!,
+            input: part.toolInput,
+            output: part.toolOutput,
+            errorText: part.errorMessage!,
+            state: part.state,
+          } as ToolUIPart;
+        }
+      }
+      throw new Error(`Unsupported part type: ${part.type}`);
+  }
+};