@assistant-ui/react-ai-sdk 0.10.1 → 0.10.2
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/index.d.ts +5 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/rsc/RSCDisplay.d.ts +3 -0
- package/dist/rsc/RSCDisplay.d.ts.map +1 -0
- package/dist/rsc/VercelRSCAdapter.d.ts +18 -0
- package/dist/rsc/VercelRSCAdapter.d.ts.map +1 -0
- package/dist/rsc/VercelRSCMessage.d.ts +8 -0
- package/dist/rsc/VercelRSCMessage.d.ts.map +1 -0
- package/dist/rsc/getVercelRSCMessage.d.ts +7 -0
- package/dist/rsc/getVercelRSCMessage.d.ts.map +1 -0
- package/dist/rsc/index.d.ts +6 -0
- package/dist/rsc/index.d.ts.map +1 -0
- package/dist/rsc/useVercelRSCRuntime.d.ts +3 -0
- package/dist/rsc/useVercelRSCRuntime.d.ts.map +1 -0
- package/dist/rsc/utils/RSCThreadExtras.d.ts +8 -0
- package/dist/rsc/utils/RSCThreadExtras.d.ts.map +1 -0
- package/dist/ui/getVercelAIMessages.d.ts +4 -0
- package/dist/ui/getVercelAIMessages.d.ts.map +1 -0
- package/dist/ui/index.d.ts +5 -0
- package/dist/ui/index.d.ts.map +1 -0
- package/dist/ui/use-assistant/useVercelUseAssistantRuntime.d.ts +7 -0
- package/dist/ui/use-assistant/useVercelUseAssistantRuntime.d.ts.map +1 -0
- package/dist/ui/use-chat/useVercelUseChatRuntime.d.ts +8 -0
- package/dist/ui/use-chat/useVercelUseChatRuntime.d.ts.map +1 -0
- package/dist/ui/utils/convertMessage.d.ts +14 -0
- package/dist/ui/utils/convertMessage.d.ts.map +1 -0
- package/dist/ui/utils/sliceMessagesUntil.d.ts +3 -0
- package/dist/ui/utils/sliceMessagesUntil.d.ts.map +1 -0
- package/dist/ui/utils/toCreateMessage.d.ts +4 -0
- package/dist/ui/utils/toCreateMessage.d.ts.map +1 -0
- package/dist/ui/utils/useInputSync.d.ts +6 -0
- package/dist/ui/utils/useInputSync.d.ts.map +1 -0
- package/dist/ui/utils/vercelAttachmentAdapter.d.ts +3 -0
- package/dist/ui/utils/vercelAttachmentAdapter.d.ts.map +1 -0
- package/dist/useChatRuntime.d.ts +5 -0
- package/dist/useChatRuntime.d.ts.map +1 -0
- package/package.json +5 -4
- package/src/index.ts +11 -0
- package/src/rsc/RSCDisplay.tsx +23 -0
- package/src/rsc/VercelRSCAdapter.tsx +23 -0
- package/src/rsc/VercelRSCMessage.tsx +9 -0
- package/src/rsc/getVercelRSCMessage.tsx +12 -0
- package/src/rsc/index.ts +5 -0
- package/src/rsc/useVercelRSCRuntime.tsx +70 -0
- package/src/rsc/utils/RSCThreadExtras.tsx +11 -0
- package/src/ui/getVercelAIMessages.tsx +9 -0
- package/src/ui/index.ts +4 -0
- package/src/ui/use-assistant/useVercelUseAssistantRuntime.tsx +58 -0
- package/src/ui/use-chat/useVercelUseChatRuntime.tsx +91 -0
- package/src/ui/utils/convertMessage.ts +158 -0
- package/src/ui/utils/sliceMessagesUntil.tsx +20 -0
- package/src/ui/utils/toCreateMessage.ts +44 -0
- package/src/ui/utils/useInputSync.tsx +24 -0
- package/src/ui/utils/vercelAttachmentAdapter.ts +29 -0
- package/src/useChatRuntime.ts +10 -0
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export * from "./rsc";
|
|
2
|
+
export * from "./ui";
|
|
3
|
+
export * from "./useChatRuntime";
|
|
4
|
+
export { toLanguageModelMessages, toLanguageModelTools, fromLanguageModelMessages, fromLanguageModelTools, useDangerousInBrowserRuntime, } from "@assistant-ui/react-edge";
|
|
5
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,OAAO,CAAC;AACtB,cAAc,MAAM,CAAC;AACrB,cAAc,kBAAkB,CAAC;AAEjC,OAAO,EACL,uBAAuB,EACvB,oBAAoB,EACpB,yBAAyB,EACzB,sBAAsB,EACtB,4BAA4B,GAC7B,MAAM,0BAA0B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"RSCDisplay.d.ts","sourceRoot":"","sources":["../../src/rsc/RSCDisplay.tsx"],"names":[],"mappings":"AAUA,OAAO,EAAE,EAAE,EAAE,MAAM,OAAO,CAAC;AAE3B,eAAO,MAAM,UAAU,EAAE,EAUxB,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { AppendMessage } from "@assistant-ui/react";
|
|
2
|
+
import type { VercelRSCMessage } from "./VercelRSCMessage";
|
|
3
|
+
import { ExternalStoreAdapter } from "@assistant-ui/react";
|
|
4
|
+
type RSCMessageConverter<T> = {
|
|
5
|
+
convertMessage: (message: T) => VercelRSCMessage;
|
|
6
|
+
};
|
|
7
|
+
type VercelRSCAdapterBase<T> = {
|
|
8
|
+
isRunning?: boolean | undefined;
|
|
9
|
+
messages: T[];
|
|
10
|
+
onNew?: (message: AppendMessage) => Promise<void>;
|
|
11
|
+
onEdit?: ((message: AppendMessage) => Promise<void>) | undefined;
|
|
12
|
+
onReload?: ((parentId: string | null) => Promise<void>) | undefined;
|
|
13
|
+
convertMessage?: ((message: T) => VercelRSCMessage) | undefined;
|
|
14
|
+
adapters?: ExternalStoreAdapter["adapters"] | undefined;
|
|
15
|
+
};
|
|
16
|
+
export type VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> & (T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
|
|
17
|
+
export {};
|
|
18
|
+
//# sourceMappingURL=VercelRSCAdapter.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"VercelRSCAdapter.d.ts","sourceRoot":"","sources":["../../src/rsc/VercelRSCAdapter.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACzD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAE3D,KAAK,mBAAmB,CAAC,CAAC,IAAI;IAC5B,cAAc,EAAE,CAAC,OAAO,EAAE,CAAC,KAAK,gBAAgB,CAAC;CAClD,CAAC;AAEF,KAAK,oBAAoB,CAAC,CAAC,IAAI;IAC7B,SAAS,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAChC,QAAQ,EAAE,CAAC,EAAE,CAAC;IAEd,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAClD,MAAM,CAAC,EAAE,CAAC,CAAC,OAAO,EAAE,aAAa,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;IACjE,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS,CAAC;IACpE,cAAc,CAAC,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,KAAK,gBAAgB,CAAC,GAAG,SAAS,CAAC;IAEhE,QAAQ,CAAC,EAAE,oBAAoB,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;CACzD,CAAC;AAEF,MAAM,MAAM,gBAAgB,CAAC,CAAC,GAAG,gBAAgB,IAAI,oBAAoB,CAAC,CAAC,CAAC,GAC1E,CAAC,CAAC,SAAS,gBAAgB,GAAG,MAAM,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"VercelRSCMessage.d.ts","sourceRoot":"","sources":["../../src/rsc/VercelRSCMessage.tsx"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,OAAO,CAAC;AAEvC,MAAM,MAAM,gBAAgB,GAAG;IAC7B,EAAE,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IACxB,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAC3B,OAAO,EAAE,SAAS,CAAC;IACnB,SAAS,CAAC,EAAE,IAAI,GAAG,SAAS,CAAC;CAC9B,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { type ThreadMessage } from "@assistant-ui/react";
|
|
2
|
+
/**
|
|
3
|
+
* @deprecated Use `getExternalStoreMessage` instead. This method was specific to Vercel RSC
|
|
4
|
+
* implementation and has been replaced by a more generic external store message handler.
|
|
5
|
+
*/
|
|
6
|
+
export declare const getVercelRSCMessage: (message: ThreadMessage) => unknown;
|
|
7
|
+
//# sourceMappingURL=getVercelRSCMessage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"getVercelRSCMessage.d.ts","sourceRoot":"","sources":["../../src/rsc/getVercelRSCMessage.tsx"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,qBAAqB,CAAC;AAE7B;;;GAGG;AACH,eAAO,MAAM,mBAAmB,GAAI,SAAS,aAAa,YAEzD,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
export { useVercelRSCRuntime } from "./useVercelRSCRuntime";
|
|
2
|
+
export { getVercelRSCMessage } from "./getVercelRSCMessage";
|
|
3
|
+
export type { VercelRSCAdapter } from "./VercelRSCAdapter";
|
|
4
|
+
export type { VercelRSCMessage } from "./VercelRSCMessage";
|
|
5
|
+
export { RSCDisplay } from "./RSCDisplay";
|
|
6
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/rsc/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,YAAY,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,YAAY,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useVercelRSCRuntime.d.ts","sourceRoot":"","sources":["../../src/rsc/useVercelRSCRuntime.tsx"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAyB3D,eAAO,MAAM,mBAAmB,GAAI,CAAC,SAAS,OAAO,EACnD,SAAS,gBAAgB,CAAC,CAAC,CAAC,mDAyC7B,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import { VercelRSCMessage } from "../VercelRSCMessage";
|
|
2
|
+
export declare const symbolInternalRSCExtras: unique symbol;
|
|
3
|
+
export type RSCThreadExtras = {
|
|
4
|
+
[symbolInternalRSCExtras]?: {
|
|
5
|
+
convertFn: (message: any) => VercelRSCMessage;
|
|
6
|
+
};
|
|
7
|
+
} | undefined;
|
|
8
|
+
//# sourceMappingURL=RSCThreadExtras.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"RSCThreadExtras.d.ts","sourceRoot":"","sources":["../../../src/rsc/utils/RSCThreadExtras.tsx"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,eAAO,MAAM,uBAAuB,eAAgC,CAAC;AACrE,MAAM,MAAM,eAAe,GACvB;IACE,CAAC,uBAAuB,CAAC,CAAC,EAAE;QAC1B,SAAS,EAAE,CAAC,OAAO,EAAE,GAAG,KAAK,gBAAgB,CAAC;KAC/C,CAAC;CACH,GACD,SAAS,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.tsx"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAEhD,eAAO,MAAM,mBAAmB,GAAI,SAAS,aAAa,KACb,OAAO,EACnD,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { useVercelUseChatRuntime } from "./use-chat/useVercelUseChatRuntime";
|
|
2
|
+
export { useVercelUseAssistantRuntime } from "./use-assistant/useVercelUseAssistantRuntime";
|
|
3
|
+
export { getVercelAIMessages } from "./getVercelAIMessages";
|
|
4
|
+
export { AISDKMessageConverter } from "./utils/convertMessage";
|
|
5
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/ui/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,uBAAuB,EAAE,MAAM,oCAAoC,CAAC;AAC7E,OAAO,EAAE,4BAA4B,EAAE,MAAM,8CAA8C,CAAC;AAC5F,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,qBAAqB,EAAE,MAAM,wBAAwB,CAAC"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { useAssistant } from "@ai-sdk/react";
|
|
2
|
+
import { ExternalStoreAdapter } from "@assistant-ui/react";
|
|
3
|
+
export type VercelUseChatAdapter = {
|
|
4
|
+
adapters?: Omit<NonNullable<ExternalStoreAdapter["adapters"]>, "attachments"> | undefined;
|
|
5
|
+
};
|
|
6
|
+
export declare const useVercelUseAssistantRuntime: (assistantHelpers: ReturnType<typeof useAssistant>, adapter?: VercelUseChatAdapter) => import("@assistant-ui/react").AssistantRuntime;
|
|
7
|
+
//# sourceMappingURL=useVercelUseAssistantRuntime.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useVercelUseAssistantRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-assistant/useVercelUseAssistantRuntime.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAKlD,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAG3D,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,CAAC,EACL,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,EAAE,aAAa,CAAC,GAClE,SAAS,CAAC;CACf,CAAC;AAEF,eAAO,MAAM,4BAA4B,GACvC,kBAAkB,UAAU,CAAC,OAAO,YAAY,CAAC,EACjD,UAAS,oBAAyB,mDAyCnC,CAAC"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
import type { useChat } from "@ai-sdk/react";
|
|
2
|
+
import { ExternalStoreAdapter } from "@assistant-ui/react";
|
|
3
|
+
export type VercelUseChatAdapter = {
|
|
4
|
+
adapters?: Omit<NonNullable<ExternalStoreAdapter["adapters"]>, "attachments"> | undefined;
|
|
5
|
+
unstable_joinStrategy?: "concat-content" | "none";
|
|
6
|
+
};
|
|
7
|
+
export declare const useVercelUseChatRuntime: (chatHelpers: ReturnType<typeof useChat>, adapter?: VercelUseChatAdapter) => import("@assistant-ui/react").AssistantRuntime;
|
|
8
|
+
//# sourceMappingURL=useVercelUseChatRuntime.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useVercelUseChatRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useVercelUseChatRuntime.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AAO7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,qBAAqB,CAAC;AAK3D,MAAM,MAAM,oBAAoB,GAAG;IACjC,QAAQ,CAAC,EACL,IAAI,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,EAAE,aAAa,CAAC,GAClE,SAAS,CAAC;IACd,qBAAqB,CAAC,EAAE,gBAAgB,GAAG,MAAM,CAAC;CACnD,CAAC;AAEF,eAAO,MAAM,uBAAuB,GAClC,aAAa,UAAU,CAAC,OAAO,OAAO,CAAC,EACvC,UAAS,oBAAyB,mDAqEnC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { Message } from "@ai-sdk/ui-utils";
|
|
2
|
+
export declare const AISDKMessageConverter: {
|
|
3
|
+
useThreadMessages: ({ messages, isRunning, joinStrategy, }: {
|
|
4
|
+
messages: Message[];
|
|
5
|
+
isRunning: boolean;
|
|
6
|
+
joinStrategy?: "concat-content" | "none" | undefined;
|
|
7
|
+
}) => import("@assistant-ui/react").ThreadMessage[];
|
|
8
|
+
toThreadMessages: (messages: Message[]) => import("@assistant-ui/react").ThreadMessage[];
|
|
9
|
+
toOriginalMessages: (input: import("@assistant-ui/react").ThreadState | import("@assistant-ui/react").ThreadMessage | import("@assistant-ui/react").ThreadMessage["content"][number]) => unknown[];
|
|
10
|
+
toOriginalMessage: (input: import("@assistant-ui/react").ThreadState | import("@assistant-ui/react").ThreadMessage | import("@assistant-ui/react").ThreadMessage["content"][number]) => {};
|
|
11
|
+
useOriginalMessage: () => Message;
|
|
12
|
+
useOriginalMessages: () => Message[];
|
|
13
|
+
};
|
|
14
|
+
//# sourceMappingURL=convertMessage.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAS3C,eAAO,MAAM,qBAAqB;;;;;;;0HAI1B,sDACI;yHAGF,sDAAqB;;;CA4I9B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"sliceMessagesUntil.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/sliceMessagesUntil.tsx"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAC;AAEhD,eAAO,MAAM,kBAAkB,GAC7B,UAAU,OAAO,EAAE,EACnB,WAAW,MAAM,GAAG,IAAI,cAezB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"toCreateMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/toCreateMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEjD,eAAO,MAAM,eAAe,GAC1B,SAAS,aAAa,KACrB,OAAO,CAAC,aAAa,CA4BvB,CAAC"}
|
|
@@ -0,0 +1,6 @@
|
|
|
1
|
+
import { useAssistant, useChat } from "@ai-sdk/react";
|
|
2
|
+
import { AssistantRuntime } from "@assistant-ui/react";
|
|
3
|
+
type VercelHelpers = ReturnType<typeof useChat> | ReturnType<typeof useAssistant>;
|
|
4
|
+
export declare const useInputSync: ({ setInput, input }: VercelHelpers, runtime: AssistantRuntime) => void;
|
|
5
|
+
export {};
|
|
6
|
+
//# sourceMappingURL=useInputSync.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useInputSync.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/useInputSync.tsx"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,eAAe,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,KAAK,aAAa,GACd,UAAU,CAAC,OAAO,OAAO,CAAC,GAC1B,UAAU,CAAC,OAAO,YAAY,CAAC,CAAC;AAEpC,eAAO,MAAM,YAAY,GACvB,qBAAqB,aAAa,EAClC,SAAS,gBAAgB,SAa1B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"vercelAttachmentAdapter.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/vercelAttachmentAdapter.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AAGxD,eAAO,MAAM,uBAAuB,EAAE,iBAyBrC,CAAC"}
|
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
import { EdgeRuntimeOptions } from "@assistant-ui/react-edge";
|
|
2
|
+
type UseChatRuntimeOptions = Omit<EdgeRuntimeOptions, "unstable_AISDKInterop">;
|
|
3
|
+
export declare const useChatRuntime: (options: UseChatRuntimeOptions) => import("@assistant-ui/react").AssistantRuntime;
|
|
4
|
+
export {};
|
|
5
|
+
//# sourceMappingURL=useChatRuntime.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../src/useChatRuntime.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,kBAAkB,EAAkB,MAAM,0BAA0B,CAAC;AAE9E,KAAK,qBAAqB,GAAG,IAAI,CAAC,kBAAkB,EAAE,uBAAuB,CAAC,CAAC;AAE/E,eAAO,MAAM,cAAc,GAAI,SAAS,qBAAqB,mDAK5D,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@assistant-ui/react-ai-sdk",
|
|
3
|
-
"version": "0.10.
|
|
3
|
+
"version": "0.10.2",
|
|
4
4
|
"license": "MIT",
|
|
5
5
|
"type": "module",
|
|
6
6
|
"exports": {
|
|
@@ -13,6 +13,7 @@
|
|
|
13
13
|
"types": "./dist/index.d.ts",
|
|
14
14
|
"files": [
|
|
15
15
|
"dist",
|
|
16
|
+
"src",
|
|
16
17
|
"README.md"
|
|
17
18
|
],
|
|
18
19
|
"sideEffects": false,
|
|
@@ -22,10 +23,10 @@
|
|
|
22
23
|
"@radix-ui/react-use-callback-ref": "^1.1.1",
|
|
23
24
|
"zod": "^3.24.3",
|
|
24
25
|
"zustand": "^5.0.3",
|
|
25
|
-
"@assistant-ui/react-edge": "0.2.
|
|
26
|
+
"@assistant-ui/react-edge": "0.2.1"
|
|
26
27
|
},
|
|
27
28
|
"peerDependencies": {
|
|
28
|
-
"@assistant-ui/react": "^0.10.
|
|
29
|
+
"@assistant-ui/react": "^0.10.1",
|
|
29
30
|
"@types/react": "*",
|
|
30
31
|
"react": "^18 || ^19 || ^19.0.0-rc"
|
|
31
32
|
},
|
|
@@ -36,7 +37,7 @@
|
|
|
36
37
|
},
|
|
37
38
|
"devDependencies": {
|
|
38
39
|
"@types/node": "^22.14.1",
|
|
39
|
-
"@types/react": "^19.1.
|
|
40
|
+
"@types/react": "^19.1.2",
|
|
40
41
|
"eslint": "^9",
|
|
41
42
|
"eslint-config-next": "15.3.1",
|
|
42
43
|
"react": "^19.1.0",
|
package/src/index.ts
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
export * from "./rsc";
|
|
2
|
+
export * from "./ui";
|
|
3
|
+
export * from "./useChatRuntime";
|
|
4
|
+
|
|
5
|
+
export {
|
|
6
|
+
toLanguageModelMessages,
|
|
7
|
+
toLanguageModelTools,
|
|
8
|
+
fromLanguageModelMessages,
|
|
9
|
+
fromLanguageModelTools,
|
|
10
|
+
useDangerousInBrowserRuntime,
|
|
11
|
+
} from "@assistant-ui/react-edge";
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import {
|
|
3
|
+
useThread,
|
|
4
|
+
useMessage,
|
|
5
|
+
getExternalStoreMessage,
|
|
6
|
+
} from "@assistant-ui/react";
|
|
7
|
+
import {
|
|
8
|
+
RSCThreadExtras,
|
|
9
|
+
symbolInternalRSCExtras,
|
|
10
|
+
} from "./utils/RSCThreadExtras";
|
|
11
|
+
import { FC } from "react";
|
|
12
|
+
|
|
13
|
+
export const RSCDisplay: FC = () => {
|
|
14
|
+
const convertFn = useThread((t) => {
|
|
15
|
+
const extras = (t.extras as RSCThreadExtras)?.[symbolInternalRSCExtras];
|
|
16
|
+
if (!extras)
|
|
17
|
+
throw new Error(
|
|
18
|
+
"This function can only be used inside a Vercel RSC runtime.",
|
|
19
|
+
);
|
|
20
|
+
return extras.convertFn;
|
|
21
|
+
});
|
|
22
|
+
return useMessage((m) => convertFn(getExternalStoreMessage(m)).display);
|
|
23
|
+
};
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import type { AppendMessage } from "@assistant-ui/react";
|
|
3
|
+
import type { VercelRSCMessage } from "./VercelRSCMessage";
|
|
4
|
+
import { ExternalStoreAdapter } from "@assistant-ui/react";
|
|
5
|
+
|
|
6
|
+
type RSCMessageConverter<T> = {
|
|
7
|
+
convertMessage: (message: T) => VercelRSCMessage;
|
|
8
|
+
};
|
|
9
|
+
|
|
10
|
+
type VercelRSCAdapterBase<T> = {
|
|
11
|
+
isRunning?: boolean | undefined;
|
|
12
|
+
messages: T[];
|
|
13
|
+
|
|
14
|
+
onNew?: (message: AppendMessage) => Promise<void>;
|
|
15
|
+
onEdit?: ((message: AppendMessage) => Promise<void>) | undefined;
|
|
16
|
+
onReload?: ((parentId: string | null) => Promise<void>) | undefined;
|
|
17
|
+
convertMessage?: ((message: T) => VercelRSCMessage) | undefined;
|
|
18
|
+
|
|
19
|
+
adapters?: ExternalStoreAdapter["adapters"] | undefined;
|
|
20
|
+
};
|
|
21
|
+
|
|
22
|
+
export type VercelRSCAdapter<T = VercelRSCMessage> = VercelRSCAdapterBase<T> &
|
|
23
|
+
(T extends VercelRSCMessage ? object : RSCMessageConverter<T>);
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
import {
|
|
2
|
+
getExternalStoreMessage,
|
|
3
|
+
type ThreadMessage,
|
|
4
|
+
} from "@assistant-ui/react";
|
|
5
|
+
|
|
6
|
+
/**
|
|
7
|
+
* @deprecated Use `getExternalStoreMessage` instead. This method was specific to Vercel RSC
|
|
8
|
+
* implementation and has been replaced by a more generic external store message handler.
|
|
9
|
+
*/
|
|
10
|
+
export const getVercelRSCMessage = (message: ThreadMessage) => {
|
|
11
|
+
return getExternalStoreMessage(message);
|
|
12
|
+
};
|
package/src/rsc/index.ts
ADDED
|
@@ -0,0 +1,5 @@
|
|
|
1
|
+
export { useVercelRSCRuntime } from "./useVercelRSCRuntime";
|
|
2
|
+
export { getVercelRSCMessage } from "./getVercelRSCMessage";
|
|
3
|
+
export type { VercelRSCAdapter } from "./VercelRSCAdapter";
|
|
4
|
+
export type { VercelRSCMessage } from "./VercelRSCMessage";
|
|
5
|
+
export { RSCDisplay } from "./RSCDisplay";
|
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
|
|
3
|
+
import type { VercelRSCAdapter } from "./VercelRSCAdapter";
|
|
4
|
+
import {
|
|
5
|
+
ExternalStoreAdapter,
|
|
6
|
+
ThreadMessageLike,
|
|
7
|
+
useExternalMessageConverter,
|
|
8
|
+
useExternalStoreRuntime,
|
|
9
|
+
} from "@assistant-ui/react";
|
|
10
|
+
import { VercelRSCMessage } from "./VercelRSCMessage";
|
|
11
|
+
import { useCallback, useMemo } from "react";
|
|
12
|
+
import { symbolInternalRSCExtras } from "./utils/RSCThreadExtras";
|
|
13
|
+
|
|
14
|
+
const vercelToThreadMessage = <T,>(
|
|
15
|
+
converter: (message: T) => VercelRSCMessage,
|
|
16
|
+
rawMessage: T,
|
|
17
|
+
): ThreadMessageLike => {
|
|
18
|
+
const message = converter(rawMessage);
|
|
19
|
+
|
|
20
|
+
return {
|
|
21
|
+
id: message.id,
|
|
22
|
+
role: message.role,
|
|
23
|
+
content: [{ type: "text", text: "[Developer: Please set up RSCDisplay]" }],
|
|
24
|
+
createdAt: message.createdAt,
|
|
25
|
+
};
|
|
26
|
+
};
|
|
27
|
+
|
|
28
|
+
export const useVercelRSCRuntime = <T extends WeakKey>(
|
|
29
|
+
adapter: VercelRSCAdapter<T>,
|
|
30
|
+
) => {
|
|
31
|
+
const onNew = adapter.onNew;
|
|
32
|
+
if (!onNew)
|
|
33
|
+
throw new Error("You must pass a onNew function to useVercelRSCRuntime");
|
|
34
|
+
|
|
35
|
+
const convertFn = useMemo(() => {
|
|
36
|
+
return (
|
|
37
|
+
adapter.convertMessage?.bind(adapter) ?? ((m: T) => m as VercelRSCMessage)
|
|
38
|
+
);
|
|
39
|
+
}, [adapter.convertMessage, adapter]);
|
|
40
|
+
const callback = useCallback(
|
|
41
|
+
(m: T) => {
|
|
42
|
+
return vercelToThreadMessage(convertFn, m);
|
|
43
|
+
},
|
|
44
|
+
[convertFn],
|
|
45
|
+
);
|
|
46
|
+
|
|
47
|
+
const messages = useExternalMessageConverter({
|
|
48
|
+
callback,
|
|
49
|
+
isRunning: adapter.isRunning ?? false,
|
|
50
|
+
messages: adapter.messages,
|
|
51
|
+
});
|
|
52
|
+
|
|
53
|
+
const eAdapter: ExternalStoreAdapter = {
|
|
54
|
+
isRunning: adapter.isRunning,
|
|
55
|
+
messages,
|
|
56
|
+
onNew,
|
|
57
|
+
onEdit: adapter.onEdit,
|
|
58
|
+
onReload: adapter.onReload,
|
|
59
|
+
adapters: adapter.adapters,
|
|
60
|
+
unstable_capabilities: {
|
|
61
|
+
copy: false,
|
|
62
|
+
},
|
|
63
|
+
extras: {
|
|
64
|
+
[symbolInternalRSCExtras]: { convertFn },
|
|
65
|
+
},
|
|
66
|
+
};
|
|
67
|
+
|
|
68
|
+
const runtime = useExternalStoreRuntime(eAdapter);
|
|
69
|
+
return runtime;
|
|
70
|
+
};
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import { VercelRSCMessage } from "../VercelRSCMessage";
|
|
3
|
+
|
|
4
|
+
export const symbolInternalRSCExtras = Symbol("internal-rsc-extras");
|
|
5
|
+
export type RSCThreadExtras =
|
|
6
|
+
| {
|
|
7
|
+
[symbolInternalRSCExtras]?: {
|
|
8
|
+
convertFn: (message: any) => VercelRSCMessage;
|
|
9
|
+
};
|
|
10
|
+
}
|
|
11
|
+
| undefined;
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import {
|
|
2
|
+
getExternalStoreMessage,
|
|
3
|
+
type ThreadMessage,
|
|
4
|
+
} from "@assistant-ui/react";
|
|
5
|
+
import type { Message } from "@ai-sdk/ui-utils";
|
|
6
|
+
|
|
7
|
+
export const getVercelAIMessages = (message: ThreadMessage) => {
|
|
8
|
+
return getExternalStoreMessage(message) as Message[];
|
|
9
|
+
};
|
package/src/ui/index.ts
ADDED
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
export { useVercelUseChatRuntime } from "./use-chat/useVercelUseChatRuntime";
|
|
2
|
+
export { useVercelUseAssistantRuntime } from "./use-assistant/useVercelUseAssistantRuntime";
|
|
3
|
+
export { getVercelAIMessages } from "./getVercelAIMessages";
|
|
4
|
+
export { AISDKMessageConverter } from "./utils/convertMessage";
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import type { useAssistant } from "@ai-sdk/react";
|
|
2
|
+
import { useExternalStoreRuntime } from "@assistant-ui/react";
|
|
3
|
+
import { useInputSync } from "../utils/useInputSync";
|
|
4
|
+
import { toCreateMessage } from "../utils/toCreateMessage";
|
|
5
|
+
import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter";
|
|
6
|
+
import { ExternalStoreAdapter } from "@assistant-ui/react";
|
|
7
|
+
import { AISDKMessageConverter } from "../utils/convertMessage";
|
|
8
|
+
|
|
9
|
+
export type VercelUseChatAdapter = {
|
|
10
|
+
adapters?:
|
|
11
|
+
| Omit<NonNullable<ExternalStoreAdapter["adapters"]>, "attachments">
|
|
12
|
+
| undefined;
|
|
13
|
+
};
|
|
14
|
+
|
|
15
|
+
export const useVercelUseAssistantRuntime = (
|
|
16
|
+
assistantHelpers: ReturnType<typeof useAssistant>,
|
|
17
|
+
adapter: VercelUseChatAdapter = {},
|
|
18
|
+
) => {
|
|
19
|
+
const messages = AISDKMessageConverter.useThreadMessages({
|
|
20
|
+
messages: assistantHelpers.messages,
|
|
21
|
+
isRunning: assistantHelpers.status === "in_progress",
|
|
22
|
+
});
|
|
23
|
+
|
|
24
|
+
const runtime = useExternalStoreRuntime({
|
|
25
|
+
isRunning: assistantHelpers.status === "in_progress",
|
|
26
|
+
messages,
|
|
27
|
+
onCancel: async () => assistantHelpers.stop(),
|
|
28
|
+
onNew: async (message) => {
|
|
29
|
+
await assistantHelpers.append(await toCreateMessage(message));
|
|
30
|
+
},
|
|
31
|
+
adapters: {
|
|
32
|
+
attachments: vercelAttachmentAdapter,
|
|
33
|
+
...adapter.adapters,
|
|
34
|
+
threadList: new Proxy(adapter.adapters?.threadList ?? {}, {
|
|
35
|
+
get(target, prop, receiver) {
|
|
36
|
+
if (prop === "onSwitchToNewThread") {
|
|
37
|
+
return () => {
|
|
38
|
+
assistantHelpers.messages = [];
|
|
39
|
+
assistantHelpers.input = "";
|
|
40
|
+
assistantHelpers.setMessages([]);
|
|
41
|
+
assistantHelpers.setInput("");
|
|
42
|
+
|
|
43
|
+
if (typeof target.onSwitchToNewThread === "function") {
|
|
44
|
+
return target.onSwitchToNewThread.call(target);
|
|
45
|
+
}
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
return Reflect.get(target, prop, receiver);
|
|
50
|
+
},
|
|
51
|
+
}),
|
|
52
|
+
},
|
|
53
|
+
});
|
|
54
|
+
|
|
55
|
+
useInputSync(assistantHelpers, runtime);
|
|
56
|
+
|
|
57
|
+
return runtime;
|
|
58
|
+
};
|
|
@@ -0,0 +1,91 @@
|
|
|
1
|
+
import type { useChat } from "@ai-sdk/react";
|
|
2
|
+
import { useExternalStoreRuntime } from "@assistant-ui/react";
|
|
3
|
+
import { useInputSync } from "../utils/useInputSync";
|
|
4
|
+
import { sliceMessagesUntil } from "../utils/sliceMessagesUntil";
|
|
5
|
+
import { toCreateMessage } from "../utils/toCreateMessage";
|
|
6
|
+
import { vercelAttachmentAdapter } from "../utils/vercelAttachmentAdapter";
|
|
7
|
+
import { getVercelAIMessages } from "../getVercelAIMessages";
|
|
8
|
+
import { ExternalStoreAdapter } from "@assistant-ui/react";
|
|
9
|
+
import { useState } from "react";
|
|
10
|
+
import { generateId } from "@ai-sdk/ui-utils";
|
|
11
|
+
import { AISDKMessageConverter } from "../utils/convertMessage";
|
|
12
|
+
|
|
13
|
+
export type VercelUseChatAdapter = {
|
|
14
|
+
adapters?:
|
|
15
|
+
| Omit<NonNullable<ExternalStoreAdapter["adapters"]>, "attachments">
|
|
16
|
+
| undefined;
|
|
17
|
+
unstable_joinStrategy?: "concat-content" | "none";
|
|
18
|
+
};
|
|
19
|
+
|
|
20
|
+
export const useVercelUseChatRuntime = (
|
|
21
|
+
chatHelpers: ReturnType<typeof useChat>,
|
|
22
|
+
adapter: VercelUseChatAdapter = {},
|
|
23
|
+
) => {
|
|
24
|
+
const messages = AISDKMessageConverter.useThreadMessages({
|
|
25
|
+
isRunning:
|
|
26
|
+
chatHelpers.status === "submitted" || chatHelpers.status == "streaming",
|
|
27
|
+
messages: chatHelpers.messages,
|
|
28
|
+
joinStrategy: adapter.unstable_joinStrategy,
|
|
29
|
+
});
|
|
30
|
+
|
|
31
|
+
const [threadId, setThreadId] = useState<string>(generateId());
|
|
32
|
+
|
|
33
|
+
const runtime = useExternalStoreRuntime({
|
|
34
|
+
isRunning: chatHelpers.isLoading,
|
|
35
|
+
messages,
|
|
36
|
+
setMessages: (messages) =>
|
|
37
|
+
chatHelpers.setMessages(messages.map(getVercelAIMessages).flat()),
|
|
38
|
+
onCancel: async () => chatHelpers.stop(),
|
|
39
|
+
onNew: async (message) => {
|
|
40
|
+
await chatHelpers.append(await toCreateMessage(message));
|
|
41
|
+
},
|
|
42
|
+
onEdit: async (message) => {
|
|
43
|
+
const newMessages = sliceMessagesUntil(
|
|
44
|
+
chatHelpers.messages,
|
|
45
|
+
message.parentId,
|
|
46
|
+
);
|
|
47
|
+
chatHelpers.setMessages(newMessages);
|
|
48
|
+
|
|
49
|
+
await chatHelpers.append(await toCreateMessage(message));
|
|
50
|
+
},
|
|
51
|
+
onReload: async (parentId: string | null) => {
|
|
52
|
+
const newMessages = sliceMessagesUntil(chatHelpers.messages, parentId);
|
|
53
|
+
chatHelpers.setMessages(newMessages);
|
|
54
|
+
|
|
55
|
+
await chatHelpers.reload();
|
|
56
|
+
},
|
|
57
|
+
onAddToolResult: ({ toolCallId, result }) => {
|
|
58
|
+
chatHelpers.addToolResult({ toolCallId, result });
|
|
59
|
+
},
|
|
60
|
+
adapters: {
|
|
61
|
+
attachments: vercelAttachmentAdapter,
|
|
62
|
+
...adapter.adapters,
|
|
63
|
+
threadList: new Proxy(adapter.adapters?.threadList ?? {}, {
|
|
64
|
+
get(target, prop, receiver) {
|
|
65
|
+
if (prop === "threadId") {
|
|
66
|
+
return target.threadId ?? threadId;
|
|
67
|
+
}
|
|
68
|
+
if (prop === "onSwitchToNewThread") {
|
|
69
|
+
return () => {
|
|
70
|
+
chatHelpers.messages = [];
|
|
71
|
+
chatHelpers.input = "";
|
|
72
|
+
chatHelpers.setMessages([]);
|
|
73
|
+
chatHelpers.setInput("");
|
|
74
|
+
setThreadId(generateId());
|
|
75
|
+
|
|
76
|
+
if (typeof target.onSwitchToNewThread === "function") {
|
|
77
|
+
return target.onSwitchToNewThread.call(target);
|
|
78
|
+
}
|
|
79
|
+
};
|
|
80
|
+
}
|
|
81
|
+
|
|
82
|
+
return Reflect.get(target, prop, receiver);
|
|
83
|
+
},
|
|
84
|
+
}),
|
|
85
|
+
},
|
|
86
|
+
});
|
|
87
|
+
|
|
88
|
+
useInputSync(chatHelpers, runtime);
|
|
89
|
+
|
|
90
|
+
return runtime;
|
|
91
|
+
};
|
|
@@ -0,0 +1,158 @@
|
|
|
1
|
+
import { Message } from "@ai-sdk/ui-utils";
|
|
2
|
+
import { unstable_createMessageConverter } from "@assistant-ui/react";
|
|
3
|
+
import { SourceContentPart } from "@assistant-ui/react";
|
|
4
|
+
import { FileContentPart } from "@assistant-ui/react";
|
|
5
|
+
import { ReasoningContentPart } from "@assistant-ui/react";
|
|
6
|
+
import { ToolCallContentPart } from "@assistant-ui/react";
|
|
7
|
+
import { TextContentPart } from "@assistant-ui/react";
|
|
8
|
+
import { CompleteAttachment } from "@assistant-ui/react";
|
|
9
|
+
|
|
10
|
+
export const AISDKMessageConverter = unstable_createMessageConverter(
|
|
11
|
+
(message: Message) => {
|
|
12
|
+
switch (message.role) {
|
|
13
|
+
case "user":
|
|
14
|
+
return {
|
|
15
|
+
role: "user",
|
|
16
|
+
id: message.id,
|
|
17
|
+
createdAt: message.createdAt,
|
|
18
|
+
content: message.content,
|
|
19
|
+
|
|
20
|
+
attachments: message.experimental_attachments?.map(
|
|
21
|
+
(attachment, idx) =>
|
|
22
|
+
({
|
|
23
|
+
id: idx.toString(),
|
|
24
|
+
type: "file",
|
|
25
|
+
name: attachment.name ?? attachment.url,
|
|
26
|
+
content: [],
|
|
27
|
+
contentType: attachment.contentType ?? "unknown/unknown",
|
|
28
|
+
status: { type: "complete" },
|
|
29
|
+
}) satisfies CompleteAttachment,
|
|
30
|
+
),
|
|
31
|
+
};
|
|
32
|
+
|
|
33
|
+
case "system":
|
|
34
|
+
return {
|
|
35
|
+
role: "system",
|
|
36
|
+
id: message.id,
|
|
37
|
+
createdAt: message.createdAt,
|
|
38
|
+
content: message.content,
|
|
39
|
+
};
|
|
40
|
+
|
|
41
|
+
case "assistant":
|
|
42
|
+
return {
|
|
43
|
+
role: "assistant",
|
|
44
|
+
id: message.id,
|
|
45
|
+
createdAt: message.createdAt,
|
|
46
|
+
content:
|
|
47
|
+
message.parts
|
|
48
|
+
?.filter((p) => p.type !== "step-start")
|
|
49
|
+
.map((part) => {
|
|
50
|
+
const type = part.type;
|
|
51
|
+
switch (type) {
|
|
52
|
+
case "text":
|
|
53
|
+
return {
|
|
54
|
+
type: "text",
|
|
55
|
+
text: part.text,
|
|
56
|
+
} satisfies TextContentPart;
|
|
57
|
+
case "tool-invocation":
|
|
58
|
+
return {
|
|
59
|
+
type: "tool-call",
|
|
60
|
+
toolName: part.toolInvocation.toolName,
|
|
61
|
+
toolCallId: part.toolInvocation.toolCallId,
|
|
62
|
+
argsText: JSON.stringify(part.toolInvocation.args),
|
|
63
|
+
args: part.toolInvocation.args,
|
|
64
|
+
result:
|
|
65
|
+
part.toolInvocation.state === "result" &&
|
|
66
|
+
part.toolInvocation.result,
|
|
67
|
+
} satisfies ToolCallContentPart;
|
|
68
|
+
case "reasoning":
|
|
69
|
+
return {
|
|
70
|
+
type: "reasoning",
|
|
71
|
+
text: part.reasoning,
|
|
72
|
+
} satisfies ReasoningContentPart;
|
|
73
|
+
|
|
74
|
+
case "source":
|
|
75
|
+
return {
|
|
76
|
+
type: "source",
|
|
77
|
+
...part.source,
|
|
78
|
+
} satisfies SourceContentPart;
|
|
79
|
+
|
|
80
|
+
case "file":
|
|
81
|
+
return {
|
|
82
|
+
type: "file",
|
|
83
|
+
data: part.data,
|
|
84
|
+
mimeType: part.mimeType,
|
|
85
|
+
} satisfies FileContentPart;
|
|
86
|
+
|
|
87
|
+
default: {
|
|
88
|
+
const _unsupported: never = type;
|
|
89
|
+
throw new Error(
|
|
90
|
+
`You have a message with an unsupported part type. The type ${_unsupported} is not supported.`,
|
|
91
|
+
);
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
}) ??
|
|
95
|
+
(message.content
|
|
96
|
+
? [
|
|
97
|
+
{
|
|
98
|
+
type: "text",
|
|
99
|
+
text: message.content,
|
|
100
|
+
} satisfies TextContentPart,
|
|
101
|
+
]
|
|
102
|
+
: []),
|
|
103
|
+
metadata: {
|
|
104
|
+
unstable_annotations: message.annotations,
|
|
105
|
+
unstable_data: Array.isArray(message.data)
|
|
106
|
+
? message.data
|
|
107
|
+
: message.data
|
|
108
|
+
? [message.data]
|
|
109
|
+
: undefined,
|
|
110
|
+
},
|
|
111
|
+
};
|
|
112
|
+
|
|
113
|
+
case "data": {
|
|
114
|
+
type MaybeSupportedDataMessage =
|
|
115
|
+
| { type?: "unsafe_other" }
|
|
116
|
+
| ToolCallContentPart
|
|
117
|
+
| {
|
|
118
|
+
type: "tool-result";
|
|
119
|
+
toolCallId: string;
|
|
120
|
+
result: any;
|
|
121
|
+
};
|
|
122
|
+
|
|
123
|
+
if (
|
|
124
|
+
!message.data ||
|
|
125
|
+
!(typeof message.data === "object") ||
|
|
126
|
+
Array.isArray(message.data)
|
|
127
|
+
)
|
|
128
|
+
return [];
|
|
129
|
+
|
|
130
|
+
const data = message.data as MaybeSupportedDataMessage;
|
|
131
|
+
|
|
132
|
+
if (data.type === "tool-call") {
|
|
133
|
+
return {
|
|
134
|
+
role: "assistant",
|
|
135
|
+
id: message.id,
|
|
136
|
+
createdAt: message.createdAt,
|
|
137
|
+
content: [data],
|
|
138
|
+
};
|
|
139
|
+
} else if (data.type === "tool-result") {
|
|
140
|
+
return {
|
|
141
|
+
role: "tool",
|
|
142
|
+
id: message.id,
|
|
143
|
+
toolCallId: data.toolCallId,
|
|
144
|
+
result: data.result,
|
|
145
|
+
};
|
|
146
|
+
}
|
|
147
|
+
return [];
|
|
148
|
+
}
|
|
149
|
+
|
|
150
|
+
default:
|
|
151
|
+
// TODO handle tool and function messages
|
|
152
|
+
const _unsupported: "function" | "tool" = message.role;
|
|
153
|
+
throw new Error(
|
|
154
|
+
`You have a message with an unsupported role. The role ${_unsupported} is not supported.`,
|
|
155
|
+
);
|
|
156
|
+
}
|
|
157
|
+
},
|
|
158
|
+
);
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import type { Message } from "@ai-sdk/ui-utils";
|
|
2
|
+
|
|
3
|
+
export const sliceMessagesUntil = (
|
|
4
|
+
messages: Message[],
|
|
5
|
+
messageId: string | null,
|
|
6
|
+
) => {
|
|
7
|
+
if (messageId == null) return [];
|
|
8
|
+
|
|
9
|
+
let messageIdx = messages.findIndex((m) => m.id === messageId);
|
|
10
|
+
if (messageIdx === -1)
|
|
11
|
+
throw new Error(
|
|
12
|
+
"useVercelAIThreadState: Message not found. This is liekly an internal bug in assistant-ui.",
|
|
13
|
+
);
|
|
14
|
+
|
|
15
|
+
while (messages[messageIdx + 1]?.role === "assistant") {
|
|
16
|
+
messageIdx++;
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
return messages.slice(0, messageIdx + 1);
|
|
20
|
+
};
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import { AppendMessage } from "@assistant-ui/react";
|
|
2
|
+
import { CreateMessage } from "@ai-sdk/ui-utils";
|
|
3
|
+
|
|
4
|
+
export const toCreateMessage = async (
|
|
5
|
+
message: AppendMessage,
|
|
6
|
+
): Promise<CreateMessage> => {
|
|
7
|
+
const content = message.content
|
|
8
|
+
.filter((part) => part.type === "text")
|
|
9
|
+
.map((t) => t.text)
|
|
10
|
+
.join("\n\n");
|
|
11
|
+
|
|
12
|
+
const images = message.content
|
|
13
|
+
.filter((part) => part.type === "image")
|
|
14
|
+
.map((part) => ({ url: part.image }));
|
|
15
|
+
|
|
16
|
+
return {
|
|
17
|
+
role: message.role,
|
|
18
|
+
content,
|
|
19
|
+
experimental_attachments: [
|
|
20
|
+
...images,
|
|
21
|
+
...(await Promise.all(
|
|
22
|
+
(message.attachments ?? []).map(async (m) => {
|
|
23
|
+
if (m.file == null)
|
|
24
|
+
throw new Error("Attachment did not contain a file");
|
|
25
|
+
return {
|
|
26
|
+
contentType: m.file.type,
|
|
27
|
+
name: m.file.name,
|
|
28
|
+
url: await getFileDataURL(m.file),
|
|
29
|
+
};
|
|
30
|
+
}),
|
|
31
|
+
)),
|
|
32
|
+
],
|
|
33
|
+
};
|
|
34
|
+
};
|
|
35
|
+
|
|
36
|
+
const getFileDataURL = (file: File) =>
|
|
37
|
+
new Promise<string>((resolve, reject) => {
|
|
38
|
+
const reader = new FileReader();
|
|
39
|
+
|
|
40
|
+
reader.onload = () => resolve(reader.result as string);
|
|
41
|
+
reader.onerror = (error) => reject(error);
|
|
42
|
+
|
|
43
|
+
reader.readAsDataURL(file);
|
|
44
|
+
});
|
|
@@ -0,0 +1,24 @@
|
|
|
1
|
+
import { useEffect } from "react";
|
|
2
|
+
import { useAssistant, useChat } from "@ai-sdk/react";
|
|
3
|
+
import { AssistantRuntime } from "@assistant-ui/react";
|
|
4
|
+
|
|
5
|
+
type VercelHelpers =
|
|
6
|
+
| ReturnType<typeof useChat>
|
|
7
|
+
| ReturnType<typeof useAssistant>;
|
|
8
|
+
|
|
9
|
+
export const useInputSync = (
|
|
10
|
+
{ setInput, input }: VercelHelpers,
|
|
11
|
+
runtime: AssistantRuntime,
|
|
12
|
+
) => {
|
|
13
|
+
// sync input from vercel to assistant-ui
|
|
14
|
+
useEffect(() => {
|
|
15
|
+
runtime.thread.composer.setText(input);
|
|
16
|
+
}, [runtime, input]);
|
|
17
|
+
|
|
18
|
+
// sync input from assistant-ui to vercel
|
|
19
|
+
useEffect(() => {
|
|
20
|
+
return runtime.thread.composer.subscribe(() => {
|
|
21
|
+
setInput(runtime.thread.composer.getState().text);
|
|
22
|
+
});
|
|
23
|
+
}, [runtime, setInput]);
|
|
24
|
+
};
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import { AttachmentAdapter } from "@assistant-ui/react";
|
|
2
|
+
import { generateId } from "@ai-sdk/ui-utils";
|
|
3
|
+
|
|
4
|
+
export const vercelAttachmentAdapter: AttachmentAdapter = {
|
|
5
|
+
accept:
|
|
6
|
+
"image/*, text/plain, text/html, text/markdown, text/csv, text/xml, text/json, text/css",
|
|
7
|
+
async add({ file }) {
|
|
8
|
+
return {
|
|
9
|
+
id: generateId(),
|
|
10
|
+
type: "file",
|
|
11
|
+
name: file.name,
|
|
12
|
+
file,
|
|
13
|
+
contentType: file.type,
|
|
14
|
+
content: [],
|
|
15
|
+
status: { type: "requires-action", reason: "composer-send" },
|
|
16
|
+
};
|
|
17
|
+
},
|
|
18
|
+
async send(attachment) {
|
|
19
|
+
// noop
|
|
20
|
+
return {
|
|
21
|
+
...attachment,
|
|
22
|
+
status: { type: "complete" },
|
|
23
|
+
content: [],
|
|
24
|
+
};
|
|
25
|
+
},
|
|
26
|
+
async remove() {
|
|
27
|
+
// noop
|
|
28
|
+
},
|
|
29
|
+
};
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import { EdgeRuntimeOptions, useEdgeRuntime } from "@assistant-ui/react-edge";
|
|
2
|
+
|
|
3
|
+
type UseChatRuntimeOptions = Omit<EdgeRuntimeOptions, "unstable_AISDKInterop">;
|
|
4
|
+
|
|
5
|
+
export const useChatRuntime = (options: UseChatRuntimeOptions) => {
|
|
6
|
+
return useEdgeRuntime({
|
|
7
|
+
...options,
|
|
8
|
+
unstable_AISDKInterop: "v2",
|
|
9
|
+
});
|
|
10
|
+
};
|