@ulearning/ai-sdk-adapter 0.0.0-stage → 1.0.0-next.0

package/src/runtime/useStreamingTiming.ts

@@ -0,0 +1,69 @@
+"use client";
+
+import type { UIMessage } from "ai";
+import type { MaybeRefOrGetter } from "vue";
+import { isToolUIPart } from "ai";
+import {
+  useStreamingTiming as useStreamingTimingPrimitive,
+  type StreamingTimingAccessors,
+} from "@ulearning/ai-core/vue";
+
+const findAssistant = (
+  messages: readonly UIMessage[],
+  messageId: string,
+): UIMessage | undefined =>
+  messages.find((m) => m.role === "assistant" && m.id === messageId);
+
+const getTextLength = (
+  messages: readonly UIMessage[],
+  messageId: string,
+): number => {
+  const message = findAssistant(messages, messageId);
+  if (!message?.parts) return 0;
+  let len = 0;
+  for (const part of message.parts) {
+    if (part?.type === "text") len += part.text?.length ?? 0;
+  }
+  return len;
+};
+
+const getToolCallCount = (
+  messages: readonly UIMessage[],
+  messageId: string,
+): number => {
+  const message = findAssistant(messages, messageId);
+  if (!message?.parts) return 0;
+  let count = 0;
+  for (const part of message.parts) {
+    if (typeof part?.type === "string" && isToolUIPart(part)) count++;
+  }
+  return count;
+};
+
+const getAssistantMessageId = (
+  messages: readonly UIMessage[],
+): string | undefined => messages.findLast((m) => m.role === "assistant")?.id;
+
+export const aiSdkStreamingTimingAccessors: StreamingTimingAccessors<UIMessage> =
+  {
+    getAssistantMessageId,
+    getTextLength,
+    getToolCallCount,
+    getMessageContent: (messages, id) => findAssistant(messages, id)?.parts,
+  };
+
+/**
+ * Tracks streaming timing for AI SDK messages client-side. Delegates to the
+ * shared `useStreamingTiming` primitive in `@ulearning/ai-core/vue`,
+ * adapted to the `UIMessage` shape. Timing is finalized when streaming ends
+ * and stored per message id.
+ */
+export const useStreamingTiming = (
+  messages: MaybeRefOrGetter<readonly UIMessage[]>,
+  isRunning: MaybeRefOrGetter<boolean>,
+) =>
+  useStreamingTimingPrimitive(
+    messages,
+    isRunning,
+    aiSdkStreamingTimingAccessors,
+  );