@assistant-ui/react-ai-sdk 1.3.20 → 1.3.23
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.
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAQzE,OAAO,KAAK,EACV,oBAAoB,EACpB,oBAAoB,EACpB,gBAAgB,EAKhB,aAAa,EAEd,MAAM,oBAAoB,CAAC;AAiB5B,MAAM,MAAM,6BAA6B,GAAG,CAC1C,UAAU,SAAS,SAAS,GAAG,SAAS,EAExC,OAAO,EAAE,aAAa,KACnB,eAAe,CAAC,UAAU,CAAC,CAAC;
|
|
1
|
+
{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAQzE,OAAO,KAAK,EACV,oBAAoB,EACpB,oBAAoB,EACpB,gBAAgB,EAKhB,aAAa,EAEd,MAAM,oBAAoB,CAAC;AAiB5B,MAAM,MAAM,6BAA6B,GAAG,CAC1C,UAAU,SAAS,SAAS,GAAG,SAAS,EAExC,OAAO,EAAE,aAAa,KACnB,eAAe,CAAC,UAAU,CAAC,CAAC;AAYjC,MAAM,MAAM,mBAAmB,GAAG;IAChC,QAAQ,CAAC,EACL,CAAC,WAAW,CAAC,oBAAoB,CAAC,UAAU,CAAC,CAAC,GAAG;QAC/C,OAAO,CAAC,EAAE,oBAAoB,GAAG,SAAS,CAAC;KAC5C,CAAC,GACF,SAAS,CAAC;IACd,eAAe,CAAC,EAAE,6BAA6B,CAAC;IAChD;;;;;;;OAOG;IACH,4BAA4B,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;CACpD,CAAC;AAEF,eAAO,MAAM,eAAe,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EACtE,aAAa,UAAU,CAAC,OAAO,OAAO,CAAC,UAAU,CAAC,CAAC,EACnD,sFAIG,mBAAwB,qBA4P5B,CAAC"}
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
"use client";
|
|
2
2
|
import { useState, useMemo, useRef } from "react";
|
|
3
|
-
import { isToolUIPart } from "ai";
|
|
3
|
+
import { isToolUIPart, generateId } from "ai";
|
|
4
4
|
import { useExternalStoreRuntime, useRuntimeAdapters, useToolInvocations, } from "@assistant-ui/core/react";
|
|
5
5
|
import { getExternalStoreMessages } from "@assistant-ui/core";
|
|
6
6
|
import { sliceMessagesUntil } from "../utils/sliceMessagesUntil.js";
|
|
@@ -11,6 +11,11 @@ import { AISDKMessageConverter } from "../utils/convertMessage.js";
|
|
|
11
11
|
import { aiSDKV6FormatAdapter, } from "../adapters/aiSDKFormatAdapter.js";
|
|
12
12
|
import { useExternalHistory, toExportedMessageRepository, } from "./useExternalHistory.js";
|
|
13
13
|
import { useStreamingTiming } from "./useStreamingTiming.js";
|
|
14
|
+
const toUIMessage = (createMessage, fallbackRole) => ({
|
|
15
|
+
...createMessage,
|
|
16
|
+
id: createMessage.id ?? generateId(),
|
|
17
|
+
role: createMessage.role ?? fallbackRole,
|
|
18
|
+
});
|
|
14
19
|
export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, } = {}) => {
|
|
15
20
|
const contextAdapters = useRuntimeAdapters();
|
|
16
21
|
const [toolStatuses, setToolStatuses] = useState({});
|
|
@@ -131,18 +136,31 @@ export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: custom
|
|
|
131
136
|
await toolInvocations.abort();
|
|
132
137
|
},
|
|
133
138
|
onNew: async (message) => {
|
|
139
|
+
const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
|
|
140
|
+
if (!(message.startRun ?? message.role === "user")) {
|
|
141
|
+
chatHelpers.setMessages((current) => [
|
|
142
|
+
...current,
|
|
143
|
+
toUIMessage(createMessage, message.role),
|
|
144
|
+
]);
|
|
145
|
+
return;
|
|
146
|
+
}
|
|
134
147
|
lastRunConfigRef.current = message.runConfig;
|
|
135
148
|
await completePendingToolCalls();
|
|
136
|
-
const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
|
|
137
149
|
await chatHelpers.sendMessage(createMessage, {
|
|
138
150
|
metadata: message.runConfig,
|
|
139
151
|
});
|
|
140
152
|
},
|
|
141
153
|
onEdit: async (message) => {
|
|
142
|
-
lastRunConfigRef.current = message.runConfig;
|
|
143
|
-
const newMessages = sliceMessagesUntil(chatHelpers.messages, message.parentId);
|
|
144
|
-
chatHelpers.setMessages(newMessages);
|
|
145
154
|
const createMessage = (customToCreateMessage ?? toCreateMessage)(message);
|
|
155
|
+
if (!(message.startRun ?? message.role === "user")) {
|
|
156
|
+
chatHelpers.setMessages((current) => [
|
|
157
|
+
...sliceMessagesUntil(current, message.parentId),
|
|
158
|
+
toUIMessage(createMessage, message.role),
|
|
159
|
+
]);
|
|
160
|
+
return;
|
|
161
|
+
}
|
|
162
|
+
lastRunConfigRef.current = message.runConfig;
|
|
163
|
+
chatHelpers.setMessages((current) => sliceMessagesUntil(current, message.parentId));
|
|
146
164
|
await chatHelpers.sendMessage(createMessage, {
|
|
147
165
|
metadata: message.runConfig,
|
|
148
166
|
});
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useAISDKRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAElD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;
|
|
1
|
+
{"version":3,"file":"useAISDKRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AAElD,OAAO,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAC9C,OAAO,EACL,uBAAuB,EACvB,kBAAkB,EAClB,kBAAkB,GAEnB,MAAM,0BAA0B,CAAC;AAYlC,OAAO,EAAE,wBAAwB,EAAE,MAAM,oBAAoB,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,uCAAoC;AACjE,OAAO,EAAE,eAAe,EAAE,oCAAiC;AAC3D,OAAO,EAAE,uBAAuB,EAAE,4CAAyC;AAC3E,OAAO,EAAE,mBAAmB,EAAE,kCAA+B;AAC7D,OAAO,EAAE,qBAAqB,EAAE,mCAAgC;AAChE,OAAO,EAEL,oBAAoB,GACrB,0CAAuC;AACxC,OAAO,EACL,kBAAkB,EAClB,2BAA2B,GAC5B,gCAA6B;AAC9B,OAAO,EAAE,kBAAkB,EAAE,gCAA6B;AAQ1D,MAAM,WAAW,GAAG,CAClB,aAA0C,EAC1C,YAAgC,EACpB,EAAE,CACd,CAAC;IACC,GAAG,aAAa;IAChB,EAAE,EAAE,aAAa,CAAC,EAAE,IAAI,UAAU,EAAE;IACpC,IAAI,EAAE,aAAa,CAAC,IAAI,IAAI,YAAY;CACzC,CAAe,CAAC;AAoBnB,MAAM,CAAC,MAAM,eAAe,GAAG,CAC7B,WAAmD,EACnD,EACE,QAAQ,EACR,eAAe,EAAE,qBAAqB,EACtC,4BAA4B,GAAG,IAAI,MACZ,EAAE,EAC3B,EAAE;IACF,MAAM,eAAe,GAAG,kBAAkB,EAAE,CAAC;IAC7C,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAE9C,EAAE,CAAC,CAAC;IACN,MAAM,wBAAwB,GAAG,MAAM,CACrC,IAAI,GAAG,EAAE,CACV,CAAC;IACF,MAAM,gBAAgB,GAAG,MAAM,CAAwB,SAAS,CAAC,CAAC;IAElE,MAAM,iBAAiB,GAAG,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,IAAI,CACxD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,KAAK,WAAW,CAC/B,CAAC;IACF,MAAM,SAAS,GACb,WAAW,CAAC,MAAM,KAAK,WAAW;QAClC,WAAW,CAAC,MAAM,KAAK,WAAW;QAClC,iBAAiB,CAAC;IAEpB,MAAM,aAAa,GAAG,kBAAkB,CAAC,WAAW,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IAE1E,MAAM,QAAQ,GAAG,qBAAqB,CAAC,iBAAiB,CAAC;QACvD,SAAS;QACT,QAAQ,EAAE,WAAW,CAAC,QAAQ;QAC9B,QAAQ,EAAE,OAAO,CACf,GAAG,EAAE,CAAC,CAAC;YACL,YAAY;YACZ,aAAa;YACb,qBAAqB,EAAE,wBAAwB,CAAC,OAAO;YACvD,GAAG,CAAC,WAAW,CAAC,KAAK,IAAI,EAAE,KAAK,EAAE,WAAW,CAAC,KAAK,CAAC,OAAO,EAAE,CAAC;SAC/D,CAAC,EACF,CAAC,YAAY,EAAE,aAAa,EAAE,WAAW,CAAC,KAAK,CAAC,CACjD;KACF,CAAC,CAAC;IAEH,MAAM,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC;QACnC,IAAI,OAAO;YACT,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC,CAAC,CAAC;IAEJ,MAAM,eAAe,GAAG,kBAAkB,CAAC;QACzC,KAAK,EAAE;YACL,QAAQ;YACR,SAAS;SACV;QACD,QAAQ,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,eAAe,EAAE,CAAC,KAAK;QACjE,QAAQ,EAAE,CAAC,OAAO,EAAE,EAAE;YACpB,IAAI,OAAO,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;gBACvC,WAAW,CAAC,aAAa,CAAC;oBACxB,IAAI,EAAE,OAAO,CAAC,QAAQ;oBACtB,UAAU,EAAE,OAAO,CAAC,UAAU;oBAC9B,MAAM,EAAE,OAAO,CAAC,MAAM;oBACtB,OAAO,EAAE,EAAE,QAAQ,EAAE,gBAAgB,CAAC,OAAO,EAAE;iBAChD,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,eAAe;KAChB,CAAC,CAAC;IAEH,MAAM,SAAS,GAAG,kBAAkB,CAClC,UAAU,EACV,QAAQ,EAAE,OAAO,IAAI,eAAe,EAAE,OAAO,EAC7C,qBAAqB,CAAC,gBAEF,EACpB,oBAGC,EACD,CAAC,QAAQ,EAAE,EAAE;QACX,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC,CACF,CAAC;IAEF,MAAM,wBAAwB,GAAG,KAAK,IAAI,EAAE;QAC1C,IAAI,CAAC,4BAA4B;YAAE,OAAO;QAE1C,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;QAE9B,2GAA2G;QAC3G,WAAW,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE,EAAE;YACnC,MAAM,WAAW,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;YACpC,IAAI,WAAW,EAAE,IAAI,KAAK,WAAW;gBAAE,OAAO,QAAQ,CAAC;YAEvD,IAAI,UAAU,GAAG,KAAK,CAAC;YACvB,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;gBAC5C,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC;oBAAE,OAAO,IAAI,CAAC;gBACrC,IAAI,IAAI,CAAC,KAAK,KAAK,kBAAkB,IAAI,IAAI,CAAC,KAAK,KAAK,cAAc;oBACpE,OAAO,IAAI,CAAC;gBAEd,UAAU,GAAG,IAAI,CAAC;gBAClB,OAAO;oBACL,GAAG,IAAI;oBACP,KAAK,EAAE,cAAuB;oBAC9B,SAAS,EAAE,oDAAoD;iBAChE,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,IAAI,CAAC,UAAU;gBAAE,OAAO,QAAQ,CAAC;YACjC,OAAO,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,GAAG,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAC/D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,MAAM,OAAO,GAAG,uBAAuB,CAAC;QACtC,SAAS;QACT,QAAQ;QACR,WAAW,EAAE,CAAC,QAAQ,EAAE,EAAE,CACxB,WAAW,CAAC,WAAW,CACrB,QAAQ;aACL,GAAG,CAAC,CAAA,mBAA+B,CAAA,CAAC;aACpC,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,EAAE,CACV;QACH,QAAQ,EAAE,CAAC,QAAQ,EAAE,EAAE,CACrB,WAAW,CAAC,WAAW,CACrB,QAAQ;aACL,GAAG,CAAC,CAAA,mBAA+B,CAAA,CAAC;aACpC,MAAM,CAAC,OAAO,CAAC;aACf,IAAI,EAAE,CACV;QACH,qBAAqB,EAAE,GAAwC,EAAE;YAC/D,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YAEpD,MAAM,gBAAgB,GAAoC,EAAE,CAAC;YAC7D,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC;YAEjD,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,QAAQ,EAAE,CAAC;gBACrC,MAAM,aAAa,GAAG,wBAAwB,CAC5C,IAAI,CAAC,OAAO,CACb,CAAC;gBACF,IAAI,QAAQ,GACV,IAAI,CAAC,QAAQ,IAAI,IAAI;oBACnB,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC;oBACtD,CAAC,CAAC,IAAI,CAAC;gBACX,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;oBACzC,gBAAgB,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;oBAC3D,QAAQ,GAAG,oBAAoB,CAAC,KAAK,CAAC,YAAyB,CAAC,CAAC;gBACnE,CAAC;gBACD,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC7B,cAAc,CAAC,GAAG,CAChB,IAAI,CAAC,OAAO,CAAC,EAAE,EACf,oBAAoB,CAAC,KAAK,CACxB,aAAa,CAAC,aAAa,CAAC,MAAM,GAAG,CAAC,CAAe,CACtD,CACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,MAAM,GAAwC;gBAClD,QAAQ,EAAE,gBAAgB;aAC3B,CAAC;YAEF,IAAI,QAAQ,CAAC,MAAM,IAAI,IAAI,EAAE,CAAC;gBAC5B,MAAM,CAAC,MAAM,GAAG,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC;YACzE,CAAC;YAED,OAAO,MAAM,CAAC;QAChB,CAAC;QACD,mBAAmB,EAAE,CAAC,IAAyC,EAAE,EAAE;YACjE,+DAA+D;YAC/D,MAAM,YAAY,GAAG,2BAA2B,CAC9C,qBAAqB,CAAC,gBAAgB,EACtC,IAAI,CACL,CAAC;YAEF,6CAA6C;YAC7C,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,QAAQ,EAAE,KAAK,IAAI,EAAE;YACnB,WAAW,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;QAChC,CAAC;QACD,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACvB,MAAM,aAAa,GAAG,CACpB,qBAAqB,IAAI,eAAe,CACzC,CAAa,OAAO,CAAC,CAAC;YAEvB,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;gBACnD,WAAW,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;oBACnC,GAAG,OAAO;oBACV,WAAW,CAAa,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC;iBACrD,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,gBAAgB,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;YAC7C,MAAM,wBAAwB,EAAE,CAAC;YACjC,MAAM,WAAW,CAAC,WAAW,CAAC,aAAa,EAAE;gBAC3C,QAAQ,EAAE,OAAO,CAAC,SAAS;aAC5B,CAAC,CAAC;QACL,CAAC;QACD,MAAM,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACxB,MAAM,aAAa,GAAG,CACpB,qBAAqB,IAAI,eAAe,CACzC,CAAa,OAAO,CAAC,CAAC;YAEvB,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,CAAC;gBACnD,WAAW,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;oBACnC,GAAG,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC;oBAChD,WAAW,CAAa,aAAa,EAAE,OAAO,CAAC,IAAI,CAAC;iBACrD,CAAC,CAAC;gBACH,OAAO;YACT,CAAC;YAED,gBAAgB,CAAC,OAAO,GAAG,OAAO,CAAC,SAAS,CAAC;YAC7C,WAAW,CAAC,WAAW,CAAC,CAAC,OAAO,EAAE,EAAE,CAClC,kBAAkB,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAC9C,CAAC;YACF,MAAM,WAAW,CAAC,WAAW,CAAC,aAAa,EAAE;gBAC3C,QAAQ,EAAE,OAAO,CAAC,SAAS;aAC5B,CAAC,CAAC;QACL,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,QAAuB,EAAE,MAAM,EAAE,EAAE;YAClD,gBAAgB,CAAC,OAAO,GAAG,MAAM,CAAC,SAAS,CAAC;YAC5C,MAAM,WAAW,GAAG,kBAAkB,CAAC,WAAW,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;YACvE,WAAW,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YAErC,MAAM,WAAW,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;QAC/D,CAAC;QACD,eAAe,EAAE,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE;YACnD,MAAM,OAAO,GAAG,EAAE,QAAQ,EAAE,gBAAgB,CAAC,OAAO,EAAE,CAAC;YACvD,IAAI,OAAO,EAAE,CAAC;gBACZ,WAAW,CAAC,aAAa,CAAC;oBACxB,KAAK,EAAE,cAAc;oBACrB,IAAI,EAAE,UAAU;oBAChB,UAAU;oBACV,SAAS,EACP,OAAO,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;oBAC9D,OAAO;iBACR,CAAC,CAAC;YACL,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,aAAa,CAAC;oBACxB,KAAK,EAAE,kBAAkB;oBACzB,IAAI,EAAE,UAAU;oBAChB,UAAU;oBACV,MAAM,EAAE,MAAM;oBACd,OAAO;iBACR,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,gBAAgB,EAAE,CAAC,OAAO,EAAE,EAAE,CAC5B,eAAe,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,OAAO,CAAC;QAC7D,QAAQ,EAAE;YACR,WAAW,EAAE,uBAAuB;YACpC,GAAG,eAAe;YAClB,GAAG,QAAQ;SACZ;QACD,SAAS;KACV,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@assistant-ui/react-ai-sdk",
|
|
3
|
-
"version": "1.3.
|
|
3
|
+
"version": "1.3.23",
|
|
4
4
|
"description": "Vercel AI SDK adapter for assistant-ui",
|
|
5
5
|
"keywords": [
|
|
6
6
|
"ai-sdk",
|
|
@@ -30,10 +30,10 @@
|
|
|
30
30
|
],
|
|
31
31
|
"sideEffects": false,
|
|
32
32
|
"dependencies": {
|
|
33
|
-
"@ai-sdk/react": "^3.0.
|
|
34
|
-
"@assistant-ui/core": "^0.
|
|
35
|
-
"@assistant-ui/store": "^0.2.
|
|
36
|
-
"ai": "^6.0.
|
|
33
|
+
"@ai-sdk/react": "^3.0.177",
|
|
34
|
+
"@assistant-ui/core": "^0.2.0",
|
|
35
|
+
"@assistant-ui/store": "^0.2.10",
|
|
36
|
+
"ai": "^6.0.175",
|
|
37
37
|
"assistant-cloud": "*"
|
|
38
38
|
},
|
|
39
39
|
"peerDependencies": {
|
|
@@ -50,11 +50,11 @@
|
|
|
50
50
|
"@types/json-schema": "^7.0.15",
|
|
51
51
|
"@types/react": "^19.2.14",
|
|
52
52
|
"@types/react-dom": "^19.2.3",
|
|
53
|
-
"jsdom": "^29.
|
|
53
|
+
"jsdom": "^29.1.1",
|
|
54
54
|
"react": "^19.2.5",
|
|
55
55
|
"vitest": "^4.1.5",
|
|
56
|
-
"@assistant-ui/x-buildutils": "0.0.
|
|
57
|
-
"assistant-stream": "0.3.
|
|
56
|
+
"@assistant-ui/x-buildutils": "0.0.7",
|
|
57
|
+
"assistant-stream": "0.3.13"
|
|
58
58
|
},
|
|
59
59
|
"publishConfig": {
|
|
60
60
|
"access": "public",
|
|
@@ -140,6 +140,88 @@ describe("useAISDKRuntime", () => {
|
|
|
140
140
|
expect(chat.messages[0].parts[1].state).toBe("output-available");
|
|
141
141
|
});
|
|
142
142
|
|
|
143
|
+
it("appends a new user message without sending when startRun is false", async () => {
|
|
144
|
+
const chat = createChatHelpers([
|
|
145
|
+
{ id: "u1", role: "user", parts: [{ type: "text", text: "earlier" }] },
|
|
146
|
+
]);
|
|
147
|
+
|
|
148
|
+
const { result } = renderHook(() => useAISDKRuntime(chat));
|
|
149
|
+
|
|
150
|
+
await waitFor(() => {
|
|
151
|
+
expect(result.current.thread.getState().messages.length).toBe(1);
|
|
152
|
+
});
|
|
153
|
+
|
|
154
|
+
act(() => {
|
|
155
|
+
result.current.thread.append({
|
|
156
|
+
role: "user",
|
|
157
|
+
content: [{ type: "text", text: "hold this" }],
|
|
158
|
+
startRun: false,
|
|
159
|
+
});
|
|
160
|
+
});
|
|
161
|
+
|
|
162
|
+
await waitFor(() => {
|
|
163
|
+
expect(chat.setMessages).toHaveBeenCalled();
|
|
164
|
+
});
|
|
165
|
+
|
|
166
|
+
expect(chat.sendMessage).not.toHaveBeenCalled();
|
|
167
|
+
expect(chat.messages).toHaveLength(2);
|
|
168
|
+
expect(chat.messages[1]).toEqual(
|
|
169
|
+
expect.objectContaining({
|
|
170
|
+
role: "user",
|
|
171
|
+
id: expect.any(String),
|
|
172
|
+
parts: expect.arrayContaining([
|
|
173
|
+
expect.objectContaining({ type: "text", text: "hold this" }),
|
|
174
|
+
]),
|
|
175
|
+
}),
|
|
176
|
+
);
|
|
177
|
+
});
|
|
178
|
+
|
|
179
|
+
it("edits without sending when startRun is false", async () => {
|
|
180
|
+
const chat = createChatHelpers([
|
|
181
|
+
{ id: "u1", role: "user", parts: [{ type: "text", text: "first" }] },
|
|
182
|
+
{
|
|
183
|
+
id: "a1",
|
|
184
|
+
role: "assistant",
|
|
185
|
+
parts: [{ type: "text", text: "first-answer" }],
|
|
186
|
+
},
|
|
187
|
+
{ id: "u2", role: "user", parts: [{ type: "text", text: "second" }] },
|
|
188
|
+
]);
|
|
189
|
+
|
|
190
|
+
const { result } = renderHook(() => useAISDKRuntime(chat));
|
|
191
|
+
|
|
192
|
+
await waitFor(() => {
|
|
193
|
+
expect(result.current.thread.getState().messages.length).toBe(3);
|
|
194
|
+
});
|
|
195
|
+
|
|
196
|
+
act(() => {
|
|
197
|
+
result.current.thread.append({
|
|
198
|
+
role: "user",
|
|
199
|
+
parentId: "u1",
|
|
200
|
+
content: [{ type: "text", text: "rewrite, no run" }],
|
|
201
|
+
startRun: false,
|
|
202
|
+
});
|
|
203
|
+
});
|
|
204
|
+
|
|
205
|
+
await waitFor(() => {
|
|
206
|
+
expect(chat.setMessages).toHaveBeenCalled();
|
|
207
|
+
});
|
|
208
|
+
|
|
209
|
+
expect(chat.sendMessage).not.toHaveBeenCalled();
|
|
210
|
+
expect(chat.messages.map((m: any) => m.id)).toEqual([
|
|
211
|
+
"u1",
|
|
212
|
+
"a1",
|
|
213
|
+
expect.any(String),
|
|
214
|
+
]);
|
|
215
|
+
expect(chat.messages[2]).toEqual(
|
|
216
|
+
expect.objectContaining({
|
|
217
|
+
role: "user",
|
|
218
|
+
parts: expect.arrayContaining([
|
|
219
|
+
expect.objectContaining({ type: "text", text: "rewrite, no run" }),
|
|
220
|
+
]),
|
|
221
|
+
}),
|
|
222
|
+
);
|
|
223
|
+
});
|
|
224
|
+
|
|
143
225
|
it("edit slices history to parentId and sends the edited message", async () => {
|
|
144
226
|
const chat = createChatHelpers([
|
|
145
227
|
{ id: "u1", role: "user", parts: [{ type: "text", text: "first" }] },
|
|
@@ -2,7 +2,7 @@
|
|
|
2
2
|
|
|
3
3
|
import { useState, useMemo, useRef } from "react";
|
|
4
4
|
import type { UIMessage, useChat, CreateUIMessage } from "@ai-sdk/react";
|
|
5
|
-
import { isToolUIPart } from "ai";
|
|
5
|
+
import { isToolUIPart, generateId } from "ai";
|
|
6
6
|
import {
|
|
7
7
|
useExternalStoreRuntime,
|
|
8
8
|
useRuntimeAdapters,
|
|
@@ -42,6 +42,16 @@ export type CustomToCreateMessageFunction = <
|
|
|
42
42
|
message: AppendMessage,
|
|
43
43
|
) => CreateUIMessage<UI_MESSAGE>;
|
|
44
44
|
|
|
45
|
+
const toUIMessage = <UI_MESSAGE extends UIMessage>(
|
|
46
|
+
createMessage: CreateUIMessage<UI_MESSAGE>,
|
|
47
|
+
fallbackRole: UI_MESSAGE["role"],
|
|
48
|
+
): UI_MESSAGE =>
|
|
49
|
+
({
|
|
50
|
+
...createMessage,
|
|
51
|
+
id: createMessage.id ?? generateId(),
|
|
52
|
+
role: createMessage.role ?? fallbackRole,
|
|
53
|
+
}) as UI_MESSAGE;
|
|
54
|
+
|
|
45
55
|
export type AISDKRuntimeAdapter = {
|
|
46
56
|
adapters?:
|
|
47
57
|
| (NonNullable<ExternalStoreAdapter["adapters"]> & {
|
|
@@ -240,27 +250,41 @@ export const useAISDKRuntime = <UI_MESSAGE extends UIMessage = UIMessage>(
|
|
|
240
250
|
await toolInvocations.abort();
|
|
241
251
|
},
|
|
242
252
|
onNew: async (message) => {
|
|
243
|
-
lastRunConfigRef.current = message.runConfig;
|
|
244
|
-
await completePendingToolCalls();
|
|
245
|
-
|
|
246
253
|
const createMessage = (
|
|
247
254
|
customToCreateMessage ?? toCreateMessage
|
|
248
255
|
)<UI_MESSAGE>(message);
|
|
256
|
+
|
|
257
|
+
if (!(message.startRun ?? message.role === "user")) {
|
|
258
|
+
chatHelpers.setMessages((current) => [
|
|
259
|
+
...current,
|
|
260
|
+
toUIMessage<UI_MESSAGE>(createMessage, message.role),
|
|
261
|
+
]);
|
|
262
|
+
return;
|
|
263
|
+
}
|
|
264
|
+
|
|
265
|
+
lastRunConfigRef.current = message.runConfig;
|
|
266
|
+
await completePendingToolCalls();
|
|
249
267
|
await chatHelpers.sendMessage(createMessage, {
|
|
250
268
|
metadata: message.runConfig,
|
|
251
269
|
});
|
|
252
270
|
},
|
|
253
271
|
onEdit: async (message) => {
|
|
254
|
-
lastRunConfigRef.current = message.runConfig;
|
|
255
|
-
const newMessages = sliceMessagesUntil(
|
|
256
|
-
chatHelpers.messages,
|
|
257
|
-
message.parentId,
|
|
258
|
-
);
|
|
259
|
-
chatHelpers.setMessages(newMessages);
|
|
260
|
-
|
|
261
272
|
const createMessage = (
|
|
262
273
|
customToCreateMessage ?? toCreateMessage
|
|
263
274
|
)<UI_MESSAGE>(message);
|
|
275
|
+
|
|
276
|
+
if (!(message.startRun ?? message.role === "user")) {
|
|
277
|
+
chatHelpers.setMessages((current) => [
|
|
278
|
+
...sliceMessagesUntil(current, message.parentId),
|
|
279
|
+
toUIMessage<UI_MESSAGE>(createMessage, message.role),
|
|
280
|
+
]);
|
|
281
|
+
return;
|
|
282
|
+
}
|
|
283
|
+
|
|
284
|
+
lastRunConfigRef.current = message.runConfig;
|
|
285
|
+
chatHelpers.setMessages((current) =>
|
|
286
|
+
sliceMessagesUntil(current, message.parentId),
|
|
287
|
+
);
|
|
264
288
|
await chatHelpers.sendMessage(createMessage, {
|
|
265
289
|
metadata: message.runConfig,
|
|
266
290
|
});
|