@depup/ai-sdk__google 3.0.43-depup.0

package/src/tool/file-search.ts

@@ -0,0 +1,51 @@
+import {
+  createProviderToolFactory,
+  lazySchema,
+  zodSchema,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+/** Tool to retrieve knowledge from the File Search Stores. */
+const fileSearchArgsBaseSchema = z
+  .object({
+    /** The names of the file_search_stores to retrieve from.
+     *  Example: `fileSearchStores/my-file-search-store-123`
+     */
+    fileSearchStoreNames: z
+      .array(z.string())
+      .describe(
+        'The names of the file_search_stores to retrieve from. Example: `fileSearchStores/my-file-search-store-123`',
+      ),
+    /** The number of file search retrieval chunks to retrieve. */
+    topK: z
+      .number()
+      .int()
+      .positive()
+      .describe('The number of file search retrieval chunks to retrieve.')
+      .optional(),
+
+    /** Metadata filter to apply to the file search retrieval documents.
+     *  See https://google.aip.dev/160 for the syntax of the filter expression.
+     */
+    metadataFilter: z
+      .string()
+      .describe(
+        'Metadata filter to apply to the file search retrieval documents. See https://google.aip.dev/160 for the syntax of the filter expression.',
+      )
+      .optional(),
+  })
+  .passthrough();
+
+export type GoogleFileSearchToolArgs = z.infer<typeof fileSearchArgsBaseSchema>;
+
+const fileSearchArgsSchema = lazySchema(() =>
+  zodSchema(fileSearchArgsBaseSchema),
+);
+
+export const fileSearch = createProviderToolFactory<
+  {},
+  GoogleFileSearchToolArgs
+>({
+  id: 'google.file_search',
+  inputSchema: fileSearchArgsSchema,
+});

package/src/tool/google-maps.ts

@@ -0,0 +1,14 @@
+import {
+  createProviderToolFactory,
+  lazySchema,
+  zodSchema,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+// https://ai.google.dev/gemini-api/docs/maps-grounding
+// https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/grounding-with-google-maps
+
+export const googleMaps = createProviderToolFactory<{}, {}>({
+  id: 'google.google_maps',
+  inputSchema: lazySchema(() => zodSchema(z.object({}))),
+});

package/src/tool/google-search.ts

@@ -0,0 +1,43 @@
+import {
+  createProviderToolFactory,
+  lazySchema,
+  zodSchema,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+// https://ai.google.dev/gemini-api/docs/google-search
+// https://ai.google.dev/api/generate-content#GroundingSupport
+// https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/grounding-with-google-search
+
+const googleSearchToolArgsBaseSchema = z
+  .object({
+    searchTypes: z
+      .object({
+        webSearch: z.object({}).optional(),
+        imageSearch: z.object({}).optional(),
+      })
+      .optional(),
+
+    timeRangeFilter: z
+      .object({
+        startTime: z.string(),
+        endTime: z.string(),
+      })
+      .optional(),
+  })
+  .passthrough();
+
+export type GoogleSearchToolArgs = z.infer<
+  typeof googleSearchToolArgsBaseSchema
+>;
+
+const googleSearchToolArgsSchema = lazySchema(() =>
+  zodSchema(googleSearchToolArgsBaseSchema),
+);
+
+export const googleSearch = createProviderToolFactory<{}, GoogleSearchToolArgs>(
+  {
+    id: 'google.google_search',
+    inputSchema: googleSearchToolArgsSchema,
+  },
+);

package/src/tool/url-context.ts

@@ -0,0 +1,16 @@
+import {
+  createProviderToolFactory,
+  lazySchema,
+  zodSchema,
+} from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+export const urlContext = createProviderToolFactory<
+  {
+    // Url context does not have any input schema, it will directly use the url from the prompt
+  },
+  {}
+>({
+  id: 'google.url_context',
+  inputSchema: lazySchema(() => zodSchema(z.object({}))),
+});

package/src/tool/vertex-rag-store.ts

@@ -0,0 +1,31 @@
+import { createProviderToolFactory } from '@ai-sdk/provider-utils';
+import { z } from 'zod/v4';
+
+// https://cloud.google.com/vertex-ai/generative-ai/docs/rag-engine/use-vertexai-search#generate-content-using-gemini-api
+// https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/rag-output-explained
+
+/**
+ * A tool that enables the model to perform RAG searches against a Vertex RAG Store.
+ *
+ * @note Only works with Vertex Gemini models.
+ */
+export const vertexRagStore = createProviderToolFactory<
+  {},
+  {
+    /**
+     * RagCorpus resource names, eg: projects/{project}/locations/{location}/ragCorpora/{rag_corpus}
+     */
+    ragCorpus: string;
+
+    /**
+     * The number of top contexts to retrieve.
+     */
+    topK?: number;
+  }
+>({
+  id: 'google.vertex_rag_store',
+  inputSchema: z.object({
+    ragCorpus: z.string(),
+    topK: z.number().optional(),
+  }),
+});

package/src/version.ts

@@ -0,0 +1,6 @@
+// Version string of this package injected at build time.
+declare const __PACKAGE_VERSION__: string | undefined;
+export const VERSION: string =
+  typeof __PACKAGE_VERSION__ !== 'undefined'
+    ? __PACKAGE_VERSION__
+    : '0.0.0-test';