@assistant-ui/react-ai-sdk 0.10.1 → 0.10.3

package/src/ui/utils/useInputSync.tsx

@@ -0,0 +1,24 @@
+import { useEffect } from "react";
+import { useAssistant, useChat } from "@ai-sdk/react";
+import { AssistantRuntime } from "@assistant-ui/react";
+
+type VercelHelpers =
+  | ReturnType<typeof useChat>
+  | ReturnType<typeof useAssistant>;
+
+export const useInputSync = (
+  { setInput, input }: VercelHelpers,
+  runtime: AssistantRuntime,
+) => {
+  // sync input from vercel to assistant-ui
+  useEffect(() => {
+    runtime.thread.composer.setText(input);
+  }, [runtime, input]);
+
+  // sync input from assistant-ui to vercel
+  useEffect(() => {
+    return runtime.thread.composer.subscribe(() => {
+      setInput(runtime.thread.composer.getState().text);
+    });
+  }, [runtime, setInput]);
+};

package/src/ui/utils/vercelAttachmentAdapter.ts

@@ -0,0 +1,29 @@
+import { AttachmentAdapter } from "@assistant-ui/react";
+import { generateId } from "@ai-sdk/ui-utils";
+
+export const vercelAttachmentAdapter: AttachmentAdapter = {
+  accept:
+    "image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css",
+  async add({ file }) {
+    return {
+      id: generateId(),
+      type: "file",
+      name: file.name,
+      file,
+      contentType: file.type,
+      content: [],
+      status: { type: "requires-action", reason: "composer-send" },
+    };
+  },
+  async send(attachment) {
+    // noop
+    return {
+      ...attachment,
+      status: { type: "complete" },
+      content: [],
+    };
+  },
+  async remove() {
+    // noop
+  },
+};

package/src/useChatRuntime.ts

@@ -0,0 +1,10 @@
+import { EdgeRuntimeOptions, useEdgeRuntime } from "@assistant-ui/react-edge";
+
+type UseChatRuntimeOptions = Omit<EdgeRuntimeOptions, "unstable_AISDKInterop">;
+
+export const useChatRuntime = (options: UseChatRuntimeOptions) => {
+  return useEdgeRuntime({
+    ...options,
+    unstable_AISDKInterop: "v2",
+  });
+};