@assistant-ui/react-ai-sdk 1.3.6 → 1.3.8
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/ui/getVercelAIMessages.d.ts.map +1 -1
- package/dist/ui/getVercelAIMessages.js.map +1 -1
- package/dist/ui/use-chat/AssistantChatTransport.d.ts.map +1 -1
- package/dist/ui/use-chat/AssistantChatTransport.js.map +1 -1
- package/dist/ui/use-chat/useAISDKRuntime.d.ts.map +1 -1
- package/dist/ui/use-chat/useAISDKRuntime.js +22 -10
- package/dist/ui/use-chat/useAISDKRuntime.js.map +1 -1
- package/dist/ui/use-chat/useChatRuntime.d.ts.map +1 -1
- package/dist/ui/use-chat/useChatRuntime.js +1 -1
- package/dist/ui/use-chat/useChatRuntime.js.map +1 -1
- package/dist/ui/use-chat/useExternalHistory.d.ts.map +1 -1
- package/dist/ui/use-chat/useExternalHistory.js +137 -24
- package/dist/ui/use-chat/useExternalHistory.js.map +1 -1
- package/dist/ui/use-chat/useStreamingTiming.d.ts +11 -0
- package/dist/ui/use-chat/useStreamingTiming.d.ts.map +1 -0
- package/dist/ui/use-chat/useStreamingTiming.js +84 -0
- package/dist/ui/use-chat/useStreamingTiming.js.map +1 -0
- package/dist/ui/utils/convertMessage.d.ts +4 -4
- package/dist/ui/utils/convertMessage.d.ts.map +1 -1
- package/dist/ui/utils/convertMessage.js +94 -76
- package/dist/ui/utils/convertMessage.js.map +1 -1
- package/dist/ui/utils/sliceMessagesUntil.d.ts.map +1 -1
- package/dist/ui/utils/sliceMessagesUntil.js.map +1 -1
- package/package.json +11 -5
- package/src/ui/use-chat/useAISDKRuntime.test.ts +225 -0
- package/src/ui/use-chat/{useAISDKRuntime.tsx → useAISDKRuntime.ts} +34 -10
- package/src/ui/use-chat/{useChatRuntime.tsx → useChatRuntime.ts} +1 -1
- package/src/ui/use-chat/useExternalHistory.ts +283 -0
- package/src/ui/use-chat/useStreamingTiming.ts +102 -0
- package/src/ui/utils/convertMessage.test.ts +125 -0
- package/src/ui/utils/convertMessage.ts +107 -82
- package/src/ui/use-chat/useExternalHistory.tsx +0 -134
- /package/src/ui/{getVercelAIMessages.tsx → getVercelAIMessages.ts} +0 -0
- /package/src/ui/use-chat/{AssistantChatTransport.tsx → AssistantChatTransport.ts} +0 -0
- /package/src/ui/utils/{sliceMessagesUntil.tsx → sliceMessagesUntil.ts} +0 -0
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.
|
|
1
|
+
{"version":3,"file":"getVercelAIMessages.d.ts","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,aAAa,EACnB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAEpC,eAAO,MAAM,mBAAmB,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAC1E,SAAS,aAAa,KAEsB,UAAU,EACvD,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"getVercelAIMessages.js","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.
|
|
1
|
+
{"version":3,"file":"getVercelAIMessages.js","sourceRoot":"","sources":["../../src/ui/getVercelAIMessages.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,wBAAwB,GAEzB,MAAM,qBAAqB,CAAC;AAG7B,MAAM,CAAC,MAAM,mBAAmB,GAAG,CACjC,OAAsB,EACtB,EAAE;IACF,OAAO,wBAAwB,CAAC,OAAO,CAAiB,CAAC;AAC3D,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AssistantChatTransport.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/AssistantChatTransport.
|
|
1
|
+
{"version":3,"file":"AssistantChatTransport.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AACvD,OAAO,EACL,oBAAoB,EACpB,4BAA4B,EAC5B,SAAS,EACV,MAAM,IAAI,CAAC;AAGZ,qBAAa,sBAAsB,CACjC,UAAU,SAAS,SAAS,CAC5B,SAAQ,oBAAoB,CAAC,UAAU,CAAC;IACxC,OAAO,CAAC,OAAO,CAA+B;gBAClC,WAAW,CAAC,EAAE,4BAA4B,CAAC,UAAU,CAAC;IAqClE,UAAU,CAAC,OAAO,EAAE,gBAAgB;CAGrC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"AssistantChatTransport.js","sourceRoot":"","sources":["../../../src/ui/use-chat/AssistantChatTransport.
|
|
1
|
+
{"version":3,"file":"AssistantChatTransport.js","sourceRoot":"","sources":["../../../src/ui/use-chat/AssistantChatTransport.ts"],"names":[],"mappings":"AACA,OAAO,EACL,oBAAoB,GAGrB,MAAM,IAAI,CAAC;AACZ,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD,MAAM,OAAO,sBAEX,SAAQ,oBAAgC;IAChC,OAAO,CAA+B;IAC9C,YAAY,WAAsD;QAChE,KAAK,CAAC;YACJ,GAAG,WAAW;YACd,0BAA0B,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;gBAC5C,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,EAAE,MAAM,CAAC,eAAe,EAAE,CAAC;gBACvD,MAAM,EAAE,GACN,CAAC,MAAM,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,EAAE,QAAQ;oBAC7D,OAAO,CAAC,EAAE,CAAC;gBAEb,MAAM,SAAS,GAAG;oBAChB,GAAG,OAAO;oBACV,IAAI,EAAE;wBACJ,YAAY,EAAE,OAAO,EAAE,YAAY;wBACnC,MAAM,EAAE,OAAO,EAAE,MAAM;wBACvB,MAAM,EAAE,OAAO,EAAE,MAAM;wBACvB,KAAK,EAAE,iBAAiB,CAAC,OAAO,EAAE,KAAK,IAAI,EAAE,CAAC;wBAC9C,GAAG,OAAO,EAAE,IAAI;qBACjB;iBACF,CAAC;gBACF,MAAM,eAAe,GACnB,MAAM,WAAW,EAAE,0BAA0B,EAAE,CAAC,SAAS,CAAC,CAAC;gBAE7D,OAAO;oBACL,GAAG,eAAe;oBAClB,IAAI,EAAE,eAAe,EAAE,IAAI,IAAI;wBAC7B,GAAG,SAAS,CAAC,IAAI;wBACjB,EAAE;wBACF,QAAQ,EAAE,OAAO,CAAC,QAAQ;wBAC1B,OAAO,EAAE,OAAO,CAAC,OAAO;wBACxB,SAAS,EAAE,OAAO,CAAC,SAAS;wBAC5B,QAAQ,EAAE,OAAO,CAAC,eAAe;qBAClC;iBACF,CAAC;YACJ,CAAC;SACF,CAAC,CAAC;IACL,CAAC;IAED,UAAU,CAAC,OAAyB;QAClC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useAISDKRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.
|
|
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;AAEzE,OAAO,EAEL,KAAK,oBAAoB,EACzB,KAAK,oBAAoB,EACzB,KAAK,gBAAgB,EAQrB,KAAK,aAAa,EAEnB,MAAM,qBAAqB,CAAC;AAgB7B,MAAM,MAAM,6BAA6B,GAAG,CAC1C,UAAU,SAAS,SAAS,GAAG,SAAS,EAExC,OAAO,EAAE,aAAa,KACnB,eAAe,CAAC,UAAU,CAAC,CAAC;AAEjC,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,qBAkO5B,CAAC"}
|
|
@@ -9,6 +9,7 @@ import { getVercelAIMessages } from "../getVercelAIMessages.js";
|
|
|
9
9
|
import { AISDKMessageConverter } from "../utils/convertMessage.js";
|
|
10
10
|
import { aiSDKV6FormatAdapter, } from "../adapters/aiSDKFormatAdapter.js";
|
|
11
11
|
import { useExternalHistory, toExportedMessageRepository, } from "./useExternalHistory.js";
|
|
12
|
+
import { useStreamingTiming } from "./useStreamingTiming.js";
|
|
12
13
|
export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: customToCreateMessage, cancelPendingToolCallsOnSend = true, } = {}) => {
|
|
13
14
|
const contextAdapters = useRuntimeAdapters();
|
|
14
15
|
const [toolStatuses, setToolStatuses] = useState({});
|
|
@@ -16,13 +17,15 @@ export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: custom
|
|
|
16
17
|
const isRunning = chatHelpers.status === "submitted" ||
|
|
17
18
|
chatHelpers.status === "streaming" ||
|
|
18
19
|
hasExecutingTools;
|
|
20
|
+
const messageTiming = useStreamingTiming(chatHelpers.messages, isRunning);
|
|
19
21
|
const messages = AISDKMessageConverter.useThreadMessages({
|
|
20
22
|
isRunning,
|
|
21
23
|
messages: chatHelpers.messages,
|
|
22
24
|
metadata: useMemo(() => ({
|
|
23
25
|
toolStatuses,
|
|
26
|
+
messageTiming,
|
|
24
27
|
...(chatHelpers.error && { error: chatHelpers.error.message }),
|
|
25
|
-
}), [toolStatuses, chatHelpers.error]),
|
|
28
|
+
}), [toolStatuses, messageTiming, chatHelpers.error]),
|
|
26
29
|
});
|
|
27
30
|
const [runtimeRef] = useState(() => ({
|
|
28
31
|
get current() {
|
|
@@ -88,18 +91,27 @@ export const useAISDKRuntime = (chatHelpers, { adapters, toCreateMessage: custom
|
|
|
88
91
|
.filter(Boolean)
|
|
89
92
|
.flat()),
|
|
90
93
|
onExportExternalState: () => {
|
|
91
|
-
// Export the thread's MessageRepository
|
|
92
94
|
const exported = runtimeRef.current.thread.export();
|
|
93
|
-
|
|
95
|
+
const expandedMessages = [];
|
|
96
|
+
const lastInnerIdMap = new Map();
|
|
97
|
+
for (const item of exported.messages) {
|
|
98
|
+
const innerMessages = getExternalStoreMessages(item.message);
|
|
99
|
+
let parentId = item.parentId != null
|
|
100
|
+
? (lastInnerIdMap.get(item.parentId) ?? item.parentId)
|
|
101
|
+
: null;
|
|
102
|
+
for (const innerMessage of innerMessages) {
|
|
103
|
+
expandedMessages.push({ parentId, message: innerMessage });
|
|
104
|
+
parentId = aiSDKV6FormatAdapter.getId(innerMessage);
|
|
105
|
+
}
|
|
106
|
+
if (innerMessages.length > 0) {
|
|
107
|
+
lastInnerIdMap.set(item.message.id, aiSDKV6FormatAdapter.getId(innerMessages[innerMessages.length - 1]));
|
|
108
|
+
}
|
|
109
|
+
}
|
|
94
110
|
const result = {
|
|
95
|
-
messages:
|
|
96
|
-
parentId: item.parentId,
|
|
97
|
-
message: getExternalStoreMessages(item.message)[0],
|
|
98
|
-
})),
|
|
111
|
+
messages: expandedMessages,
|
|
99
112
|
};
|
|
100
|
-
|
|
101
|
-
|
|
102
|
-
result.headId = exported.headId;
|
|
113
|
+
if (exported.headId != null) {
|
|
114
|
+
result.headId = lastInnerIdMap.get(exported.headId) ?? exported.headId;
|
|
103
115
|
}
|
|
104
116
|
return result;
|
|
105
117
|
},
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useAISDKRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useAISDKRuntime.
|
|
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,OAAO,CAAC;AAE1C,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAClC,OAAO,EACL,uBAAuB,EAQvB,kBAAkB,EAClB,QAAQ,EAGR,wBAAwB,GACzB,MAAM,qBAAqB,CAAC;AAC7B,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;AA0B1D,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;IAEN,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,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,QAAQ,CAAC,kBAAkB,CAAC;QAClD,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;iBACvB,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,wBAAwB,EAAE,CAAC;YAEjC,MAAM,aAAa,GAAG,CACpB,qBAAqB,IAAI,eAAe,CACzC,CAAa,OAAO,CAAC,CAAC;YACvB,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,WAAW,GAAG,kBAAkB,CACpC,WAAW,CAAC,QAAQ,EACpB,OAAO,CAAC,QAAQ,CACjB,CAAC;YACF,WAAW,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YAErC,MAAM,aAAa,GAAG,CACpB,qBAAqB,IAAI,eAAe,CACzC,CAAa,OAAO,CAAC,CAAC;YACvB,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,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,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;iBAC/D,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;iBACf,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"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.
|
|
1
|
+
{"version":3,"file":"useChatRuntime.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"names":[],"mappings":"AAEA,OAAO,EAAW,KAAK,SAAS,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACtD,OAAO,EACL,gBAAgB,EAIjB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,KAAK,mBAAmB,EACxB,KAAK,6BAA6B,EACnC,6BAA0B;AAC3B,OAAO,EAAE,QAAQ,EAAiB,MAAM,IAAI,CAAC;AAI7C,MAAM,MAAM,qBAAqB,CAAC,UAAU,SAAS,SAAS,GAAG,SAAS,IACxE,QAAQ,CAAC,UAAU,CAAC,GAAG;IACrB,KAAK,CAAC,EAAE,cAAc,GAAG,SAAS,CAAC;IACnC,QAAQ,CAAC,EAAE,mBAAmB,CAAC,UAAU,CAAC,GAAG,SAAS,CAAC;IACvD,eAAe,CAAC,EAAE,6BAA6B,CAAC;CACjD,CAAC;AA0DJ,eAAO,MAAM,cAAc,GAAI,UAAU,SAAS,SAAS,GAAG,SAAS,EAAE,wBAGtE,qBAAqB,CAAC,UAAU,CAAM,KAAG,gBAS3C,CAAC"}
|
|
@@ -22,7 +22,7 @@ const useDynamicChatTransport = (transport) => {
|
|
|
22
22
|
const useChatThreadRuntime = (options) => {
|
|
23
23
|
const { adapters, transport: transportOptions, toCreateMessage, ...chatOptions } = options ?? {};
|
|
24
24
|
const transport = useDynamicChatTransport(transportOptions ?? new AssistantChatTransport());
|
|
25
|
-
const id = useAuiState((
|
|
25
|
+
const id = useAuiState((s) => s.threadListItem.id);
|
|
26
26
|
const chat = useChat({
|
|
27
27
|
...chatOptions,
|
|
28
28
|
id,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useChatRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.
|
|
1
|
+
{"version":3,"file":"useChatRuntime.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useChatRuntime.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,OAAO,EAAkB,MAAM,eAAe,CAAC;AAExD,OAAO,EAEL,kCAAkC,EAClC,mCAAmC,EACnC,WAAW,GACZ,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,eAAe,GAGhB,6BAA0B;AAE3B,OAAO,EAAE,sBAAsB,EAAE,oCAAiC;AAClE,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC;AASnD,MAAM,uBAAuB,GAAG,CAC9B,SAAoC,EACT,EAAE;IAC7B,MAAM,YAAY,GAAG,MAAM,CAA4B,SAAS,CAAC,CAAC;IAClE,SAAS,CAAC,GAAG,EAAE;QACb,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;IACnC,CAAC,CAAC,CAAC;IACH,MAAM,gBAAgB,GAAG,OAAO,CAC9B,GAAG,EAAE,CACH,IAAI,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE;QAC9B,GAAG,CAAC,CAAC,EAAE,IAAI;YACT,MAAM,GAAG,GACP,YAAY,CAAC,OAAO,CAAC,IAAuC,CAAC,CAAC;YAChE,OAAO,OAAO,GAAG,KAAK,UAAU;gBAC9B,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC;gBAChC,CAAC,CAAC,GAAG,CAAC;QACV,CAAC;KACF,CAAC,EACJ,EAAE,CACH,CAAC;IACF,OAAO,gBAAgB,CAAC;AAC1B,CAAC,CAAC;AAEF,MAAM,oBAAoB,GAAG,CAC3B,OAA2C,EACzB,EAAE;IACpB,MAAM,EACJ,QAAQ,EACR,SAAS,EAAE,gBAAgB,EAC3B,eAAe,EACf,GAAG,WAAW,EACf,GAAG,OAAO,IAAI,EAAE,CAAC;IAElB,MAAM,SAAS,GAAG,uBAAuB,CACvC,gBAAgB,IAAI,IAAI,sBAAsB,EAAE,CACjD,CAAC;IAEF,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;IACnD,MAAM,IAAI,GAAG,OAAO,CAAC;QACnB,GAAG,WAAW;QACd,EAAE;QACF,SAAS;KACV,CAAC,CAAC;IAEH,MAAM,OAAO,GAAG,eAAe,CAAC,IAAI,EAAE;QACpC,QAAQ;QACR,GAAG,CAAC,eAAe,IAAI,EAAE,eAAe,EAAE,CAAC;KAC5C,CAAC,CAAC;IAEH,IAAI,SAAS,YAAY,sBAAsB,EAAE,CAAC;QAChD,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,cAAc,GAAG,CAA2C,EACvE,KAAK,EACL,GAAG,OAAO,KAC2B,EAAE,EAAoB,EAAE;IAC7D,MAAM,YAAY,GAAG,kCAAkC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IACnE,OAAO,mCAAmC,CAAC;QACzC,WAAW,EAAE,SAAS,WAAW;YAC/B,OAAO,oBAAoB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC;QACD,OAAO,EAAE,YAAY;QACrB,YAAY,EAAE,IAAI;KACnB,CAAC,CAAC;AACL,CAAC,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useExternalHistory.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useExternalHistory.
|
|
1
|
+
{"version":3,"file":"useExternalHistory.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"names":[],"mappings":"AAEA,OAAO,EACL,gBAAgB,EAChB,oBAAoB,EACpB,aAAa,EACb,oBAAoB,EAEpB,uBAAuB,EACvB,yBAAyB,EAG1B,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAA+B,SAAS,EAAe,MAAM,OAAO,CAAC;AAI5E,eAAO,MAAM,2BAA2B,GAAI,QAAQ,EAClD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,UAAU,uBAAuB,CAAC,QAAQ,CAAC,KAC1C,yBAWF,CAAC;AAEF,eAAO,MAAM,kBAAkB,GAAI,QAAQ,EACzC,YAAY,SAAS,CAAC,gBAAgB,CAAC,EACvC,gBAAgB,oBAAoB,GAAG,SAAS,EAChD,kBAAkB,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,aAAa,EAAE,EAC3D,sBAAsB,oBAAoB,CAAC,QAAQ,EAAE,GAAG,CAAC,EACzD,eAAe,CAAC,QAAQ,EAAE,QAAQ,EAAE,KAAK,IAAI,YAoP9C,CAAC"}
|
|
@@ -20,11 +20,10 @@ export const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages,
|
|
|
20
20
|
const optionalThreadListItem = useCallback(() => (aui.threadListItem.source ? aui.threadListItem() : null), [aui]);
|
|
21
21
|
const [isLoading, setIsLoading] = useState(false);
|
|
22
22
|
const historyIds = useRef(new Set());
|
|
23
|
-
const onSetMessagesRef = useRef(
|
|
23
|
+
const onSetMessagesRef = useRef(onSetMessages);
|
|
24
24
|
useEffect(() => {
|
|
25
25
|
onSetMessagesRef.current = onSetMessages;
|
|
26
26
|
});
|
|
27
|
-
// Load messages from history adapter on mount
|
|
28
27
|
useEffect(() => {
|
|
29
28
|
if (!historyAdapter || loadedRef.current)
|
|
30
29
|
return;
|
|
@@ -51,14 +50,12 @@ export const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages,
|
|
|
51
50
|
setIsLoading(false);
|
|
52
51
|
}
|
|
53
52
|
};
|
|
54
|
-
|
|
55
|
-
|
|
56
|
-
|
|
57
|
-
|
|
58
|
-
return;
|
|
59
|
-
}
|
|
60
|
-
loadHistory();
|
|
53
|
+
loadedRef.current = true;
|
|
54
|
+
if (!optionalThreadListItem()?.getState().remoteId) {
|
|
55
|
+
setIsLoading(false);
|
|
56
|
+
return;
|
|
61
57
|
}
|
|
58
|
+
loadHistory();
|
|
62
59
|
}, [
|
|
63
60
|
historyAdapter,
|
|
64
61
|
storageFormatAdapter,
|
|
@@ -66,27 +63,143 @@ export const useExternalHistory = (runtimeRef, historyAdapter, toThreadMessages,
|
|
|
66
63
|
runtimeRef,
|
|
67
64
|
optionalThreadListItem,
|
|
68
65
|
]);
|
|
66
|
+
const runStartRef = useRef(null);
|
|
67
|
+
const persistTimerRef = useRef(null);
|
|
68
|
+
const stepBoundariesRef = useRef([]);
|
|
69
|
+
const wasRunningRef = useRef(false);
|
|
70
|
+
const toolCallCountRef = useRef(0);
|
|
69
71
|
useEffect(() => {
|
|
70
|
-
|
|
71
|
-
const {
|
|
72
|
-
|
|
72
|
+
const unsubscribe = runtimeRef.current.thread.subscribe(() => {
|
|
73
|
+
const { isRunning } = runtimeRef.current.thread.getState();
|
|
74
|
+
const wasRunning = wasRunningRef.current;
|
|
75
|
+
wasRunningRef.current = isRunning;
|
|
76
|
+
// Track step boundaries by content changes (more reliable than isRunning)
|
|
77
|
+
if (runStartRef.current != null) {
|
|
78
|
+
const lastMsg = runtimeRef.current.thread.getState().messages.at(-1);
|
|
79
|
+
if (lastMsg?.role === "assistant") {
|
|
80
|
+
const currentToolCallCount = lastMsg.content.filter((p) => p.type === "tool-call").length;
|
|
81
|
+
while (toolCallCountRef.current < currentToolCallCount) {
|
|
82
|
+
stepBoundariesRef.current.push(Date.now() - runStartRef.current);
|
|
83
|
+
toolCallCountRef.current++;
|
|
84
|
+
}
|
|
85
|
+
}
|
|
86
|
+
}
|
|
87
|
+
if (isRunning) {
|
|
88
|
+
if (runStartRef.current == null) {
|
|
89
|
+
runStartRef.current = Date.now();
|
|
90
|
+
stepBoundariesRef.current = [];
|
|
91
|
+
toolCallCountRef.current = 0;
|
|
92
|
+
}
|
|
93
|
+
// Cancel any pending persist — isRunning went back to true
|
|
94
|
+
if (persistTimerRef.current) {
|
|
95
|
+
clearTimeout(persistTimerRef.current);
|
|
96
|
+
persistTimerRef.current = null;
|
|
97
|
+
}
|
|
73
98
|
return;
|
|
74
|
-
|
|
75
|
-
|
|
76
|
-
|
|
77
|
-
|
|
78
|
-
|
|
79
|
-
|
|
99
|
+
}
|
|
100
|
+
// Only act on the true→false transition
|
|
101
|
+
if (!wasRunning)
|
|
102
|
+
return;
|
|
103
|
+
// Record step boundary offset (synchronous for accuracy)
|
|
104
|
+
if (runStartRef.current != null) {
|
|
105
|
+
stepBoundariesRef.current.push(Date.now() - runStartRef.current);
|
|
106
|
+
}
|
|
107
|
+
// Debounce: wait one macrotask so agentic step flickers are absorbed
|
|
108
|
+
if (persistTimerRef.current)
|
|
109
|
+
clearTimeout(persistTimerRef.current);
|
|
110
|
+
persistTimerRef.current = setTimeout(async () => {
|
|
111
|
+
persistTimerRef.current = null;
|
|
112
|
+
// Re-read latest state — may have changed since the timeout was scheduled
|
|
113
|
+
const latest = runtimeRef.current.thread.getState();
|
|
114
|
+
if (latest.isRunning)
|
|
115
|
+
return; // was just a flicker
|
|
116
|
+
// Derive durationMs from the last boundary (covers all steps)
|
|
117
|
+
const boundaries = stepBoundariesRef.current;
|
|
118
|
+
const durationMs = boundaries.length > 0 ? boundaries.at(-1) : undefined;
|
|
119
|
+
// Fallback: if only 1 boundary but message has multiple steps, distribute evenly
|
|
120
|
+
if (boundaries.length === 1 && durationMs != null) {
|
|
121
|
+
const lastAssistant = latest.messages.findLast((m) => m.role === "assistant");
|
|
122
|
+
if (lastAssistant) {
|
|
123
|
+
const tcCount = lastAssistant.content.filter((p) => p.type === "tool-call").length;
|
|
124
|
+
if (tcCount > 0) {
|
|
125
|
+
const totalSteps = tcCount + 1;
|
|
126
|
+
const stepDur = durationMs / totalSteps;
|
|
127
|
+
boundaries.length = 0;
|
|
128
|
+
for (let i = 0; i < totalSteps; i++) {
|
|
129
|
+
boundaries.push(Math.round((i + 1) * stepDur));
|
|
130
|
+
}
|
|
131
|
+
}
|
|
132
|
+
}
|
|
133
|
+
}
|
|
134
|
+
// Build per-step timestamps when there are multiple steps
|
|
135
|
+
const stepTimestamps = boundaries.length > 1
|
|
136
|
+
? boundaries.map((endMs, i) => ({
|
|
137
|
+
start_ms: i === 0 ? 0 : boundaries[i - 1],
|
|
138
|
+
end_ms: endMs,
|
|
139
|
+
}))
|
|
140
|
+
: undefined;
|
|
141
|
+
runStartRef.current = null;
|
|
142
|
+
stepBoundariesRef.current = [];
|
|
143
|
+
const telemetryOptions = {
|
|
144
|
+
...(durationMs != null ? { durationMs } : undefined),
|
|
145
|
+
...(stepTimestamps != null ? { stepTimestamps } : undefined),
|
|
146
|
+
};
|
|
147
|
+
const { messages } = latest;
|
|
148
|
+
let lastInnerMessageId = null;
|
|
149
|
+
const getLastInnerId = (msgs) => msgs.length > 0 ? storageFormatAdapter.getId(msgs.at(-1)) : null;
|
|
150
|
+
const toBatchItems = (msgs) => msgs.map((msg, idx) => ({
|
|
151
|
+
parentId: idx === 0
|
|
152
|
+
? lastInnerMessageId
|
|
153
|
+
: storageFormatAdapter.getId(msgs[idx - 1]),
|
|
154
|
+
message: msg,
|
|
155
|
+
}));
|
|
156
|
+
for (const message of messages) {
|
|
157
|
+
const innerMessages = getExternalStoreMessages(message);
|
|
158
|
+
const isReady = message.status === undefined ||
|
|
159
|
+
message.status.type === "complete" ||
|
|
160
|
+
message.status.type === "incomplete";
|
|
161
|
+
if (!isReady) {
|
|
162
|
+
lastInnerMessageId =
|
|
163
|
+
getLastInnerId(innerMessages) ?? lastInnerMessageId;
|
|
80
164
|
continue;
|
|
165
|
+
}
|
|
166
|
+
if (historyIds.current.has(message.id)) {
|
|
167
|
+
if (durationMs !== undefined) {
|
|
168
|
+
const formatAdapter = historyAdapter?.withFormat?.(storageFormatAdapter);
|
|
169
|
+
let parentId = lastInnerMessageId;
|
|
170
|
+
for (const innerMessage of innerMessages) {
|
|
171
|
+
try {
|
|
172
|
+
await formatAdapter?.update?.({ parentId, message: innerMessage }, storageFormatAdapter.getId(innerMessage));
|
|
173
|
+
}
|
|
174
|
+
catch {
|
|
175
|
+
// ignore update failures to avoid breaking the message processing loop
|
|
176
|
+
}
|
|
177
|
+
parentId = storageFormatAdapter.getId(innerMessage);
|
|
178
|
+
}
|
|
179
|
+
}
|
|
180
|
+
lastInnerMessageId =
|
|
181
|
+
getLastInnerId(innerMessages) ?? lastInnerMessageId;
|
|
182
|
+
continue;
|
|
183
|
+
}
|
|
81
184
|
historyIds.current.add(message.id);
|
|
82
|
-
const
|
|
83
|
-
|
|
84
|
-
|
|
85
|
-
|
|
86
|
-
}
|
|
185
|
+
const formatAdapter = historyAdapter?.withFormat?.(storageFormatAdapter);
|
|
186
|
+
const batchItems = toBatchItems(innerMessages);
|
|
187
|
+
for (const item of batchItems) {
|
|
188
|
+
await formatAdapter?.append(item);
|
|
189
|
+
}
|
|
190
|
+
lastInnerMessageId =
|
|
191
|
+
getLastInnerId(innerMessages) ?? lastInnerMessageId;
|
|
192
|
+
formatAdapter?.reportTelemetry?.(batchItems, telemetryOptions);
|
|
87
193
|
}
|
|
88
|
-
}
|
|
194
|
+
}, 0);
|
|
89
195
|
});
|
|
196
|
+
return () => {
|
|
197
|
+
unsubscribe();
|
|
198
|
+
if (persistTimerRef.current) {
|
|
199
|
+
clearTimeout(persistTimerRef.current);
|
|
200
|
+
persistTimerRef.current = null;
|
|
201
|
+
}
|
|
202
|
+
};
|
|
90
203
|
}, [historyAdapter, storageFormatAdapter, runtimeRef]);
|
|
91
204
|
return isLoading;
|
|
92
205
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"useExternalHistory.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useExternalHistory.
|
|
1
|
+
{"version":3,"file":"useExternalHistory.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useExternalHistory.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAKL,wBAAwB,EAGxB,QAAQ,EACR,MAAM,GACP,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAa,WAAW,EAAE,MAAM,OAAO,CAAC;AAE5E,MAAM,EAAE,iBAAiB,EAAE,GAAG,QAAQ,CAAC;AAEvC,MAAM,CAAC,MAAM,2BAA2B,GAAG,CACzC,gBAA2D,EAC3D,QAA2C,EAChB,EAAE;IAC7B,OAAO;QACL,MAAM,EAAE,QAAQ,CAAC,MAAO;QACxB,QAAQ,EAAE,QAAQ,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACpC,MAAM,OAAO,GAAG,gBAAgB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAE,CAAC;YAClD,OAAO;gBACL,GAAG,CAAC;gBACJ,OAAO;aACR,CAAC;QACJ,CAAC,CAAC;KACH,CAAC;AACJ,CAAC,CAAC;AAEF,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,UAAuC,EACvC,cAAgD,EAChD,gBAA2D,EAC3D,oBAAyD,EACzD,aAA6C,EAC7C,EAAE;IACF,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAEhC,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;IACrB,MAAM,sBAAsB,GAAG,WAAW,CACxC,GAAG,EAAE,CAAC,CAAC,GAAG,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,CAAC,cAAc,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,EAC/D,CAAC,GAAG,CAAC,CACN,CAAC;IAEF,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAElD,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,GAAG,EAAU,CAAC,CAAC;IAE7C,MAAM,gBAAgB,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;IAC/C,SAAS,CAAC,GAAG,EAAE;QACb,gBAAgB,CAAC,OAAO,GAAG,aAAa,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,cAAc,IAAI,SAAS,CAAC,OAAO;YAAE,OAAO;QAEjD,MAAM,WAAW,GAAG,KAAK,IAAI,EAAE;YAC7B,YAAY,CAAC,IAAI,CAAC,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,IAAI,GAAG,MAAM,cAAc;qBAC9B,UAAU,EAAE,CAAC,oBAAoB,CAAC;qBAClC,IAAI,EAAE,CAAC;gBACV,IAAI,IAAI,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACrC,MAAM,SAAS,GAAG,2BAA2B,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;oBACtE,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;oBAE5C,MAAM,QAAQ,GAAG,IAAI,iBAAiB,EAAE,CAAC;oBACzC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;oBAC3B,MAAM,QAAQ,GAAG,QAAQ,CAAC,WAAW,EAAE,CAAC;oBAExC,gBAAgB,CAAC,OAAO,CACtB,QAAQ,CAAC,GAAG,CAAC,CAAA,wBAAkC,CAAA,CAAC,CAAC,IAAI,EAAE,CACxD,CAAC;oBAEF,UAAU,CAAC,OAAO,GAAG,IAAI,GAAG,CAC1B,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAC5C,CAAC;gBACJ,CAAC;YACH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YAC1D,CAAC;oBAAS,CAAC;gBACT,YAAY,CAAC,KAAK,CAAC,CAAC;YACtB,CAAC;QACH,CAAC,CAAC;QAEF,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC;QAEzB,IAAI,CAAC,sBAAsB,EAAE,EAAE,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC;YACnD,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,OAAO;QACT,CAAC;QAED,WAAW,EAAE,CAAC;IAChB,CAAC,EAAE;QACD,cAAc;QACd,oBAAoB;QACpB,gBAAgB;QAChB,UAAU;QACV,sBAAsB;KACvB,CAAC,CAAC;IAEH,MAAM,WAAW,GAAG,MAAM,CAAgB,IAAI,CAAC,CAAC;IAChD,MAAM,eAAe,GAAG,MAAM,CAAuC,IAAI,CAAC,CAAC;IAC3E,MAAM,iBAAiB,GAAG,MAAM,CAAW,EAAE,CAAC,CAAC;IAC/C,MAAM,aAAa,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IACpC,MAAM,gBAAgB,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;IAEnC,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE;YAC3D,MAAM,EAAE,SAAS,EAAE,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YAC3D,MAAM,UAAU,GAAG,aAAa,CAAC,OAAO,CAAC;YACzC,aAAa,CAAC,OAAO,GAAG,SAAS,CAAC;YAElC,0EAA0E;YAC1E,IAAI,WAAW,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;gBAChC,MAAM,OAAO,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;gBACrE,IAAI,OAAO,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;oBAClC,MAAM,oBAAoB,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CACjD,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAC9B,CAAC,MAAM,CAAC;oBACT,OAAO,gBAAgB,CAAC,OAAO,GAAG,oBAAoB,EAAE,CAAC;wBACvD,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;wBACjE,gBAAgB,CAAC,OAAO,EAAE,CAAC;oBAC7B,CAAC;gBACH,CAAC;YACH,CAAC;YAED,IAAI,SAAS,EAAE,CAAC;gBACd,IAAI,WAAW,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;oBAChC,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;oBACjC,iBAAiB,CAAC,OAAO,GAAG,EAAE,CAAC;oBAC/B,gBAAgB,CAAC,OAAO,GAAG,CAAC,CAAC;gBAC/B,CAAC;gBACD,2DAA2D;gBAC3D,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;oBAC5B,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;oBACtC,eAAe,CAAC,OAAO,GAAG,IAAI,CAAC;gBACjC,CAAC;gBACD,OAAO;YACT,CAAC;YAED,wCAAwC;YACxC,IAAI,CAAC,UAAU;gBAAE,OAAO;YAExB,yDAAyD;YACzD,IAAI,WAAW,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC;gBAChC,iBAAiB,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,WAAW,CAAC,OAAO,CAAC,CAAC;YACnE,CAAC;YAED,qEAAqE;YACrE,IAAI,eAAe,CAAC,OAAO;gBAAE,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;YACnE,eAAe,CAAC,OAAO,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;gBAC9C,eAAe,CAAC,OAAO,GAAG,IAAI,CAAC;gBAE/B,0EAA0E;gBAC1E,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;gBACpD,IAAI,MAAM,CAAC,SAAS;oBAAE,OAAO,CAAC,qBAAqB;gBAEnD,8DAA8D;gBAC9D,MAAM,UAAU,GAAG,iBAAiB,CAAC,OAAO,CAAC;gBAC7C,MAAM,UAAU,GACd,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAExD,iFAAiF;gBACjF,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,UAAU,IAAI,IAAI,EAAE,CAAC;oBAClD,MAAM,aAAa,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAC5C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAC9B,CAAC;oBACF,IAAI,aAAa,EAAE,CAAC;wBAClB,MAAM,OAAO,GAAG,aAAa,CAAC,OAAO,CAAC,MAAM,CAC1C,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAC9B,CAAC,MAAM,CAAC;wBACT,IAAI,OAAO,GAAG,CAAC,EAAE,CAAC;4BAChB,MAAM,UAAU,GAAG,OAAO,GAAG,CAAC,CAAC;4BAC/B,MAAM,OAAO,GAAG,UAAU,GAAG,UAAU,CAAC;4BACxC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC;4BACtB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;gCACpC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;4BACjD,CAAC;wBACH,CAAC;oBACH,CAAC;gBACH,CAAC;gBAED,0DAA0D;gBAC1D,MAAM,cAAc,GAClB,UAAU,CAAC,MAAM,GAAG,CAAC;oBACnB,CAAC,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;wBAC5B,QAAQ,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,CAAE;wBAC1C,MAAM,EAAE,KAAK;qBACd,CAAC,CAAC;oBACL,CAAC,CAAC,SAAS,CAAC;gBAEhB,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;gBAC3B,iBAAiB,CAAC,OAAO,GAAG,EAAE,CAAC;gBAE/B,MAAM,gBAAgB,GAAG;oBACvB,GAAG,CAAC,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;oBACpD,GAAG,CAAC,cAAc,IAAI,IAAI,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;iBAC7D,CAAC;gBAEF,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;gBAC5B,IAAI,kBAAkB,GAAkB,IAAI,CAAC;gBAE7C,MAAM,cAAc,GAAG,CAAC,IAAgB,EAAiB,EAAE,CACzD,IAAI,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAEpE,MAAM,YAAY,GAAG,CAAC,IAAgB,EAAE,EAAE,CACxC,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;oBACtB,QAAQ,EACN,GAAG,KAAK,CAAC;wBACP,CAAC,CAAC,kBAAkB;wBACpB,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,CAAE,CAAC;oBAChD,OAAO,EAAE,GAAG;iBACb,CAAC,CAAC,CAAC;gBAEN,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;oBAC/B,MAAM,aAAa,GAAG,wBAAwB,CAAW,OAAO,CAAC,CAAC;oBAElE,MAAM,OAAO,GACX,OAAO,CAAC,MAAM,KAAK,SAAS;wBAC5B,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,UAAU;wBAClC,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,YAAY,CAAC;oBAEvC,IAAI,CAAC,OAAO,EAAE,CAAC;wBACb,kBAAkB;4BAChB,cAAc,CAAC,aAAa,CAAC,IAAI,kBAAkB,CAAC;wBACtD,SAAS;oBACX,CAAC;oBAED,IAAI,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,CAAC;wBACvC,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;4BAC7B,MAAM,aAAa,GACjB,cAAc,EAAE,UAAU,EAAE,CAAC,oBAAoB,CAAC,CAAC;4BACrD,IAAI,QAAQ,GAAG,kBAAkB,CAAC;4BAClC,KAAK,MAAM,YAAY,IAAI,aAAa,EAAE,CAAC;gCACzC,IAAI,CAAC;oCACH,MAAM,aAAa,EAAE,MAAM,EAAE,CAC3B,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,EACnC,oBAAoB,CAAC,KAAK,CAAC,YAAY,CAAC,CACzC,CAAC;gCACJ,CAAC;gCAAC,MAAM,CAAC;oCACP,uEAAuE;gCACzE,CAAC;gCACD,QAAQ,GAAG,oBAAoB,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;4BACtD,CAAC;wBACH,CAAC;wBACD,kBAAkB;4BAChB,cAAc,CAAC,aAAa,CAAC,IAAI,kBAAkB,CAAC;wBACtD,SAAS;oBACX,CAAC;oBACD,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;oBAEnC,MAAM,aAAa,GACjB,cAAc,EAAE,UAAU,EAAE,CAAC,oBAAoB,CAAC,CAAC;oBAErD,MAAM,UAAU,GAAG,YAAY,CAAC,aAAa,CAAC,CAAC;oBAC/C,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;wBAC9B,MAAM,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;oBACpC,CAAC;oBAED,kBAAkB;wBAChB,cAAc,CAAC,aAAa,CAAC,IAAI,kBAAkB,CAAC;oBAEtD,aAAa,EAAE,eAAe,EAAE,CAAC,UAAU,EAAE,gBAAgB,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC,EAAE,CAAC,CAAC,CAAC;QACR,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE;YACV,WAAW,EAAE,CAAC;YACd,IAAI,eAAe,CAAC,OAAO,EAAE,CAAC;gBAC5B,YAAY,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACtC,eAAe,CAAC,OAAO,GAAG,IAAI,CAAC;YACjC,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,cAAc,EAAE,oBAAoB,EAAE,UAAU,CAAC,CAAC,CAAC;IAEvD,OAAO,SAAS,CAAC;AACnB,CAAC,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { UIMessage } from "@ai-sdk/react";
|
|
2
|
+
import type { MessageTiming } from "@assistant-ui/react";
|
|
3
|
+
/**
|
|
4
|
+
* Tracks streaming timing for AI SDK messages client-side.
|
|
5
|
+
*
|
|
6
|
+
* Observes `isRunning` transitions and content changes to estimate
|
|
7
|
+
* timing metrics (TTFT, duration, tok/s). Timing is finalized when
|
|
8
|
+
* streaming ends and stored per message ID.
|
|
9
|
+
*/
|
|
10
|
+
export declare const useStreamingTiming: (messages: UIMessage[], isRunning: boolean) => Record<string, MessageTiming>;
|
|
11
|
+
//# sourceMappingURL=useStreamingTiming.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useStreamingTiming.d.ts","sourceRoot":"","sources":["../../../src/ui/use-chat/useStreamingTiming.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AA4BzD;;;;;;GAMG;AACH,eAAO,MAAM,kBAAkB,GAC7B,UAAU,SAAS,EAAE,EACrB,WAAW,OAAO,KACjB,MAAM,CAAC,MAAM,EAAE,aAAa,CA0D9B,CAAC"}
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
"use client";
|
|
2
|
+
import { useEffect, useRef, useState } from "react";
|
|
3
|
+
import { isToolUIPart } from "ai";
|
|
4
|
+
function getTextLength(message) {
|
|
5
|
+
if (!message?.parts)
|
|
6
|
+
return 0;
|
|
7
|
+
let len = 0;
|
|
8
|
+
for (const part of message.parts) {
|
|
9
|
+
if (part.type === "text")
|
|
10
|
+
len += part.text.length;
|
|
11
|
+
}
|
|
12
|
+
return len;
|
|
13
|
+
}
|
|
14
|
+
function getToolCallCount(message) {
|
|
15
|
+
if (!message?.parts)
|
|
16
|
+
return 0;
|
|
17
|
+
let count = 0;
|
|
18
|
+
for (const part of message.parts) {
|
|
19
|
+
if (isToolUIPart(part))
|
|
20
|
+
count++;
|
|
21
|
+
}
|
|
22
|
+
return count;
|
|
23
|
+
}
|
|
24
|
+
/**
|
|
25
|
+
* Tracks streaming timing for AI SDK messages client-side.
|
|
26
|
+
*
|
|
27
|
+
* Observes `isRunning` transitions and content changes to estimate
|
|
28
|
+
* timing metrics (TTFT, duration, tok/s). Timing is finalized when
|
|
29
|
+
* streaming ends and stored per message ID.
|
|
30
|
+
*/
|
|
31
|
+
export const useStreamingTiming = (messages, isRunning) => {
|
|
32
|
+
const [timings, setTimings] = useState({});
|
|
33
|
+
const trackRef = useRef(null);
|
|
34
|
+
useEffect(() => {
|
|
35
|
+
const lastAssistant = messages.findLast((m) => m.role === "assistant");
|
|
36
|
+
if (isRunning && lastAssistant) {
|
|
37
|
+
// Start tracking if not already or if message changed
|
|
38
|
+
if (!trackRef.current ||
|
|
39
|
+
trackRef.current.messageId !== lastAssistant.id) {
|
|
40
|
+
trackRef.current = {
|
|
41
|
+
messageId: lastAssistant.id,
|
|
42
|
+
startTime: Date.now(),
|
|
43
|
+
lastContentLength: 0,
|
|
44
|
+
totalChunks: 0,
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
// Track content changes
|
|
48
|
+
const t = trackRef.current;
|
|
49
|
+
const len = getTextLength(lastAssistant);
|
|
50
|
+
if (len > t.lastContentLength) {
|
|
51
|
+
if (t.firstTokenTime === undefined) {
|
|
52
|
+
t.firstTokenTime = Date.now() - t.startTime;
|
|
53
|
+
}
|
|
54
|
+
t.totalChunks++;
|
|
55
|
+
t.lastContentLength = len;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
else if (!isRunning && trackRef.current) {
|
|
59
|
+
// Streaming ended — finalize timing
|
|
60
|
+
const t = trackRef.current;
|
|
61
|
+
const totalStreamTime = Date.now() - t.startTime;
|
|
62
|
+
const tokenCount = Math.ceil(t.lastContentLength / 4);
|
|
63
|
+
const toolCallCount = getToolCallCount(lastAssistant);
|
|
64
|
+
const timing = {
|
|
65
|
+
streamStartTime: t.startTime,
|
|
66
|
+
totalStreamTime,
|
|
67
|
+
totalChunks: t.totalChunks,
|
|
68
|
+
toolCallCount,
|
|
69
|
+
...(t.firstTokenTime !== undefined && {
|
|
70
|
+
firstTokenTime: t.firstTokenTime,
|
|
71
|
+
}),
|
|
72
|
+
...(tokenCount > 0 && { tokenCount }),
|
|
73
|
+
...(totalStreamTime > 0 &&
|
|
74
|
+
tokenCount > 0 && {
|
|
75
|
+
tokensPerSecond: tokenCount / (totalStreamTime / 1000),
|
|
76
|
+
}),
|
|
77
|
+
};
|
|
78
|
+
setTimings((prev) => ({ ...prev, [t.messageId]: timing }));
|
|
79
|
+
trackRef.current = null;
|
|
80
|
+
}
|
|
81
|
+
}, [messages, isRunning]);
|
|
82
|
+
return timings;
|
|
83
|
+
};
|
|
84
|
+
//# sourceMappingURL=useStreamingTiming.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"useStreamingTiming.js","sourceRoot":"","sources":["../../../src/ui/use-chat/useStreamingTiming.ts"],"names":[],"mappings":"AAAA,YAAY,CAAC;AAEb,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AAEpD,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAC;AAWlC,SAAS,aAAa,CAAC,OAA8B;IACnD,IAAI,CAAC,OAAO,EAAE,KAAK;QAAE,OAAO,CAAC,CAAC;IAC9B,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,IAAI,CAAC,IAAI,KAAK,MAAM;YAAE,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IACpD,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,gBAAgB,CAAC,OAA8B;IACtD,IAAI,CAAC,OAAO,EAAE,KAAK;QAAE,OAAO,CAAC,CAAC;IAC9B,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,IAAI,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC;QACjC,IAAI,YAAY,CAAC,IAAI,CAAC;YAAE,KAAK,EAAE,CAAC;IAClC,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED;;;;;;GAMG;AACH,MAAM,CAAC,MAAM,kBAAkB,GAAG,CAChC,QAAqB,EACrB,SAAkB,EACa,EAAE;IACjC,MAAM,CAAC,OAAO,EAAE,UAAU,CAAC,GAAG,QAAQ,CAAgC,EAAE,CAAC,CAAC;IAC1E,MAAM,QAAQ,GAAG,MAAM,CAAuB,IAAI,CAAC,CAAC;IAEpD,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,aAAa,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAEvE,IAAI,SAAS,IAAI,aAAa,EAAE,CAAC;YAC/B,sDAAsD;YACtD,IACE,CAAC,QAAQ,CAAC,OAAO;gBACjB,QAAQ,CAAC,OAAO,CAAC,SAAS,KAAK,aAAa,CAAC,EAAE,EAC/C,CAAC;gBACD,QAAQ,CAAC,OAAO,GAAG;oBACjB,SAAS,EAAE,aAAa,CAAC,EAAE;oBAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;oBACrB,iBAAiB,EAAE,CAAC;oBACpB,WAAW,EAAE,CAAC;iBACf,CAAC;YACJ,CAAC;YAED,wBAAwB;YACxB,MAAM,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC;YAC3B,MAAM,GAAG,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;YACzC,IAAI,GAAG,GAAG,CAAC,CAAC,iBAAiB,EAAE,CAAC;gBAC9B,IAAI,CAAC,CAAC,cAAc,KAAK,SAAS,EAAE,CAAC;oBACnC,CAAC,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC;gBAC9C,CAAC;gBACD,CAAC,CAAC,WAAW,EAAE,CAAC;gBAChB,CAAC,CAAC,iBAAiB,GAAG,GAAG,CAAC;YAC5B,CAAC;QACH,CAAC;aAAM,IAAI,CAAC,SAAS,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YAC1C,oCAAoC;YACpC,MAAM,CAAC,GAAG,QAAQ,CAAC,OAAO,CAAC;YAC3B,MAAM,eAAe,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,SAAS,CAAC;YACjD,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,iBAAiB,GAAG,CAAC,CAAC,CAAC;YACtD,MAAM,aAAa,GAAG,gBAAgB,CAAC,aAAa,CAAC,CAAC;YAEtD,MAAM,MAAM,GAAkB;gBAC5B,eAAe,EAAE,CAAC,CAAC,SAAS;gBAC5B,eAAe;gBACf,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,aAAa;gBACb,GAAG,CAAC,CAAC,CAAC,cAAc,KAAK,SAAS,IAAI;oBACpC,cAAc,EAAE,CAAC,CAAC,cAAc;iBACjC,CAAC;gBACF,GAAG,CAAC,UAAU,GAAG,CAAC,IAAI,EAAE,UAAU,EAAE,CAAC;gBACrC,GAAG,CAAC,eAAe,GAAG,CAAC;oBACrB,UAAU,GAAG,CAAC,IAAI;oBAChB,eAAe,EAAE,UAAU,GAAG,CAAC,eAAe,GAAG,IAAI,CAAC;iBACvD,CAAC;aACL,CAAC;YACF,UAAU,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC;YAC3D,QAAQ,CAAC,OAAO,GAAG,IAAI,CAAC;QAC1B,CAAC;IACH,CAAC,EAAE,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC;IAE1B,OAAO,OAAO,CAAC;AACjB,CAAC,CAAC"}
|
|
@@ -6,10 +6,10 @@ export declare const AISDKMessageConverter: {
|
|
|
6
6
|
isRunning: boolean;
|
|
7
7
|
joinStrategy?: "concat-content" | "none" | undefined;
|
|
8
8
|
metadata?: useExternalMessageConverter.Metadata;
|
|
9
|
-
}) => import("@assistant-ui/
|
|
10
|
-
toThreadMessages: (messages: UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[], isRunning?: boolean, metadata?: useExternalMessageConverter.Metadata) => import("@assistant-ui/
|
|
11
|
-
toOriginalMessages: (input: import("@assistant-ui/
|
|
12
|
-
toOriginalMessage: (input: import("@assistant-ui/
|
|
9
|
+
}) => import("@assistant-ui/core").ThreadMessage[];
|
|
10
|
+
toThreadMessages: (messages: UIMessage<unknown, import("ai").UIDataTypes, import("ai").UITools>[], isRunning?: boolean, metadata?: useExternalMessageConverter.Metadata) => import("@assistant-ui/core").ThreadMessage[];
|
|
11
|
+
toOriginalMessages: (input: import("@assistant-ui/core").ThreadState | import("@assistant-ui/core").ThreadMessage | import("@assistant-ui/core").ThreadMessage["content"][number]) => unknown[];
|
|
12
|
+
toOriginalMessage: (input: import("@assistant-ui/core").ThreadState | import("@assistant-ui/core").ThreadMessage | import("@assistant-ui/core").ThreadMessage["content"][number]) => {};
|
|
13
13
|
useOriginalMessage: () => {};
|
|
14
14
|
useOriginalMessages: () => unknown[];
|
|
15
15
|
};
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,KAAK,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/D,OAAO,EAOL,KAAK,2BAA2B,EAEjC,MAAM,qBAAqB,CAAC;
|
|
1
|
+
{"version":3,"file":"convertMessage.d.ts","sourceRoot":"","sources":["../../../src/ui/utils/convertMessage.ts"],"names":[],"mappings":"AAAA,OAAO,EAA6B,KAAK,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/D,OAAO,EAOL,KAAK,2BAA2B,EAEjC,MAAM,qBAAqB,CAAC;AA2J7B,eAAO,MAAM,qBAAqB;;;;;;;;wHA7IL,qDAAyB;uHAGb,qDACtB;;;CAmMlB,CAAC"}
|