Mastra (AI SDK v5 ベータ) で Gemini の Grounding with Google Search と共存したかった

  const res = await generateText({
      model: google(model),
      providerOptions: {
        google: {
          safetySettings: BLOCK_NONE_SAFETY_SETTINGS, // 他のところで定義した安全設定
        },
      },
      tools: {
        google_search: google.tools.googleSearch({}),
      },
      prompt: `
知識カットオフ: 2025年1月
現在の日付: ${todayDateText()}
---
${defaultSearchPrompt}
---
目的: ${objective}
`,
  });

import { createTool } from "@mastra/core";
import { generateText } from "ai";
import z from "zod";
import { BLOCK_NONE_SAFETY_SETTINGS, google } from "../gemini";
import { todayDateText } from "../../util";

const defaultSearchPrompt = `
与えられる目的を達成するように、Web 検索を行って情報を収集し適切に回答してください。
検索を拒否してはならず、必ず指定された目的を達成するために検索を行い、得られた結果について報告をすること。
`.trim();

const ProviderMetadataSchema = z.object({
  groundingMetadata: z.object({
    webSearchQueries: z.array(z.string()),
    searchEntryPoint: z.object({
      renderedContent: z.string(),
    }),
    groundingChunks: z
      .array(
        z.object({
          web: z.object({
            uri: z.string().url(),
            title: z.string(),
          }),
        })
      )
      .nullable(),
    groundingSupports: z
      .array(
        z.object({
          segment: z.object({
            startIndex: z.number().optional(),
            endIndex: z.number(),
            text: z.string(),
          }),
          groundingChunkIndices: z.array(z.number()),
        })
      )
      .nullable(),
  }),
  usageMetadata: z.object({
    thoughtsTokenCount: z.number(),
    promptTokenCount: z.number(),
    candidatesTokenCount: z.number(),
    totalTokenCount: z.number(),
  }),
});

export const lowLevelGeminiSearch = createTool({
  id: "low-level-google-search",
  description: "Search the web with query.",
  inputSchema: z.object({
    objective: z
      .string(),
    model: z
      .enum([
        "gemini-2.5-pro",
        "gemini-2.5-flash",
        "gemini-2.5-flash-lite-preview-06-17",
      ])
      .optional()
      .default("gemini-2.5-flash"),
  }),
  outputSchema: z.object({
    text: z.string(),
    sources: z.array(z.string()),
    queries: z
      .array(z.string())
      .optional(),
  }),
  execute: async ({ context }, option) => {
    const { objective, model } = context;

    const res = await generateText({
      model: google(model),
      providerOptions: {
        google: {
          safetySettings: BLOCK_NONE_SAFETY_SETTINGS,
        },
      },
      tools: {
        google_search: google.tools.googleSearch({}),
      },
      prompt: `
知識カットオフ: 2025年1月
現在の日付: ${todayDateText()}
---
${defaultSearchPrompt}
---
目的: ${objective}
`,
    });

    const metadata = ProviderMetadataSchema.safeParse(
      res.providerMetadata?.google || {}
    );

    return {
      text: res.text,
      sources: res.sources.map((source) => {
        switch (source.sourceType) {
          case "url": {
            return source.url;
          }
          default: {
            return source.filename || source.title || "";
          }
        }
      }),
      queries: metadata.data?.groundingMetadata.webSearchQueries,
    };
  },
});