@tuturuuu/ai 0.2.1 → 0.3.0

package/README.md

@@ -61,6 +61,11 @@ yarn add @tuturuuu/ai
 bun add @tuturuuu/ai
 ```
 
+Published artifacts replace local `workspace:*` dependencies with exact
+registry versions before packing. If a public consumer cannot install this
+package, rerun the package release from the current production commit so its
+manifest points at published Tuturuuu package versions.
+
 ## Development
 
 ```bash

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@tuturuuu/ai",
   "license": "MIT",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "repository": {
     "type": "git",
     "url": "https://github.com/tutur3u/platform",
@@ -19,96 +19,96 @@
   ],
   "scripts": {
     "test": "vitest run",
-    "type-check": "tsgo --project tsconfig.typecheck.json"
+    "type-check": "tsc --project tsconfig.typecheck.json"
   },
   "dependencies": {
-    "@ai-sdk/amazon-bedrock": "^4.0.113",
-    "@ai-sdk/anthropic": "^3.0.81",
-    "@ai-sdk/azure": "^3.0.70",
-    "@ai-sdk/cerebras": "^2.0.53",
-    "@ai-sdk/cohere": "^3.0.36",
-    "@ai-sdk/deepgram": "^2.0.33",
-    "@ai-sdk/deepinfra": "^2.0.51",
-    "@ai-sdk/deepseek": "^2.0.34",
-    "@ai-sdk/elevenlabs": "^2.0.33",
-    "@ai-sdk/fal": "^2.0.34",
-    "@ai-sdk/fireworks": "^2.0.52",
-    "@ai-sdk/gateway": "^3.0.125",
-    "@ai-sdk/gladia": "^2.0.33",
-    "@ai-sdk/google": "^3.0.80",
-    "@ai-sdk/google-vertex": "^4.0.142",
-    "@ai-sdk/groq": "^3.0.39",
-    "@ai-sdk/hume": "^2.0.33",
-    "@ai-sdk/lmnt": "^2.0.33",
-    "@ai-sdk/luma": "^2.0.33",
-    "@ai-sdk/mistral": "^3.0.37",
-    "@ai-sdk/openai": "^3.0.68",
-    "@ai-sdk/openai-compatible": "^2.0.47",
-    "@ai-sdk/perplexity": "^3.0.33",
-    "@ai-sdk/react": "^3.0.199",
-    "@ai-sdk/replicate": "^2.0.33",
-    "@ai-sdk/revai": "^2.0.33",
-    "@ai-sdk/togetherai": "^2.0.52",
-    "@ai-sdk/xai": "^3.0.93",
+    "@ai-sdk/amazon-bedrock": "^5.0.24",
+    "@ai-sdk/anthropic": "^4.0.16",
+    "@ai-sdk/azure": "^4.0.17",
+    "@ai-sdk/cerebras": "^3.0.12",
+    "@ai-sdk/cohere": "^4.0.11",
+    "@ai-sdk/deepgram": "^3.0.11",
+    "@ai-sdk/deepinfra": "^3.0.12",
+    "@ai-sdk/deepseek": "^3.0.12",
+    "@ai-sdk/elevenlabs": "^3.0.12",
+    "@ai-sdk/fal": "^3.0.12",
+    "@ai-sdk/fireworks": "^3.0.13",
+    "@ai-sdk/gateway": "^4.0.23",
+    "@ai-sdk/gladia": "^3.0.11",
+    "@ai-sdk/google": "^4.0.18",
+    "@ai-sdk/google-vertex": "^5.0.22",
+    "@ai-sdk/groq": "^4.0.12",
+    "@ai-sdk/hume": "^3.0.11",
+    "@ai-sdk/lmnt": "^3.0.11",
+    "@ai-sdk/luma": "^3.0.12",
+    "@ai-sdk/mistral": "^4.0.13",
+    "@ai-sdk/openai": "^4.0.16",
+    "@ai-sdk/openai-compatible": "^3.0.12",
+    "@ai-sdk/perplexity": "^4.0.12",
+    "@ai-sdk/react": "^4.0.34",
+    "@ai-sdk/replicate": "^3.0.12",
+    "@ai-sdk/revai": "^3.0.11",
+    "@ai-sdk/togetherai": "^3.0.13",
+    "@ai-sdk/xai": "^4.0.16",
     "@beeper/chat-adapter-matrix": "^0.2.0",
     "@bitbasti/chat-adapter-webex": "^0.1.0",
-    "@chat-adapter/discord": "^4.30.0",
-    "@chat-adapter/gchat": "^4.30.0",
-    "@chat-adapter/github": "^4.30.0",
-    "@chat-adapter/linear": "^4.30.0",
-    "@chat-adapter/messenger": "^4.30.0",
-    "@chat-adapter/slack": "^4.30.0",
-    "@chat-adapter/state-ioredis": "^4.30.0",
-    "@chat-adapter/state-memory": "^4.30.0",
-    "@chat-adapter/state-pg": "^4.30.0",
-    "@chat-adapter/state-redis": "^4.30.0",
-    "@chat-adapter/teams": "^4.30.0",
-    "@chat-adapter/telegram": "^4.30.0",
-    "@chat-adapter/web": "^4.30.0",
-    "@chat-adapter/whatsapp": "^4.30.0",
+    "@chat-adapter/discord": "^4.34.0",
+    "@chat-adapter/gchat": "^4.34.0",
+    "@chat-adapter/github": "^4.34.0",
+    "@chat-adapter/linear": "^4.34.0",
+    "@chat-adapter/messenger": "^4.34.0",
+    "@chat-adapter/slack": "^4.34.0",
+    "@chat-adapter/state-ioredis": "^4.34.0",
+    "@chat-adapter/state-memory": "^4.34.0",
+    "@chat-adapter/state-pg": "^4.34.0",
+    "@chat-adapter/state-redis": "^4.34.0",
+    "@chat-adapter/teams": "^4.34.0",
+    "@chat-adapter/telegram": "^4.34.0",
+    "@chat-adapter/web": "^4.34.0",
+    "@chat-adapter/whatsapp": "^4.34.0",
     "@json-render/core": "^0.19.0",
     "@json-render/react": "^0.19.0",
-    "@liveblocks/chat-sdk-adapter": "^3.19.4",
+    "@liveblocks/chat-sdk-adapter": "^3.22.0",
     "@octokit/rest": "^22.0.1",
     "@resend/chat-sdk-adapter": "^0.2.2",
     "@streamdown/cjk": "^1.0.3",
     "@streamdown/code": "^1.1.1",
     "@streamdown/math": "^1.0.2",
     "@streamdown/mermaid": "^1.0.2",
-    "@tuturuuu/google": "0.0.1",
-    "@tuturuuu/internal-api": "0.4.1",
-    "@tuturuuu/supabase": "0.3.3",
-    "@tuturuuu/utils": "0.5.1",
-    "@vercel/sandbox": "^2.1.1",
-    "@zernio/chat-sdk-adapter": "^0.3.0",
-    "ai": "^6.0.197",
-    "bash-tool": "^1.3.17",
-    "chat": "^4.30.0",
+    "@tuturuuu/google": "0.0.3",
+    "@tuturuuu/internal-api": "0.20.0",
+    "@tuturuuu/supabase": "0.4.1",
+    "@tuturuuu/utils": "0.17.0",
+    "@vercel/sandbox": "^2.7.1",
+    "@zernio/chat-sdk-adapter": "^0.4.0",
+    "ai": "^7.0.31",
+    "bash-tool": "^1.3.18",
+    "chat": "^4.34.0",
     "chat-adapter-baileys": "^2.0.3",
     "chat-adapter-blooio": "^0.1.0",
-    "chat-adapter-imessage": "^0.1.1",
+    "chat-adapter-imessage": "^1.1.0",
     "chat-adapter-mattermost": "^1.1.3",
     "chat-adapter-sendblue": "^0.2.0",
     "chat-adapter-zalo": "^0.1.0",
     "chat-state-cloudflare-do": "^0.2.0",
     "chat-state-mysql": "^0.1.0",
     "dayjs": "^1.11.20",
-    "next": "^16.2.7",
+    "next": "^16.3.0-preview.6",
     "qrcode": "^1.5.4",
     "react": "^19.2.7",
     "react-dom": "^19.2.7",
     "streamdown": "^2.5.0",
-    "uuid": "^14.0.0",
+    "uuid": "^14.0.1",
     "zca-js": "^2.1.2",
     "zod": "^4.4.3"
   },
   "devDependencies": {
-    "@tuturuuu/types": "0.6.1",
+    "@tuturuuu/types": "0.19.0",
     "@tuturuuu/typescript-config": "0.1.1",
-    "@types/node": "^25.9.2",
+    "@types/node": "^26.1.1",
     "@types/qrcode": "^1.5.6",
     "@types/react": "^19.2.17",
-    "typescript": "^6.0.3"
+    "typescript": "7.0.2"
   },
   "exports": {
     "./*": "./src/*.ts"

package/src/chat/google/route.ts

@@ -18,6 +18,7 @@ import {
   resolvePlanModel,
 } from '../../credits/resolve-plan-model';
 import { withAiMemory } from '../../memory';
+import { createGoogleSearchToolSet } from '../../tools/google-search-tool';
 import type { CreditSource as SharedCreditSource } from '../credit-source';
 import {
   shouldForceGoogleSearchForLatestUserMessage,
@@ -495,9 +496,7 @@ export function createPOST(
       );
 
       // Provider-native Google Search tool for non-Mira mode.
-      const googleSearchTool = {
-        google_search: google.tools.googleSearch({}),
-      };
+      const googleSearchTool = createGoogleSearchToolSet();
 
       type PrepareStep = NonNullable<
         NonNullable<Parameters<typeof streamText>[0]>['prepareStep']

package/src/chat-sdk/registry.ts

@@ -83,18 +83,12 @@ async function loadChatSdkFactory(
 }
 
 function isAdapter(value: unknown): value is Adapter {
-  return (
-    typeof value === 'object' &&
-    value !== null &&
-    'handleWebhook' in value &&
-    'name' in value
-  );
+  return value instanceof Object && 'handleWebhook' in value && 'name' in value;
 }
 
 function isStateAdapter(value: unknown): value is StateAdapter {
   return (
-    typeof value === 'object' &&
-    value !== null &&
+    value instanceof Object &&
     'get' in value &&
     'set' in value &&
     'delete' in value

package/src/chat-sdk/zalo-personal.ts

@@ -1,6 +1,7 @@
 import {
   type Adapter,
   type AdapterPostableMessage,
+  type Attachment,
   type ChatInstance,
   type FetchOptions,
   type FetchResult,
@@ -163,6 +164,7 @@ export interface ZaloPersonalPhoneSyncOptions {
   maxPulls?: number;
   minSeqId?: number;
   pullDelayMs?: number;
+  signal?: AbortSignal;
   tempKey?: string;
   useListenerWakeup?: boolean;
 }
@@ -211,8 +213,8 @@ const DEFAULT_HISTORY_MAX_PAGES_PER_TYPE = 1000;
 const DEFAULT_HISTORY_MESSAGES_PER_GROUP = 200;
 const DEFAULT_HISTORY_PAGE_TIMEOUT_MS = 8000;
 const HISTORY_CONNECT_TIMEOUT_MS = 5000;
-const DEFAULT_PHONE_SYNC_MAX_PULLS = 90;
-const DEFAULT_PHONE_SYNC_PULL_DELAY_MS = 1000;
+const DEFAULT_PHONE_SYNC_MAX_PULLS = 4;
+const DEFAULT_PHONE_SYNC_PULL_DELAY_MS = 20_000;
 const PHONE_SYNC_PC_NAME_MAX_LENGTH = 80;
 const ZALO_LISTENER_SOCKET_OPEN_STATE = 1;
 const ZALO_PHONE_SYNC_REQUEST_CMD = 590;
@@ -1023,15 +1025,16 @@ export function createZaloPersonalAdapter(
         });
       }
 
+      const attachments = extractZaloPersonalAttachments(raw, config);
       const text =
         typeof raw.data.content === 'string'
           ? raw.data.content
-          : '[Unsupported Zalo message]';
+          : zaloAttachmentFallbackText(raw.data.msgType, attachments);
       const authorId =
         raw.data.uidFrom || raw.data.userId || (raw.isSelf ? status.ownId : '');
 
       return new Message<ZaloPersonalRawMessage>({
-        attachments: [],
+        attachments,
         author: {
           fullName: raw.data.dName || authorId || 'Zalo user',
           isBot: raw.isSelf,
@@ -1245,6 +1248,8 @@ export function createZaloPersonalAdapter(
       const useListenerWakeup = options.useListenerWakeup ?? true;
 
       try {
+        assertPhoneSyncNotAborted(options.signal);
+
         if (useListenerWakeup) {
           attachListeners(apiInstance);
           await adapter.startPersonalListener();
@@ -1276,9 +1281,8 @@ export function createZaloPersonalAdapter(
         }
 
         for (let attempt = 0; attempt < maxPulls; attempt += 1) {
-          if (attempt > 0) {
-            await delay(pullDelayMs);
-          }
+          await delay(pullDelayMs, options.signal);
+          assertPhoneSyncNotAborted(options.signal);
 
           try {
             const pullResponse = await transferApi.tuturuuuPullMobileMessages?.(
@@ -1332,7 +1336,11 @@ export function createZaloPersonalAdapter(
           }
         }
       } catch (error) {
-        lastError = error instanceof Error ? error.message : String(error);
+        lastError = options.signal?.aborted
+          ? 'zalo_personal_phone_sync_cancelled'
+          : error instanceof Error
+            ? error.message
+            : String(error);
       } finally {
         if (requestAccepted && messages.length > 0) {
           cleaned =
@@ -2124,8 +2132,229 @@ function stringValue(value: unknown) {
   return '';
 }
 
-function delay(ms: number) {
-  return new Promise((resolve) => setTimeout(resolve, ms));
+const ZALO_MEDIA_URL_KEYS = [
+  'downloadUrl',
+  'fileUrl',
+  'hdUrl',
+  'href',
+  'normalUrl',
+  'originUrl',
+  'thumb',
+  'thumbnail',
+  'thumbUrl',
+  'url',
+  'voiceUrl',
+] as const;
+
+function extractZaloPersonalAttachments(
+  raw: ZcaMessage,
+  config: ZaloPersonalAdapterConfig
+): Attachment[] {
+  const content = normalizeZaloAttachmentContent(raw.data.content);
+  if (!content) return [];
+
+  const urls = collectZaloMediaUrls(content);
+  if (urls.length === 0) return [];
+
+  const type = zaloAttachmentType(raw.data.msgType);
+  const preferredUrl = pickPreferredZaloMediaUrl(urls);
+  const name = zaloAttachmentFilename(content, preferredUrl, type);
+  const mimeType = inferZaloAttachmentMimeType(name, type);
+  const size = numericValue(
+    content.fileSize ?? content.file_size ?? content.size ?? content.totalSize
+  );
+
+  return [
+    {
+      fetchData: async () => {
+        const response = await fetch(preferredUrl, {
+          headers: {
+            Cookie: zaloCookieHeader(config.cookieJson),
+            'User-Agent': config.userAgent,
+          },
+          signal: AbortSignal.timeout(30_000),
+        });
+        if (!response.ok) {
+          throw new Error(
+            `zalo_personal_media_fetch_failed_${response.status}`
+          );
+        }
+        const declaredSize = Number(response.headers.get('content-length'));
+        if (Number.isFinite(declaredSize) && declaredSize > 25 * 1024 * 1024) {
+          throw new Error('zalo_personal_media_too_large');
+        }
+        return Buffer.from(await response.arrayBuffer());
+      },
+      height: numericValue(content.height),
+      mimeType,
+      name,
+      size,
+      type,
+      url: preferredUrl,
+      width: numericValue(content.width),
+    },
+  ];
+}
+
+function normalizeZaloAttachmentContent(value: unknown) {
+  if (isRecord(value)) return value;
+  if (typeof value !== 'string') return null;
+
+  const trimmed = value.trim();
+  if (!trimmed.startsWith('{')) return null;
+
+  try {
+    const parsed = JSON.parse(trimmed);
+    return isRecord(parsed) ? parsed : null;
+  } catch {
+    return null;
+  }
+}
+
+function collectZaloMediaUrls(value: unknown) {
+  const urls: string[] = [];
+
+  visitRecordValues(value, (key, candidate) => {
+    if (
+      ZALO_MEDIA_URL_KEYS.includes(
+        key as (typeof ZALO_MEDIA_URL_KEYS)[number]
+      ) &&
+      typeof candidate === 'string' &&
+      /^https?:\/\//u.test(candidate)
+    ) {
+      urls.push(candidate);
+    }
+  });
+
+  return [...new Set(urls)];
+}
+
+function pickPreferredZaloMediaUrl(urls: string[]) {
+  return (
+    urls.find((url) => /(?:hd|origin|download|file)/iu.test(url)) ?? urls[0]!
+  );
+}
+
+function zaloAttachmentType(msgType: string): Attachment['type'] {
+  if (/voice|audio/iu.test(msgType)) return 'audio';
+  if (/video/iu.test(msgType)) return 'video';
+  if (/photo|image|sticker|gif|doodle/iu.test(msgType)) return 'image';
+  return 'file';
+}
+
+function zaloAttachmentFilename(
+  content: Record<string, unknown>,
+  url: string,
+  type: Attachment['type']
+) {
+  const explicit =
+    stringValue(content.fileName) ||
+    stringValue(content.filename) ||
+    stringValue(content.title);
+  if (explicit) return explicit.slice(0, 240);
+
+  try {
+    const pathname = new URL(url).pathname;
+    const candidate = decodeURIComponent(pathname.split('/').at(-1) ?? '');
+    if (candidate?.includes('.')) return candidate.slice(0, 240);
+  } catch {
+    // Fall through to a stable generic name.
+  }
+
+  const extension =
+    type === 'image'
+      ? 'jpg'
+      : type === 'video'
+        ? 'mp4'
+        : type === 'audio'
+          ? 'm4a'
+          : 'bin';
+  return `zalo-${type}.${extension}`;
+}
+
+function inferZaloAttachmentMimeType(name: string, type: Attachment['type']) {
+  const extension = name.split('.').at(-1)?.toLowerCase();
+  const byExtension: Record<string, string> = {
+    gif: 'image/gif',
+    jpeg: 'image/jpeg',
+    jpg: 'image/jpeg',
+    m4a: 'audio/mp4',
+    mov: 'video/quicktime',
+    mp3: 'audio/mpeg',
+    mp4: 'video/mp4',
+    ogg: 'audio/ogg',
+    pdf: 'application/pdf',
+    png: 'image/png',
+    webm: 'video/webm',
+    webp: 'image/webp',
+  };
+
+  return (
+    (extension && byExtension[extension]) ||
+    (type === 'file' ? 'application/octet-stream' : `${type}/*`)
+  );
+}
+
+function zaloAttachmentFallbackText(
+  msgType: string,
+  attachments: Attachment[]
+) {
+  if (attachments.length === 0) return '[Unsupported Zalo message]';
+  const type = zaloAttachmentType(msgType);
+  return `[Zalo ${type}]`;
+}
+
+function numericValue(value: unknown) {
+  const numeric =
+    typeof value === 'number'
+      ? value
+      : typeof value === 'string'
+        ? Number.parseInt(value, 10)
+        : Number.NaN;
+  return Number.isFinite(numeric) && numeric > 0 ? numeric : undefined;
+}
+
+function zaloCookieHeader(cookieJson: string) {
+  try {
+    const parsed = JSON.parse(cookieJson) as unknown;
+    if (!Array.isArray(parsed)) return '';
+    return parsed
+      .filter(isRecord)
+      .map((cookie) => {
+        const name = stringValue(cookie.name);
+        const value = stringValue(cookie.value);
+        return name && value ? `${name}=${value}` : '';
+      })
+      .filter(Boolean)
+      .join('; ');
+  } catch {
+    return '';
+  }
+}
+
+function delay(ms: number, signal?: AbortSignal) {
+  if (signal?.aborted) {
+    return Promise.reject(new Error('zalo_personal_phone_sync_cancelled'));
+  }
+
+  return new Promise<void>((resolve, reject) => {
+    const timer = setTimeout(() => {
+      signal?.removeEventListener('abort', onAbort);
+      resolve();
+    }, ms);
+    const onAbort = () => {
+      clearTimeout(timer);
+      reject(new Error('zalo_personal_phone_sync_cancelled'));
+    };
+
+    signal?.addEventListener('abort', onAbort, { once: true });
+  });
+}
+
+function assertPhoneSyncNotAborted(signal?: AbortSignal) {
+  if (signal?.aborted) {
+    throw new Error('zalo_personal_phone_sync_cancelled');
+  }
 }
 
 function dateFromZaloTimestamp(value: string) {

package/src/effect.ts

@@ -0,0 +1,66 @@
+import type {
+  CreditCheckResult,
+  CreditDeductionResult,
+  DeductCreditsParams,
+} from '@tuturuuu/ai/credits/types';
+import {
+  Context,
+  Effect,
+  fromPromise,
+  Layer,
+  type TuturuuuEffectError,
+} from '@tuturuuu/utils/effect';
+import { checkAiCredits, deductAiCredits } from './credits/check-credits';
+import type { AiFeature } from './credits/constants';
+
+export interface AiCreditsServiceShape {
+  readonly checkAiCredits: typeof checkAiCredits;
+  readonly deductAiCredits: typeof deductAiCredits;
+}
+
+export class AiCreditsService extends Context.Tag(
+  '@tuturuuu/ai/AiCreditsService'
+)<AiCreditsService, AiCreditsServiceShape>() {}
+
+export const AiCreditsLive = Layer.succeed(AiCreditsService, {
+  checkAiCredits,
+  deductAiCredits,
+});
+
+export function checkAiCreditsEffect(
+  wsId: string | undefined,
+  modelId: string,
+  feature: AiFeature,
+  opts?: { userId?: string; estimatedInputTokens?: number }
+): Effect.Effect<CreditCheckResult, TuturuuuEffectError, AiCreditsService> {
+  return Effect.gen(function* () {
+    const service = yield* AiCreditsService;
+
+    return yield* fromPromise(
+      () => service.checkAiCredits(wsId, modelId, feature, opts),
+      {
+        code: 'AI_CREDIT_CHECK_FAILED',
+        message: 'AI credit check failed.',
+        context: { feature, modelId, wsId },
+      }
+    );
+  });
+}
+
+export function deductAiCreditsEffect(
+  params: DeductCreditsParams
+): Effect.Effect<CreditDeductionResult, TuturuuuEffectError, AiCreditsService> {
+  return Effect.gen(function* () {
+    const service = yield* AiCreditsService;
+
+    return yield* fromPromise(() => service.deductAiCredits(params), {
+      code: 'AI_CREDIT_DEDUCTION_FAILED',
+      message: 'AI credit deduction failed.',
+      context: {
+        feature: params.feature,
+        modelId: params.modelId,
+        wsId: params.wsId,
+      },
+    });
+  });
+}

package/src/generate/route.ts

@@ -259,7 +259,7 @@ export function createPOST(
             usage: {
               inputTokens: usage.inputTokens ?? 0,
               outputTokens: usage.outputTokens ?? 0,
-              reasoningTokens: usage.reasoningTokens ?? 0,
+              reasoningTokens: usage.outputTokenDetails?.reasoningTokens ?? 0,
               totalTokens: usage.totalTokens ?? 0,
             },
             finishReason,
@@ -274,7 +274,7 @@ export function createPOST(
             finish_reason: String(finishReason),
             input_tokens: usage.inputTokens ?? 0,
             output_tokens: usage.outputTokens ?? 0,
-            reasoning_tokens: usage.reasoningTokens ?? 0,
+            reasoning_tokens: usage.outputTokenDetails?.reasoningTokens ?? 0,
             total_tokens: usage.totalTokens ?? 0,
             system_prompt: configs.systemPrompt ?? '',
           };
@@ -296,10 +296,7 @@ export function createPOST(
             modelId: effectiveModel,
             inputTokens: usage.inputTokens ?? 0,
             outputTokens: usage.outputTokens ?? 0,
-            reasoningTokens:
-              usage.outputTokenDetails?.reasoningTokens ??
-              usage.reasoningTokens ??
-              0,
+            reasoningTokens: usage.outputTokenDetails?.reasoningTokens ?? 0,
             feature: 'generate',
             executionId: execData?.id,
           }).catch((err) => console.error('Failed to deduct AI credits:', err));

package/src/mind/route.ts

@@ -5,7 +5,13 @@ import {
   normalizeWorkspaceId,
   verifyWorkspaceMembershipType,
 } from '@tuturuuu/utils/workspace-helper';
-import { consumeStream, gateway, stepCountIs, streamText } from 'ai';
+import {
+  consumeStream,
+  gateway,
+  type LanguageModelUsage,
+  stepCountIs,
+  streamText,
+} from 'ai';
 import { type NextRequest, NextResponse } from 'next/server';
 import { z } from 'zod';
 import type { CreditSource as SharedCreditSource } from '../chat/credit-source';
@@ -31,6 +37,11 @@ import { createMiraStreamTools } from '../tools/mira-tools';
 import { createMindStreamTools, type MindToolCallbacks } from './tools';
 
 type AuthOk = Extract<AiRouteAuthResult, { ok: true }>;
+type MindUsage = Partial<
+  Pick<LanguageModelUsage, 'inputTokens' | 'outputTokens' | 'totalTokens'>
+> & {
+  outputTokenDetails?: Partial<LanguageModelUsage['outputTokenDetails']>;
+};
 
 const MindChatBodySchema = z.object({
   boardId: z.guid().nullable().optional(),
@@ -589,22 +600,34 @@ export function createPOST(callbacks: MindRouteCallbacks) {
         userId: auth.user.id,
         wsId: access.wsId,
       });
+      const telemetryRuntimeContext = {
+        boardId: parsedBody.boardId ?? 'none',
+        clientRunId: parsedBody.clientRunId ?? 'none',
+        creditSource: requestedCreditSource,
+        creditWsId: billingWsId ?? access.wsId,
+        model: resolvedModelId,
+        threadId,
+        thinkingMode: parsedBody.thinkingMode ?? 'fast',
+        writeMode,
+        wsId: access.wsId,
+      };
 
       const result = streamText({
         abortSignal: request.signal,
-        experimental_telemetry: {
+        runtimeContext: telemetryRuntimeContext,
+        telemetry: {
           functionId: 'mind.ai.stream',
           isEnabled: true,
-          metadata: {
-            boardId: parsedBody.boardId ?? 'none',
-            clientRunId: parsedBody.clientRunId ?? 'none',
-            creditSource: requestedCreditSource,
-            creditWsId: billingWsId ?? access.wsId,
-            model: resolvedModelId,
-            threadId,
-            thinkingMode: parsedBody.thinkingMode ?? 'fast',
-            writeMode,
-            wsId: access.wsId,
+          includeRuntimeContext: {
+            boardId: true,
+            clientRunId: true,
+            creditSource: true,
+            creditWsId: true,
+            model: true,
+            threadId: true,
+            thinkingMode: true,
+            writeMode: true,
+            wsId: true,
           },
         },
         maxRetries: 0,
@@ -636,7 +659,7 @@ export function createPOST(callbacks: MindRouteCallbacks) {
           const toolResults = response.steps?.flatMap(
             (step) => step.toolResults ?? []
           );
-          const usage = response.totalUsage ?? response.usage ?? {};
+          const usage: MindUsage = response.totalUsage ?? response.usage ?? {};
           await callbacks.persistMessage({
             boardId: parsedBody.boardId ?? null,
             content: response.text || '',
@@ -653,7 +676,7 @@ export function createPOST(callbacks: MindRouteCallbacks) {
             usage: {
               inputTokens: usage.inputTokens ?? 0,
               outputTokens: usage.outputTokens ?? 0,
-              reasoningTokens: usage.reasoningTokens ?? 0,
+              reasoningTokens: usage.outputTokenDetails?.reasoningTokens ?? 0,
               totalTokens: usage.totalTokens ?? 0,
             },
             userId: auth.user.id,
@@ -664,7 +687,7 @@ export function createPOST(callbacks: MindRouteCallbacks) {
             inputTokens: usage.inputTokens ?? 0,
             modelId: resolvedModelId,
             outputTokens: usage.outputTokens ?? 0,
-            reasoningTokens: usage.reasoningTokens ?? 0,
+            reasoningTokens: usage.outputTokenDetails?.reasoningTokens ?? 0,
             userId: auth.user.id,
             wsId: billingWsId ?? access.wsId,
           });

package/src/scheduling/duration-optimizer.ts

@@ -542,12 +542,9 @@ export function calculateIdealStartTimeForTask(
   now: Date
 ): Date {
   // Effective slot start (must be >= now, rounded to next 15-min)
-  let effectiveSlotStart = new Date(slot.start);
-  if (now > slot.start) {
-    effectiveSlotStart = roundToNext15Minutes(now);
-  } else {
-    effectiveSlotStart = roundToNext15Minutes(slot.start);
-  }
+  const effectiveSlotStart = roundToNext15Minutes(
+    now > slot.start ? now : slot.start
+  );
 
   // Latest possible start time to fit the duration
   const latestStart = new Date(slot.end.getTime() - duration * 60000);

package/src/tools/executors/parallel-checks.ts

@@ -123,7 +123,7 @@ function extractMeteredUsage(usage: LanguageModelUsage): MeteredUsage {
     inputTokens: normalizeTokenCount(usage.inputTokens),
     outputTokens: normalizeTokenCount(usage.outputTokens),
     reasoningTokens: normalizeTokenCount(
-      usage.outputTokenDetails.reasoningTokens ?? usage.reasoningTokens
+      usage.outputTokenDetails.reasoningTokens
     ),
   };
 }

package/src/tools/executors/search.ts

@@ -2,6 +2,7 @@ import { google } from '@ai-sdk/google';
 import { generateText, stepCountIs } from 'ai';
 import { z } from 'zod';
 import { withAiMemory } from '../../memory';
+import { createGoogleSearchToolSet } from '../google-search-tool';
 import type { MiraToolContext } from '../mira-tools';
 
 const SEARCH_WRAPPER_MODEL = 'gemini-3.1-flash-lite';
@@ -85,9 +86,7 @@ async function runGoogleSearchWrapper(
       userId: ctx.userId,
       wsId: ctx.workspaceContext?.wsId ?? ctx.wsId,
     }),
-    tools: {
-      google_search: google.tools.googleSearch({}),
-    },
+    tools: createGoogleSearchToolSet(),
     prompt,
     stopWhen: stepCountIs(4),
     ...(forceTool ? { toolChoice: 'required' as const } : {}),

package/src/tools/google-search-tool.ts

@@ -0,0 +1,10 @@
+import { google } from '@ai-sdk/google';
+import type { ToolSet } from 'ai';
+
+export function createGoogleSearchToolSet(): ToolSet {
+  return {
+    // Google executes this provider tool itself; AI SDK's public ToolSet type
+    // still narrows provider tools through the local-tool generic union.
+    google_search: google.tools.googleSearch({}) as unknown as ToolSet[string],
+  };
+}