From 2e205d7400f449646e24e9a5cc7bcdc623c5e07b Mon Sep 17 00:00:00 2001
From: Tat Dat Duong <david@duong.cz>
Date: Mon, 17 Feb 2025 05:15:11 +0100
Subject: [PATCH] Fix TS and build

---
 libs/langgraph-api/scripts/build.mjs | 3 +++
 libs/langgraph-api/src/ui/load.mts   | 3 +--
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/libs/langgraph-api/scripts/build.mjs b/libs/langgraph-api/scripts/build.mjs
index d83c7b4..d614af6 100644
--- a/libs/langgraph-api/scripts/build.mjs
+++ b/libs/langgraph-api/scripts/build.mjs
@@ -11,5 +11,8 @@ await $`pnpm tsc --module nodenext --outDir dist/src/cli -d src/cli/spawn.mts`;
 await $`cp src/graph/parser/schema/types.template.mts dist/src/graph/parser/schema`;
 await $`rm -rf dist/src/graph/parser/schema/types.template.mjs`;
 
+await $`cp src/ui/render.mts dist/src/ui`;
+await $`rm -rf dist/src/ui/render.mjs`;
+
 await $`mv dist/src/* dist`;
 await $`rm -rf dist/src dist/tests`;
diff --git a/libs/langgraph-api/src/ui/load.mts b/libs/langgraph-api/src/ui/load.mts
index b55b375..99a2681 100644
--- a/libs/langgraph-api/src/ui/load.mts
+++ b/libs/langgraph-api/src/ui/load.mts
@@ -147,8 +147,7 @@ api.get("/ui/:agent/:basename", async (c) => {
   const file = GRAPH_UI[agent]?.find((item) => item.basename === basename);
   if (!file) return c.text("File not found", 404);
 
-  // @ts-expect-error TODO weird TS error?
-  return c.body(file.contents, {
+  return c.body(file.contents as unknown as ArrayBuffer, {
     headers: { "Content-Type": getMimeType(file.basename) ?? "text/plain" },
   });
 });