From 857fd3578fe86192af8f50bd2fc3b5d5c08bb71c Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Tue, 11 Mar 2025 15:02:42 +0100
Subject: [PATCH] feat(sdk-js): cleanup types for ui payloads

---
 libs/sdk-js/src/react-ui/client.tsx       |  4 ++--
 libs/sdk-js/src/react-ui/server/server.ts | 10 +++++-----
 libs/sdk-js/src/react-ui/types.ts         |  4 ++--
 3 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/libs/sdk-js/src/react-ui/client.tsx b/libs/sdk-js/src/react-ui/client.tsx
index b1ca064ac..ada4b8c11 100644
--- a/libs/sdk-js/src/react-ui/client.tsx
+++ b/libs/sdk-js/src/react-ui/client.tsx
@@ -175,7 +175,7 @@ export function LoadExternalComponent({
   ]);
 
   if (hasClientComponent) {
-    return React.createElement(clientComponent, message.content);
+    return React.createElement(clientComponent, message.props);
   }
 
   return (
@@ -185,7 +185,7 @@ export function LoadExternalComponent({
       <UseStreamContext.Provider value={{ stream, meta }}>
         {state?.target != null
           ? ReactDOM.createPortal(
-              React.createElement(state.comp, message.content),
+              React.createElement(state.comp, message.props),
               state.target,
             )
           : fallback}
diff --git a/libs/sdk-js/src/react-ui/server/server.ts b/libs/sdk-js/src/react-ui/server/server.ts
index 62263ae3c..cde477bc1 100644
--- a/libs/sdk-js/src/react-ui/server/server.ts
+++ b/libs/sdk-js/src/react-ui/server/server.ts
@@ -30,8 +30,8 @@ export const typedUi = <Decl extends Record<string, ElementType>>(config: {
     message: {
       id?: string;
       name: K;
-      content: PropMap[K];
-      additional_kwargs?: Record<string, unknown>;
+      props: PropMap[K];
+      metadata?: Record<string, unknown>;
     },
     options?: { message?: MessageLike },
   ): UIMessage => {
@@ -39,10 +39,10 @@ export const typedUi = <Decl extends Record<string, ElementType>>(config: {
       type: "ui" as const,
       id: message?.id ?? uuidv4(),
       name: message?.name,
-      content: message?.content,
-      additional_kwargs: {
+      props: message?.props,
+      metadata: {
         ...metadata,
-        ...message?.additional_kwargs,
+        ...message?.metadata,
         ...(options?.message ? { message_id: options.message.id } : null),
       },
     };
diff --git a/libs/sdk-js/src/react-ui/types.ts b/libs/sdk-js/src/react-ui/types.ts
index 86f538e0c..c3f52b5b8 100644
--- a/libs/sdk-js/src/react-ui/types.ts
+++ b/libs/sdk-js/src/react-ui/types.ts
@@ -3,8 +3,8 @@ export interface UIMessage {
 
   id: string;
   name: string;
-  content: Record<string, unknown>;
-  additional_kwargs: {
+  props: Record<string, unknown>;
+  metadata: {
     run_id: string;
     message_id?: string;
     [key: string]: unknown;