@mast-ai/google-genai 0.3.0 → 0.5.0

package/dist/GoogleGenAIAdapter.d.ts

@@ -1,3 +1,4 @@
+import type { Tool } from '@google/genai';
 import type { LlmAdapter, AdapterRequest, AdapterResponse, AdapterStreamChunk } from '@mast-ai/core';
 /** Token-usage statistics reported by the Gemini API. */
 export interface UsageMetadata {
@@ -15,12 +16,17 @@ export declare class GoogleGenAIAdapter implements LlmAdapter {
     private client;
     private modelName;
     private onUsageUpdate?;
+    private nativeTools?;
     /**
      * @param apiKey - Google AI API key.
      * @param modelName - Gemini model identifier (defaults to `"gemini-3.1-flash-lite-preview"`).
      * @param onUsageUpdate - Optional callback invoked with token-usage data after each response.
+     * @param nativeTools - Optional Gemini built-in tools (e.g. `{ googleSearch: {} }`,
+     *   `{ codeExecution: {} }`, `{ urlContext: {} }`) that are appended to the tools array on
+     *   every request alongside any function declarations from the registry.
      */
-    constructor(apiKey: string, modelName?: string, onUsageUpdate?: (usage: UsageMetadata) => void);
+    constructor(apiKey: string, modelName?: string, onUsageUpdate?: (usage: UsageMetadata) => void, nativeTools?: Tool[]);
+    private buildTools;
     /** {@inheritDoc LlmAdapter.generate} */
     generate(request: AdapterRequest): Promise<AdapterResponse>;
     /** {@inheritDoc LlmAdapter.generateStream} */

package/dist/GoogleGenAIAdapter.js

@@ -11,23 +11,34 @@ export class GoogleGenAIAdapter {
     client;
     modelName;
     onUsageUpdate;
+    nativeTools;
     /**
      * @param apiKey - Google AI API key.
      * @param modelName - Gemini model identifier (defaults to `"gemini-3.1-flash-lite-preview"`).
      * @param onUsageUpdate - Optional callback invoked with token-usage data after each response.
+     * @param nativeTools - Optional Gemini built-in tools (e.g. `{ googleSearch: {} }`,
+     *   `{ codeExecution: {} }`, `{ urlContext: {} }`) that are appended to the tools array on
+     *   every request alongside any function declarations from the registry.
      */
-    constructor(apiKey, modelName = 'gemini-3.1-flash-lite-preview', onUsageUpdate) {
+    constructor(apiKey, modelName = 'gemini-3.1-flash-lite-preview', onUsageUpdate, nativeTools) {
         this.client = new GoogleGenAI({ apiKey });
         this.modelName = modelName;
         this.onUsageUpdate = onUsageUpdate;
+        this.nativeTools = nativeTools;
+    }
+    buildTools(request) {
+        const functionDeclarations = request.tools.map((t) => this.mapTool(t));
+        const allTools = [
+            ...(functionDeclarations.length > 0 ? [{ functionDeclarations }] : []),
+            ...(this.nativeTools ?? []),
+        ];
+        return allTools.length > 0 ? allTools : undefined;
     }
     /** {@inheritDoc LlmAdapter.generate} */
     async generate(request) {
         const contents = this.mapMessages(request.messages);
         const systemInstruction = this.mapSystemInstruction(request.system);
-        const tools = request.tools.length > 0
-            ? [{ functionDeclarations: request.tools.map((t) => this.mapTool(t)) }]
-            : undefined;
+        const tools = this.buildTools(request);
         const outputSchema = request.outputSchema;
         const response = await this.client.models.generateContent({
             model: this.modelName,
@@ -84,9 +95,7 @@ export class GoogleGenAIAdapter {
     async *generateStream(request) {
         const contents = this.mapMessages(request.messages);
         const systemInstruction = this.mapSystemInstruction(request.system);
-        const tools = request.tools.length > 0
-            ? [{ functionDeclarations: request.tools.map((t) => this.mapTool(t)) }]
-            : undefined;
+        const tools = this.buildTools(request);
         const responseStream = await this.client.models.generateContentStream({
             model: this.modelName,
             contents,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mast-ai/google-genai",
-  "version": "0.3.0",
+  "version": "0.5.0",
   "license": "Apache-2.0",
   "type": "module",
   "main": "./dist/index.js",
@@ -28,7 +28,7 @@
   },
   "dependencies": {
     "@google/genai": "^1.50.1",
-    "@mast-ai/core": "^0.3.0"
+    "@mast-ai/core": "^0.5.0"
   },
   "devDependencies": {
     "typescript": "~6.0.2",