Merge pull request #15 from langchain-ai/dqbd/update-types

feat: update types, refine ui sdk

src/components/thread/messages/ai.tsx

@@ -4,7 +4,7 @@ import { getContentString } from "../utils";
 import { BranchSwitcher, CommandBar } from "./shared";
 import { Avatar, AvatarFallback } from "@/components/ui/avatar";
 import { MarkdownText } from "../markdown-text";
-import { LoadExternalComponent } from "@langchain/langgraph-sdk/react-ui/client";
+import { LoadExternalComponent } from "@langchain/langgraph-sdk/react-ui";
 import { cn } from "@/lib/utils";
 import { ToolCalls, ToolResult } from "./tool-calls";
 
@@ -36,6 +36,7 @@ function CustomComponent({
           assistantId="agent"
           stream={thread}
           message={customComponent}
+          meta={{ ui: customComponent }}
         />
       )}
     </div>

src/providers/Stream.tsx

@@ -4,7 +4,7 @@ import { type Message } from "@langchain/langgraph-sdk";
 import type {
   UIMessage,
   RemoveUIMessage,
-} from "@langchain/langgraph-sdk/react-ui/types";
+} from "@langchain/langgraph-sdk/react-ui";
 import { useQueryParam, StringParam } from "use-query-params";
 import { Input } from "@/components/ui/input";
 import { Button } from "@/components/ui/button";