feat: upgrade to 0.0.52

agent/open-code/nodes/executor.ts

@@ -106,17 +106,22 @@ export async function executor(
 
   const fullWriteAccess = !!config.configurable?.permissions?.full_write_access;
 
-  const msg = ui.create("proposed-change", {
-    toolCallId,
-    change: updateFileContents,
-    planItem: nextPlanItem,
-    fullWriteAccess,
-  });
-  msg.additional_kwargs["message_id"] = aiMessage.id;
+  ui.push(
+    {
+      name: "proposed-change",
+      content: {
+        toolCallId,
+        change: updateFileContents,
+        planItem: nextPlanItem,
+        fullWriteAccess,
+      },
+    },
+    { message: aiMessage },
+  );
 
   return {
     messages: [aiMessage],
-    ui: [msg],
+    ui: ui.items,
     timestamp: Date.now(),
   };
 }

agent/open-code/nodes/planner.ts

@@ -86,13 +86,18 @@ export async function planner(
     ],
   };
 
-  const msg = ui.create("code-plan", {
-    toolCallId,
-    executedPlans,
-    rejectedPlans,
-    remainingPlans,
-  });
-  msg.additional_kwargs["message_id"] = aiMessage.id;
+  ui.push(
+    {
+      name: "code-plan",
+      content: {
+        toolCallId,
+        executedPlans,
+        rejectedPlans,
+        remainingPlans,
+      },
+    },
+    { message: aiMessage },
+  );
 
   const toolMessage: ToolMessage = {
     type: "tool",
@@ -103,7 +108,7 @@ export async function planner(
 
   return {
     messages: [aiMessage, toolMessage],
-    ui: [msg],
+    ui: ui.items,
     timestamp: Date.now(),
   };
 }

agent/stockbroker/nodes/tools.tsx

@@ -176,29 +176,38 @@ export async function callTools(
 
   if (stockbrokerToolCall) {
     const prices = await getPricesForTicker(stockbrokerToolCall.args.ticker);
-    ui.write("stock-price", {
-      ticker: stockbrokerToolCall.args.ticker,
-      ...prices,
-    });
+    ui.push(
+      {
+        name: "stock-price",
+        content: { ticker: stockbrokerToolCall.args.ticker, ...prices },
+      },
+      { message },
+    );
   }
   if (portfolioToolCall) {
-    ui.write("portfolio", {});
+    ui.push({ name: "portfolio", content: {} }, { message });
   }
   if (buyStockToolCall) {
     const snapshot = await getPriceSnapshotForTicker(
       buyStockToolCall.args.ticker,
     );
-    ui.write("buy-stock", {
-      toolCallId:
-        message.tool_calls?.find((tc) => tc.name === "buy-stock")?.id ?? "",
-      snapshot,
-      quantity: buyStockToolCall.args.quantity,
-    });
+    ui.push(
+      {
+        name: "buy-stock",
+        content: {
+          toolCallId:
+            message.tool_calls?.find((tc) => tc.name === "buy-stock")?.id ?? "",
+          snapshot,
+          quantity: buyStockToolCall.args.quantity,
+        },
+      },
+      { message },
+    );
   }
 
   return {
     messages: [message],
-    ui: ui.collect as StockbrokerUpdate["ui"],
+    ui: ui.items,
     timestamp: Date.now(),
   };
 }

agent/trip-planner/nodes/tools.tsx

@@ -87,32 +87,56 @@ export async function callTools(
   }
 
   if (tripPlan.listAccommodations) {
-    ui.write("accommodations-list", {
-      toolCallId,
-      ...getAccommodationsListProps(state.tripDetails),
-    });
+    ui.push(
+      {
+        name: "accommodations-list",
+        content: {
+          toolCallId,
+          ...getAccommodationsListProps(state.tripDetails),
+        },
+      },
+      { message: response },
+    );
   }
   if (tripPlan.bookAccommodation && tripPlan.accommodationName) {
-    ui.write("book-accommodation", {
-      tripDetails: state.tripDetails,
-      accommodationName: tripPlan.accommodationName,
-    });
+    ui.push(
+      {
+        name: "book-accommodation",
+        content: {
+          tripDetails: state.tripDetails,
+          accommodationName: tripPlan.accommodationName,
+        },
+      },
+      { message: response },
+    );
   }
 
   if (tripPlan.listRestaurants) {
-    ui.write("restaurants-list", { tripDetails: state.tripDetails });
+    ui.push(
+      {
+        name: "restaurants-list",
+        content: { tripDetails: state.tripDetails },
+      },
+      { message: response },
+    );
   }
 
   if (tripPlan.bookRestaurant && tripPlan.restaurantName) {
-    ui.write("book-restaurant", {
-      tripDetails: state.tripDetails,
-      restaurantName: tripPlan.restaurantName,
-    });
+    ui.push(
+      {
+        name: "book-restaurant",
+        content: {
+          tripDetails: state.tripDetails,
+          restaurantName: tripPlan.restaurantName,
+        },
+      },
+      { message: response },
+    );
   }
 
   return {
     messages: [response],
-    ui: ui.collect as TripPlannerUpdate["ui"],
+    ui: ui.items,
     timestamp: Date.now(),
   };
 }