feat(ai-sdk): Adapters support (attachment, feedback, speech, threadM…

…anager) (#1037) * feat(ai-sdk): Adapters support (attachment, feedback, speech, threadManager) * add changesets
Yonom · Oct 21, 2024 · b38165d · b38165d
1 parent fadb2ec
commit b38165d
Show file tree

Hide file tree

Showing 9 changed files with 39 additions and 6 deletions.
diff --git a/.changeset/calm-countries-argue.md b/.changeset/calm-countries-argue.md
@@ -0,0 +1,5 @@
+---
+"@assistant-ui/react": patch
+---
+
+feat: export useAttachmentRuntime, useAttachment, FeedbackAdapter
diff --git a/.changeset/small-avocados-unite.md b/.changeset/small-avocados-unite.md
@@ -0,0 +1,6 @@
+---
+"@assistant-ui/react-ai-sdk": patch
+"@assistant-ui/react": patch
+---
+
+feat(ai-sdk): Adapters support (attachment, feedback, speech, threadManager)
diff --git a/packages/react-ai-sdk/src/rsc/VercelRSCAdapter.tsx b/packages/react-ai-sdk/src/rsc/VercelRSCAdapter.tsx
@@ -1,6 +1,7 @@
 "use client";
 import type { AppendMessage } from "@assistant-ui/react";
 import type { VercelRSCMessage } from "./VercelRSCMessage";
+import { ExternalStoreAdapter } from "@assistant-ui/react";
 
 type RSCMessageConverter<T> = {
   convertMessage: (message: T) => VercelRSCMessage;
@@ -15,6 +16,8 @@ type VercelRSCAdapterBase<T> = {
   onReload?: ((parentId: string | null) => Promise<void>) | undefined;
   convertMessage?: ((message: T) => VercelRSCMessage) | undefined;
 
+  adapters?: ExternalStoreAdapter["adapters"] | undefined;
+
   /**
    * @deprecated Use `onNew` instead. This will be removed in 0.6.0.
    */

diff --git a/packages/react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx b/packages/react-ai-sdk/src/rsc/useVercelRSCRuntime.tsx
@@ -39,6 +39,7 @@ export const useVercelRSCRuntime = <T extends WeakKey>(
         adapter.convertMessage ?? ((m) => m as VercelRSCMessage),
         m,
       ),
+    adapters: adapter.adapters,
     unstable_capabilities: {
       copy: false,
     },

diff --git a/packages/react-ai-sdk/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx b/packages/react-ai-sdk/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx
@@ -7,9 +7,15 @@ import { convertMessage } from "../utils/convertMessage";
 import { useInputSync } from "../utils/useInputSync";
 import { toCreateMessage } from "../utils/toCreateMessage";
 import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter";
+import { ExternalStoreAdapter } from "@assistant-ui/react";
+
+export type VercelUseChatAdapter = {
+  adapters?: Omit<ExternalStoreAdapter["adapters"], "attachments"> | undefined;
+};
 
 export const useVercelUseAssistantRuntime = (
   assistantHelpers: ReturnType<typeof useAssistant>,
+  adapter: VercelUseChatAdapter = {},
 ) => {
   const messages = useExternalMessageConverter({
     callback: convertMessage,
@@ -31,6 +37,7 @@ export const useVercelUseAssistantRuntime = (
     },
     adapters: {
       attachments: vercelAttachmentAdapter,
+      ...adapter.adapters,
     },
   });
 

diff --git a/packages/react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx b/packages/react-ai-sdk/src/ui/use-chat/useVercelUseChatRuntime.tsx
@@ -9,9 +9,15 @@ import { sliceMessagesUntil } from "../utils/sliceMessagesUntil";
 import { toCreateMessage } from "../utils/toCreateMessage";
 import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter";
 import { getVercelAIMessages } from "../getVercelAIMessages";
+import { ExternalStoreAdapter } from "@assistant-ui/react";
+
+export type VercelUseChatAdapter = {
+  adapters?: Omit<ExternalStoreAdapter["adapters"], "attachments"> | undefined;
+};
 
 export const useVercelUseChatRuntime = (
   chatHelpers: ReturnType<typeof useChat>,
+  adapter: VercelUseChatAdapter = {},
 ) => {
   const messages = useExternalMessageConverter({
     callback: convertMessage,
@@ -54,6 +60,7 @@ export const useVercelUseChatRuntime = (
     },
     adapters: {
       attachments: vercelAttachmentAdapter,
+      ...adapter.adapters,
     },
   });
 

diff --git a/packages/react/src/runtimes/external-store/ExternalStoreAdapter.tsx b/packages/react/src/runtimes/external-store/ExternalStoreAdapter.tsx
@@ -60,12 +60,14 @@ type ExternalStoreAdapterBase<T> = {
     | ((options: AddToolResultOptions) => Promise<void> | void)
     | undefined;
   convertMessage?: ExternalStoreMessageConverter<T> | undefined;
-  adapters?: {
-    attachments?: AttachmentAdapter | undefined;
-    speech?: SpeechSynthesisAdapter | undefined;
-    feedback?: FeedbackAdapter | undefined;
-    threadManager?: ExternalStoreThreadManagerAdapter | undefined;
-  };
+  adapters?:
+    | {
+        attachments?: AttachmentAdapter | undefined;
+        speech?: SpeechSynthesisAdapter | undefined;
+        feedback?: FeedbackAdapter | undefined;
+        threadManager?: ExternalStoreThreadManagerAdapter | undefined;
+      }
+    | undefined;
   unstable_capabilities?:
     | {
         copy?: boolean | undefined;

diff --git a/packages/react/src/runtimes/feedback/index.ts b/packages/react/src/runtimes/feedback/index.ts
@@ -0,0 +1 @@
+export { type FeedbackAdapter } from "./FeedbackAdapter";
diff --git a/packages/react/src/runtimes/index.ts b/packages/react/src/runtimes/index.ts
@@ -5,3 +5,4 @@ export * from "./external-store";
 export * from "./dangerous-in-browser";
 export * from "./speech";
 export * from "./attachment";
+export * from "./feedback";
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		export { type FeedbackAdapter } from "./FeedbackAdapter";