@tuturuuu/ai 0.13.0 → 0.15.0

package/src/chat/google/mira-step-preparation.ts

@@ -3,17 +3,20 @@ import {
   buildActiveToolsFromSelected,
   countRenderUiAttemptsInSteps,
   extractSelectedToolsFromSteps,
+  getWorkspaceArtifactProgress,
   hasRenderableRenderUiInSteps,
   hasSuccessfulWorkspaceContextResolutionInSteps,
   hasToolCallInSteps,
   removeWorkspaceDiscoveryTools,
   wasToolEverSelectedInSteps,
 } from '../mira-render-ui-policy';
+import { getMiraToolLoopReason } from '../mira-tool-loop-guard';
 
 export type PrepareMiraToolStepInput = {
   steps: unknown[];
   forceGoogleSearch: boolean;
   forceRenderUi: boolean;
+  forceWorkspaceArtifact?: boolean;
   needsParallelChecks: boolean;
   needsWorkspaceContextResolution: boolean;
   needsWorkspaceMembersTool: boolean;
@@ -24,26 +27,41 @@ export function prepareMiraToolStep({
   steps,
   forceGoogleSearch,
   forceRenderUi,
+  forceWorkspaceArtifact,
   needsParallelChecks,
   needsWorkspaceContextResolution,
   needsWorkspaceMembersTool,
   preferMarkdownTables,
 }: PrepareMiraToolStepInput): {
-  toolChoice?: 'required';
+  toolChoice?: 'required' | 'none';
   activeTools: string[];
 } {
+  if (forceWorkspaceArtifact) forceRenderUi = false;
+  const artifact = getWorkspaceArtifactProgress(steps);
+  if (forceWorkspaceArtifact && !artifact.completed && artifact.attempts >= 3)
+    return { toolChoice: 'none', activeTools: [] };
+  if (
+    forceWorkspaceArtifact &&
+    !artifact.completed &&
+    (!needsWorkspaceContextResolution ||
+      hasSuccessfulWorkspaceContextResolutionInSteps(steps))
+  ) {
+    return { toolChoice: 'required', activeTools: ['show_workspace_artifact'] };
+  }
+  if (getMiraToolLoopReason(steps))
+    return { toolChoice: 'none', activeTools: [] };
   if (steps.length === 0) {
     if (forceGoogleSearch) {
       return {
         toolChoice: 'required',
-        activeTools: ['google_search', 'select_tools'],
+        activeTools: ['google_search'],
       };
     }
 
     if (needsParallelChecks) {
       return {
         toolChoice: 'required',
-        activeTools: ['run_parallel_checks', 'select_tools'],
+        activeTools: ['run_parallel_checks', 'search_tools', 'select_tools'],
       };
     }
 
@@ -54,6 +72,7 @@ export function prepareMiraToolStep({
           'list_accessible_workspaces',
           'get_workspace_context',
           'set_workspace_context',
+          'search_tools',
           'select_tools',
         ],
       };
@@ -65,13 +84,14 @@ export function prepareMiraToolStep({
         activeTools: [
           'get_workspace_context',
           'list_workspace_members',
+          'search_tools',
           'select_tools',
         ],
       };
     }
 
     return {
-      activeTools: ['select_tools', 'no_action_needed'],
+      activeTools: ['search_tools', 'select_tools', 'no_action_needed'],
     };
   }
 
@@ -106,11 +126,17 @@ export function prepareMiraToolStep({
     return {
       toolChoice: 'required',
       activeTools: hasListedAccessibleWorkspaces
-        ? ['get_workspace_context', 'set_workspace_context', 'select_tools']
+        ? [
+            'get_workspace_context',
+            'set_workspace_context',
+            'search_tools',
+            'select_tools',
+          ]
         : [
             'list_accessible_workspaces',
             'get_workspace_context',
             'set_workspace_context',
+            'search_tools',
             'select_tools',
           ],
     };
@@ -131,6 +157,7 @@ export function prepareMiraToolStep({
       'get_workspace_context',
       'list_workspace_members',
       ...selected,
+      'search_tools',
       'select_tools',
     ];
 
@@ -141,14 +168,7 @@ export function prepareMiraToolStep({
   }
 
   if (forceGoogleSearch && !hasToolCallInSteps(steps, 'google_search')) {
-    const active = buildActiveToolsFromSelected(toolsForBuild)
-      .filter((toolName) => toolName !== 'no_action_needed')
-      .concat('google_search', 'select_tools');
-
-    return {
-      toolChoice: 'required',
-      activeTools: Array.from(new Set(active)),
-    };
+    return { toolChoice: 'required', activeTools: ['google_search'] };
   }
 
   if (
@@ -157,7 +177,7 @@ export function prepareMiraToolStep({
   ) {
     const active = buildActiveToolsFromSelected(toolsForBuild)
       .filter((toolName) => toolName !== 'no_action_needed')
-      .concat('run_parallel_checks', 'select_tools');
+      .concat('run_parallel_checks', 'search_tools', 'select_tools');
 
     return {
       toolChoice: 'required',
@@ -178,6 +198,7 @@ export function prepareMiraToolStep({
           toolName !== 'select_tools' && toolName !== 'no_action_needed'
       ),
       'render_ui',
+      'search_tools',
       'select_tools',
     ];
 
@@ -199,7 +220,7 @@ export function prepareMiraToolStep({
   ) {
     const active = buildActiveToolsFromSelected(toolsForBuild)
       .filter((toolName) => toolName !== 'no_action_needed')
-      .concat('render_ui', 'select_tools');
+      .concat('render_ui', 'search_tools', 'select_tools');
     return {
       toolChoice: 'required',
       activeTools: Array.from(new Set(active)),
@@ -209,7 +230,7 @@ export function prepareMiraToolStep({
   if (hasRenderableRenderUiInSteps(steps)) {
     const active = buildActiveToolsFromSelected(toolsForBuild)
       .filter((toolName) => toolName !== 'render_ui')
-      .concat('select_tools');
+      .concat('search_tools', 'select_tools');
     return {
       activeTools: Array.from(new Set(active)),
     };

package/src/chat/google/route-chat-resolution.ts

@@ -59,10 +59,6 @@ export async function resolveChatIdForUser(
 }
 
 type MoveTempFilesToThreadParams = {
-  loadThread: () => PromiseLike<{
-    data: { role?: string }[] | null;
-    error: { message: string } | null;
-  }>;
   listFiles: (tempStoragePath: string) => PromiseLike<{
     data: { name: string }[] | null;
     error: AdminClientLike | null;
@@ -77,7 +73,6 @@ type MoveTempFilesToThreadParams = {
 };
 
 export async function moveTempFilesToThread({
-  loadThread,
   listFiles,
   moveFile,
   wsId,
@@ -88,30 +83,21 @@ export async function moveTempFilesToThread({
     return null;
   }
 
-  const { data: thread, error: threadError } = await loadThread();
-
-  if (threadError) {
-    console.error('Error getting thread:', threadError);
-    return new Response(threadError.message, { status: 500 });
-  }
-
-  if (!thread || thread.length === 0) {
-    return null;
-  }
-
   const tempStoragePath = `${wsId}/chats/ai/resources/temp/${userId}`;
   const { data: files, error: listError } = await listFiles(tempStoragePath);
 
   if (listError) {
     console.error('Error getting files:', listError);
-    return null;
+    return new Response('Could not load chat attachments. Please retry.', {
+      status: 502,
+    });
   }
 
   if (!files?.length) {
     return null;
   }
 
-  await Promise.all(
+  const moves = await Promise.all(
     files.map(async (file) => {
       const fileName = file.name;
       const { error: copyError } = await moveFile(
@@ -122,8 +108,15 @@ export async function moveTempFilesToThread({
       if (copyError) {
         console.error('File copy error:', { fileName, copyError });
       }
+      return copyError;
     })
   );
 
+  if (moves.some(Boolean)) {
+    return new Response('Could not attach every file. Please retry.', {
+      status: 502,
+    });
+  }
+
   return null;
 }

package/src/chat/google/route-credits.ts

@@ -44,15 +44,14 @@ export async function performCreditPreflight({
       )
     : null;
 
-  if (
-    cappedMaxOutput === null &&
-    creditCheck &&
-    creditCheck.remainingCredits <= 0
-  ) {
+  if (cappedMaxOutput === null) {
     return {
       error: Response.json(
-        { error: 'AI credits insufficient', code: 'CREDITS_EXHAUSTED' },
-        { status: 403 }
+        {
+          error: 'AI request budget could not be verified',
+          code: 'CREDIT_CHECK_FAILED',
+        },
+        { status: 503 }
       ),
     };
   }

package/src/chat/google/route-message-preparation.ts

@@ -377,3 +377,36 @@ export async function persistLatestUserMessage({
   console.log('User message saved to database');
   return null;
 }
+
+export function splitSystemMessages(messages: ModelMessage[]) {
+  const systemMessages: string[] = [];
+  const nonSystemMessages: ModelMessage[] = [];
+  for (const message of messages) {
+    if (message.role === 'system') {
+      const systemMessage = getTextContent(message.content);
+      if (systemMessage) systemMessages.push(systemMessage);
+      continue;
+    }
+    nonSystemMessages.push(message);
+  }
+  return {
+    messages: nonSystemMessages,
+    system: systemMessages.join('\n\n').trim(),
+  };
+}
+function getTextContent(content: ModelMessage['content']) {
+  if (typeof content === 'string') return content.trim();
+  if (!Array.isArray(content)) return '';
+  return content
+    .map((part) => (part.type === 'text' ? part.text.trim() : ''))
+    .filter(Boolean)
+    .join('\n')
+    .trim();
+}
+
+export function mergeSystemInstructions(...instructions: Array<string | null>) {
+  return instructions
+    .map((instruction) => instruction?.trim())
+    .filter(Boolean)
+    .join('\n\n');
+}

package/src/chat/google/route-mira-runtime.ts

@@ -1,19 +1,21 @@
 import type { TypedSupabaseClient } from '@tuturuuu/supabase/next/client';
+import type { PermissionId } from '@tuturuuu/types';
 import { normalizeWorkspaceContextId } from '@tuturuuu/utils/constants';
 import { getPermissions } from '@tuturuuu/utils/workspace-helper';
 import type { NextRequest } from 'next/server';
-import { buildMiraContext } from '../../tools/context-builder';
+import { createMiraSdkToolSearch } from '../../tools/mira-sdk-tool-search';
 import {
   createMiraStreamTools,
   type MiraToolContext,
 } from '../../tools/mira-tools';
 import type { MiraWorkspaceContextState } from '../../tools/workspace-context';
 import { resolveWorkspaceContextState } from '../../tools/workspace-context';
+import { buildMiraPrompt } from '../mira-prompt';
 import { buildMiraSystemInstruction } from '../mira-system-instruction';
 import type { ChatRequestTaskBoardContext } from './chat-request-schema';
 
 type PermissionResultLike = {
-  withoutPermission?: (permission: unknown) => boolean;
+  withoutPermission?: (permission: PermissionId) => boolean;
 };
 
 type SupabaseClientLike = TypedSupabaseClient;
@@ -171,6 +173,9 @@ export async function prepareMiraRuntime({
 }: PrepareMiraRuntimeParams): Promise<{
   miraSystemPrompt?: string;
   miraTools?: ReturnType<typeof createMiraStreamTools>;
+  prepareMiraDiscoveryStep?: ReturnType<
+    typeof createMiraSdkToolSearch
+  >['prepareStep'];
 }> {
   if (!isMiraMode || !wsId) {
     return {};
@@ -188,26 +193,29 @@ export async function prepareMiraRuntime({
     });
   } catch (workspaceContextErr) {
     console.error(
-      'Failed to resolve Mira workspace context, falling back to current workspace:',
+      'Failed to verify Mira workspace context:',
       workspaceContextErr
     );
-    resolvedWorkspaceContext = {
-      workspaceContextId: wsId,
-      wsId,
-      name: 'Current workspace',
-      personal: false,
-      memberCount: 0,
+    return {
+      miraSystemPrompt:
+        'You are Mira. Workspace access could not be verified for this request. Do not claim to read or change workspace data. Explain that the user can retry after access is restored; answer unrelated general questions normally.',
     };
   }
 
   let withoutPermission: PermissionResultLike['withoutPermission'];
   const denyPermissionByDefault = () => true;
   withoutPermission = denyPermissionByDefault;
+  const permissionCache = new Map<
+    string,
+    Promise<PermissionResultLike | null>
+  >();
   try {
-    const permissionsResult = (await getPermissions({
+    const pending = getPermissions({
       wsId: resolvedWorkspaceContext.wsId,
       ...(user ? { user } : { request }),
-    })) as PermissionResultLike | null;
+    });
+    permissionCache.set(resolvedWorkspaceContext.wsId, pending);
+    const permissionsResult = await pending;
     if (permissionsResult?.withoutPermission) {
       withoutPermission = permissionsResult.withoutPermission;
     }
@@ -223,36 +231,46 @@ export async function prepareMiraRuntime({
     chatId,
     supabase: miraSupabase,
     timezone,
+    requestHeaders: request.headers,
+    authorizeWorkspaceTools: async (targetWsId, permissions) => {
+      let pending = permissionCache.get(targetWsId);
+      if (!pending) {
+        pending = getPermissions({
+          wsId: targetWsId,
+          ...(user ? { user } : { request }),
+        });
+        permissionCache.set(targetWsId, pending);
+      }
+      try {
+        const resolved = await pending;
+        return (
+          !!resolved?.withoutPermission &&
+          permissions.every(
+            (permission) => !resolved.withoutPermission?.(permission)
+          )
+        );
+      } catch {
+        return false;
+      }
+    },
   };
 
   let miraSystemPrompt: string;
   try {
-    const { contextString, soul, isFirstInteraction } = await buildMiraContext({
+    const taskBoardInstruction = await buildTaskBoardContextInstruction({
+      resolvedWorkspaceContext,
+      supabase: miraSupabase,
+      taskBoardContext,
+    });
+    miraSystemPrompt = await buildMiraPrompt({
       userId,
       wsId: resolvedWorkspaceContext.wsId,
       supabase: miraSupabase,
       timezone,
       withoutPermission,
+      workspaceContext: resolvedWorkspaceContext,
+      taskBoardInstruction,
     });
-    const dynamicInstruction = buildMiraSystemInstruction({
-      soul,
-      isFirstInteraction,
-      withoutPermission,
-    });
-    const workspaceContextInstruction = `## Workspace Context\n\nCurrent task/calendar/finance workspace context: ${resolvedWorkspaceContext.name} (${resolvedWorkspaceContext.personal ? 'personal' : 'shared'} workspace).\nUse this workspace for "my tasks", "my calendar", and "my finance" requests. Only switch to another workspace when the user explicitly names a different workspace.`;
-    const taskBoardContextInstruction = await buildTaskBoardContextInstruction({
-      resolvedWorkspaceContext,
-      supabase: miraSupabase,
-      taskBoardContext,
-    });
-    miraSystemPrompt = [
-      contextString,
-      workspaceContextInstruction,
-      taskBoardContextInstruction,
-      dynamicInstruction,
-    ]
-      .filter(Boolean)
-      .join('\n\n');
   } catch (ctxErr) {
     console.error(
       'Failed to build Mira context (continuing with default instruction):',
@@ -261,8 +279,14 @@ export async function prepareMiraRuntime({
     miraSystemPrompt = buildMiraSystemInstruction({ withoutPermission });
   }
 
+  const discovery = createMiraSdkToolSearch(
+    createMiraStreamTools(ctx, withoutPermission, getSteps),
+    ctx,
+    withoutPermission
+  );
   return {
     miraSystemPrompt,
-    miraTools: createMiraStreamTools(ctx, withoutPermission, getSteps),
+    miraTools: discovery.tools,
+    prepareMiraDiscoveryStep: discovery.prepareStep,
   };
 }

package/src/chat/google/route.ts

@@ -4,13 +4,7 @@ import {
   normalizeWorkspaceId,
   verifyWorkspaceMembershipType,
 } from '@tuturuuu/utils/workspace-helper';
-import {
-  consumeStream,
-  type ModelMessage,
-  smoothStream,
-  stepCountIs,
-  streamText,
-} from 'ai';
+import { consumeStream, smoothStream, stepCountIs, streamText } from 'ai';
 import { type NextRequest, NextResponse } from 'next/server';
 import { normalizeStableModelId } from '../../credits/model-mapping';
 import {
@@ -18,8 +12,8 @@ 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 { shouldPresentWorkspaceArtifact } from '../mira-artifact-intent';
 import {
   shouldForceGoogleSearchForLatestUserMessage,
   shouldForceRenderUiForLatestUserMessage,
@@ -29,6 +23,7 @@ import {
   shouldUseParallelChecksForLatestUserMessage,
 } from '../mira-render-ui-policy';
 import { ChatRequestBodySchema, mapToUIMessages } from './chat-request-schema';
+import { resolveChatTools } from './chat-tools';
 import { systemInstruction } from './default-system-instruction';
 import { prepareMiraToolStep } from './mira-step-preparation';
 import { resolveChatReasoningSettings } from './reasoning-settings';
@@ -50,9 +45,11 @@ import {
 import { performCreditPreflight } from './route-credits';
 import {
   extractLatestUserMessageContent,
+  mergeSystemInstructions,
   persistLatestUserMessage,
   persistRequestScopedUserMessage,
   prepareProcessedMessages,
+  splitSystemMessages,
 } from './route-message-preparation';
 import { prepareMiraRuntime } from './route-mira-runtime';
 import {
@@ -60,44 +57,6 @@ import {
   persistAssistantResponse,
 } from './stream-finish-persistence';
 
-function splitSystemMessages(messages: ModelMessage[]) {
-  const systemMessages: string[] = [];
-  const nonSystemMessages: ModelMessage[] = [];
-
-  for (const message of messages) {
-    if (message.role === 'system') {
-      const systemMessage = getTextContent(message.content);
-      if (systemMessage) systemMessages.push(systemMessage);
-      continue;
-    }
-
-    nonSystemMessages.push(message);
-  }
-
-  return {
-    messages: nonSystemMessages,
-    system: systemMessages.join('\n\n').trim(),
-  };
-}
-
-function getTextContent(content: ModelMessage['content']) {
-  if (typeof content === 'string') return content.trim();
-  if (!Array.isArray(content)) return '';
-
-  return content
-    .map((part) => (part.type === 'text' ? part.text.trim() : ''))
-    .filter(Boolean)
-    .join('\n')
-    .trim();
-}
-
-function mergeSystemInstructions(...instructions: Array<string | null>) {
-  return instructions
-    .map((instruction) => instruction?.trim())
-    .filter(Boolean)
-    .join('\n\n');
-}
-
 export function createPOST(
   _options: {
     serverAPIKeyFallback?: boolean;
@@ -368,12 +327,6 @@ export function createPOST(
       const chatId = resolvedChatId.chatId;
 
       const moveFilesError = await moveTempFilesToThread({
-        loadThread: () =>
-          sbAdmin
-            .from('ai_chat_messages')
-            .select('role, ai_chats!chat_id!inner(creator_id)')
-            .eq('chat_id', chatId)
-            .eq('ai_chats.creator_id', user.id),
         listFiles: (tempStoragePath) =>
           sbAdmin.storage.from('workspaces').list(tempStoragePath),
         moveFile: (fromPath, toPath) =>
@@ -480,21 +433,22 @@ export function createPOST(
 
       const stepsRef: { current: unknown[] } = { current: [] };
 
-      const { miraSystemPrompt, miraTools } = await prepareMiraRuntime({
-        isMiraMode,
-        wsId: normalizedWsId ?? undefined,
-        workspaceContextId,
-        creditWsId: billingWsId ?? normalizedWsId ?? undefined,
-        request: req,
-        user,
-        userId: user.id,
-        chatId,
-        supabase,
-        toolSupabase: sbAdmin as typeof supabase,
-        timezone,
-        taskBoardContext,
-        getSteps: () => stepsRef.current,
-      });
+      const { miraSystemPrompt, miraTools, prepareMiraDiscoveryStep } =
+        await prepareMiraRuntime({
+          isMiraMode,
+          wsId: normalizedWsId ?? undefined,
+          workspaceContextId,
+          creditWsId: billingWsId ?? normalizedWsId ?? undefined,
+          request: req,
+          user,
+          userId: user.id,
+          chatId,
+          supabase,
+          toolSupabase: sbAdmin as typeof supabase,
+          timezone,
+          taskBoardContext,
+          getSteps: () => stepsRef.current,
+        });
 
       const effectiveSource = isMiraMode ? 'Mira' : 'Rewise';
 
@@ -509,6 +463,9 @@ export function createPOST(
       });
 
       const reasoningSettings = resolveChatReasoningSettings(thinkingMode);
+      const forceWorkspaceArtifact = shouldPresentWorkspaceArtifact(
+        promptMessages.messages
+      );
       const forceRenderUi = shouldForceRenderUiForLatestUserMessage(
         promptMessages.messages
       );
@@ -529,14 +486,13 @@ export function createPOST(
         promptMessages.messages
       );
 
-      const googleSearchTool = createGoogleSearchToolSet();
-
       type PrepareStep = NonNullable<
         NonNullable<Parameters<typeof streamText>[0]>['prepareStep']
       >;
-      const prepareStep: PrepareStep = ({ steps }) => {
+      const prepareStep: PrepareStep = async ({ steps }) => {
         stepsRef.current = steps;
-        return prepareMiraToolStep({
+        const policy = prepareMiraToolStep({
+          forceWorkspaceArtifact,
           steps,
           forceGoogleSearch,
           forceRenderUi,
@@ -545,6 +501,9 @@ export function createPOST(
           needsWorkspaceMembersTool,
           preferMarkdownTables,
         });
+        return prepareMiraDiscoveryStep
+          ? prepareMiraDiscoveryStep(policy)
+          : policy;
       };
 
       const { lease: claimedLease, response: leaseResponse } =
@@ -597,20 +556,16 @@ export function createPOST(
           promptMessages.system
         ),
         ...(cappedMaxOutput ? { maxOutputTokens: cappedMaxOutput } : {}),
+        tools: resolveChatTools(miraTools),
         ...(miraTools
           ? {
-              tools: { ...miraTools, ...googleSearchTool } as NonNullable<
-                Parameters<typeof streamText>[0]
-              >['tools'],
               stopWhen: stepCountIs(25),
               toolChoice: 'auto' as const,
               prepareStep: prepareStep as NonNullable<
                 NonNullable<Parameters<typeof streamText>[0]>['prepareStep']
               >,
             }
-          : {
-              tools: googleSearchTool,
-            }),
+          : {}),
         providerOptions: {
           google: {
             thinkingConfig: {