@elizaos/plugin-google-genai 2.0.0-alpha.4 → 2.0.0-alpha.537

package/README.md

@@ -0,0 +1,122 @@
+# @elizaos/plugin-google-genai
+
+Google Generative AI (Gemini) plugin for elizaOS with native support for TypeScript, Python, and Rust.
+
+## Features
+
+- **Text Generation**: Generate text using Gemini 2.0 Flash and 2.5 Pro models
+- **Embeddings**: Generate text embeddings with text-embedding-004
+- **Image Analysis**: Analyze and describe images with multimodal capabilities
+- **JSON Object Generation**: Generate structured JSON with schema validation
+
+## Available Models
+
+| Model Type        | Default Model                | Description                        |
+| ----------------- | ---------------------------- | ---------------------------------- |
+| TEXT_SMALL        | gemini-2.0-flash-001         | Fast, efficient for everyday tasks |
+| TEXT_LARGE        | gemini-2.5-pro-preview-03-25 | Most capable for complex tasks     |
+| TEXT_EMBEDDING    | text-embedding-004           | Text embeddings (768 dimensions)   |
+| IMAGE_DESCRIPTION | gemini-2.5-pro-preview-03-25 | Multimodal image analysis          |
+| OBJECT_SMALL      | gemini-2.0-flash-001         | Fast JSON generation               |
+| OBJECT_LARGE      | gemini-2.5-pro-preview-03-25 | Complex JSON generation            |
+
+## Installation
+
+### TypeScript/JavaScript (npm)
+
+```bash
+npm install @elizaos/plugin-google-genai
+# or
+bun add @elizaos/plugin-google-genai
+```
+## Configuration
+
+Set the following environment variables:
+
+| Variable                       | Required | Description                                                                  |
+| ------------------------------ | -------- | ---------------------------------------------------------------------------- |
+| `GOOGLE_GENERATIVE_AI_API_KEY` | Yes      | Your Google AI API key from [Google AI Studio](https://aistudio.google.com/) |
+| `GOOGLE_SMALL_MODEL`           | No       | Override small model (default: gemini-2.0-flash-001)                         |
+| `GOOGLE_LARGE_MODEL`           | No       | Override large model (default: gemini-2.5-pro-preview-03-25)                 |
+| `GOOGLE_EMBEDDING_MODEL`       | No       | Override embedding model (default: text-embedding-004)                       |
+| `GOOGLE_IMAGE_MODEL`           | No       | Override image analysis model                                                |
+| `GOOGLE_TIMEOUT_SECONDS`       | No       | Request timeout (default: 60)                                                |
+
+## Usage
+
+### TypeScript (elizaOS Plugin)
+
+```typescript
+import { googleGenAIPlugin } from "@elizaos/plugin-google-genai";
+
+// Register the plugin with your elizaOS agent
+const agent = new Agent({
+  plugins: [googleGenAIPlugin],
+});
+
+// Use via runtime
+const text = await runtime.useModel(ModelType.TEXT_LARGE, {
+  prompt: "Explain quantum mechanics in simple terms.",
+});
+
+const embedding = await runtime.useModel(ModelType.TEXT_EMBEDDING, {
+  text: "Hello, world!",
+});
+
+const object = await runtime.useModel(ModelType.OBJECT_SMALL, {
+  prompt: "Generate a person profile with name and age.",
+  schema: {
+    type: "object",
+    properties: {
+      name: { type: "string" },
+      age: { type: "number" },
+    },
+  },
+});
+```
+## Project Structure
+
+```
+plugin-google-genai/
+├── typescript/          # TypeScript implementation
+│   ├── index.ts         # Main plugin entry
+│   ├── models/          # Model handlers
+│   ├── utils/           # Utility functions
+│   └── __tests__/       # Unit and integration tests
+├── package.json         # npm publishing config
+└── README.md            # This file
+```
+
+## Development
+
+### TypeScript
+
+```bash
+# Install dependencies
+bun install
+
+# Build
+bun run build
+
+# Run tests
+bun run test
+
+# Type checking
+bun run typecheck
+```
+
+## Publishing
+
+### npm (TypeScript)
+
+```bash
+npm publish --access public
+```
+
+## License
+
+MIT
+
+## Contributing
+
+See the main [elizaOS contribution guidelines](https://github.com/elizaos/eliza/blob/main/CONTRIBUTING.md).

package/dist/browser/index.browser.js

@@ -1,5 +1,6 @@
 // index.ts
-import { logger as logger7, ModelType as ModelType3 } from "@elizaos/core";
+import * as ElizaCore3 from "@elizaos/core";
+import { logger as logger7, ModelType as ModelType4 } from "@elizaos/core";
 import { GoogleGenAI as GoogleGenAI3 } from "@google/genai";
 
 // init.ts
@@ -29,9 +30,24 @@ function getApiKey(runtime) {
 function getSmallModel(runtime) {
   return getSetting(runtime, "GOOGLE_SMALL_MODEL") ?? getSetting(runtime, "SMALL_MODEL", "gemini-2.0-flash-001") ?? "gemini-2.0-flash-001";
 }
+function getNanoModel(runtime) {
+  return getSetting(runtime, "GOOGLE_NANO_MODEL") ?? getSetting(runtime, "NANO_MODEL") ?? getSmallModel(runtime);
+}
+function getMediumModel(runtime) {
+  return getSetting(runtime, "GOOGLE_MEDIUM_MODEL") ?? getSetting(runtime, "MEDIUM_MODEL") ?? getSmallModel(runtime);
+}
 function getLargeModel(runtime) {
   return getSetting(runtime, "GOOGLE_LARGE_MODEL") ?? getSetting(runtime, "LARGE_MODEL", "gemini-2.5-pro-preview-03-25") ?? "gemini-2.5-pro-preview-03-25";
 }
+function getMegaModel(runtime) {
+  return getSetting(runtime, "GOOGLE_MEGA_MODEL") ?? getSetting(runtime, "MEGA_MODEL") ?? getLargeModel(runtime);
+}
+function getResponseHandlerModel(runtime) {
+  return getSetting(runtime, "GOOGLE_RESPONSE_HANDLER_MODEL") ?? getSetting(runtime, "GOOGLE_SHOULD_RESPOND_MODEL") ?? getSetting(runtime, "RESPONSE_HANDLER_MODEL") ?? getSetting(runtime, "SHOULD_RESPOND_MODEL") ?? getNanoModel(runtime);
+}
+function getActionPlannerModel(runtime) {
+  return getSetting(runtime, "GOOGLE_ACTION_PLANNER_MODEL") ?? getSetting(runtime, "GOOGLE_PLANNER_MODEL") ?? getSetting(runtime, "ACTION_PLANNER_MODEL") ?? getSetting(runtime, "PLANNER_MODEL") ?? getMediumModel(runtime);
+}
 function getImageModel(runtime) {
   return getSetting(runtime, "GOOGLE_IMAGE_MODEL") ?? getSetting(runtime, "IMAGE_MODEL", "gemini-2.5-pro-preview-03-25") ?? "gemini-2.5-pro-preview-03-25";
 }
@@ -90,12 +106,13 @@ function initializeGoogleGenAI(_config, runtime) {
 }
 
 // models/embedding.ts
-import { logger as logger3, MAX_EMBEDDING_TOKENS, ModelType } from "@elizaos/core";
+import * as ElizaCore from "@elizaos/core";
+import { logger as logger3 } from "@elizaos/core";
 
 // utils/events.ts
-import { EventType } from "@elizaos/core";
+var MODEL_USED_EVENT = "MODEL_USED";
 function emitModelUsageEvent(runtime, type, _prompt, usage) {
-  runtime.emitEvent(EventType.MODEL_USED, {
+  runtime.emitEvent(MODEL_USED_EVENT, {
     runtime,
     source: "plugin-google-genai",
     type,
@@ -113,6 +130,7 @@ async function countTokens(text) {
 }
 
 // models/embedding.ts
+var TEXT_EMBEDDING_MODEL_TYPE = ElizaCore.ModelType?.TEXT_EMBEDDING ?? "TEXT_EMBEDDING";
 async function handleTextEmbedding(runtime, params) {
   const genAI = createGoogleGenAI(runtime);
   if (!genAI) {
@@ -128,9 +146,9 @@ async function handleTextEmbedding(runtime, params) {
     logger3.warn("Empty text for embedding");
     return Array(768).fill(0);
   }
-  const maxChars = MAX_EMBEDDING_TOKENS * 4;
+  const maxChars = 8192 * 4;
   if (text.length > maxChars) {
-    logger3.warn(`[Google GenAI] Embedding input too long (~${Math.ceil(text.length / 4)} tokens), truncating to ~${MAX_EMBEDDING_TOKENS} tokens`);
+    logger3.warn(`[Google GenAI] Embedding input too long (~${Math.ceil(text.length / 4)} tokens), truncating to ~8192 tokens`);
     text = text.slice(0, maxChars);
   }
   try {
@@ -140,7 +158,7 @@ async function handleTextEmbedding(runtime, params) {
     });
     const embedding = response.embeddings?.[0]?.values || [];
     const promptTokens = await countTokens(text);
-    emitModelUsageEvent(runtime, ModelType.TEXT_EMBEDDING, text, {
+    emitModelUsageEvent(runtime, TEXT_EMBEDDING_MODEL_TYPE, text, {
       promptTokens,
       completionTokens: 0,
       totalTokens: promptTokens
@@ -289,19 +307,94 @@ async function handleObjectLarge(runtime, params) {
   return generateObjectByModelType(runtime, params, "OBJECT_LARGE", getLargeModel);
 }
 // models/text.ts
-import { logger as logger6, ModelType as ModelType2 } from "@elizaos/core";
-async function handleTextSmall(runtime, { prompt, stopSequences = [], maxTokens = 8192, temperature = 0.7 }) {
+import * as ElizaCore2 from "@elizaos/core";
+import { logger as logger6 } from "@elizaos/core";
+var CORE_MODEL_TYPES = ElizaCore2.ModelType ?? {};
+var TEXT_NANO_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_NANO ?? "TEXT_NANO";
+var TEXT_MEDIUM_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_MEDIUM ?? "TEXT_MEDIUM";
+var TEXT_SMALL_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_SMALL ?? "TEXT_SMALL";
+var TEXT_LARGE_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_LARGE ?? "TEXT_LARGE";
+var TEXT_MEGA_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_MEGA ?? "TEXT_MEGA";
+var RESPONSE_HANDLER_MODEL_TYPE = CORE_MODEL_TYPES.RESPONSE_HANDLER ?? "RESPONSE_HANDLER";
+var ACTION_PLANNER_MODEL_TYPE = CORE_MODEL_TYPES.ACTION_PLANNER ?? "ACTION_PLANNER";
+function buildPromptParts(prompt, attachments) {
+  const parts = [{ text: prompt }];
+  for (const attachment of attachments ?? []) {
+    if (attachment.data instanceof URL) {
+      parts.push({
+        fileData: {
+          mimeType: attachment.mediaType,
+          fileUri: attachment.data.toString()
+        }
+      });
+      continue;
+    }
+    if (typeof attachment.data === "string" && /^https?:\/\//i.test(attachment.data)) {
+      parts.push({
+        fileData: {
+          mimeType: attachment.mediaType,
+          fileUri: attachment.data
+        }
+      });
+      continue;
+    }
+    if (typeof attachment.data === "string") {
+      const dataUrlMatch = attachment.data.match(/^data:([^;,]+);base64,(.+)$/i);
+      parts.push({
+        inlineData: {
+          mimeType: dataUrlMatch?.[1] ?? attachment.mediaType,
+          data: dataUrlMatch?.[2] ?? attachment.data
+        }
+      });
+      continue;
+    }
+    parts.push({
+      inlineData: {
+        mimeType: attachment.mediaType,
+        data: Buffer.from(attachment.data).toString("base64")
+      }
+    });
+  }
+  return parts;
+}
+function getModelNameForType(runtime, modelType) {
+  switch (modelType) {
+    case TEXT_NANO_MODEL_TYPE:
+      return getNanoModel(runtime);
+    case TEXT_MEDIUM_MODEL_TYPE:
+      return getMediumModel(runtime);
+    case TEXT_SMALL_MODEL_TYPE:
+      return getSmallModel(runtime);
+    case TEXT_LARGE_MODEL_TYPE:
+      return getLargeModel(runtime);
+    case TEXT_MEGA_MODEL_TYPE:
+      return getMegaModel(runtime);
+    case RESPONSE_HANDLER_MODEL_TYPE:
+      return getResponseHandlerModel(runtime);
+    case ACTION_PLANNER_MODEL_TYPE:
+      return getActionPlannerModel(runtime);
+    default:
+      return getLargeModel(runtime);
+  }
+}
+async function handleTextSmall(runtime, {
+  prompt,
+  stopSequences = [],
+  maxTokens = 8192,
+  temperature = 0.7,
+  attachments
+}) {
   const genAI = createGoogleGenAI(runtime);
   if (!genAI) {
     throw new Error("Google Generative AI client not initialized");
   }
-  const modelName = getSmallModel(runtime);
+  const modelName = getModelNameForType(runtime, TEXT_SMALL_MODEL_TYPE);
   logger6.log(`[TEXT_SMALL] Using model: ${modelName}`);
   try {
     const systemInstruction = runtime.character.system || undefined;
     const response = await genAI.models.generateContent({
       model: modelName,
-      contents: prompt,
+      contents: (attachments?.length ?? 0) > 0 ? [{ role: "user", parts: buildPromptParts(prompt, attachments) }] : prompt,
       config: {
         temperature,
         topK: 40,
@@ -315,7 +408,7 @@ async function handleTextSmall(runtime, { prompt, stopSequences = [], maxTokens
     const text = response.text || "";
     const promptTokens = await countTokens(prompt);
     const completionTokens = await countTokens(text);
-    emitModelUsageEvent(runtime, ModelType2.TEXT_SMALL, prompt, {
+    emitModelUsageEvent(runtime, TEXT_SMALL_MODEL_TYPE, prompt, {
       promptTokens,
       completionTokens,
       totalTokens: promptTokens + completionTokens
@@ -326,18 +419,24 @@ async function handleTextSmall(runtime, { prompt, stopSequences = [], maxTokens
     throw error;
   }
 }
-async function handleTextLarge(runtime, { prompt, stopSequences = [], maxTokens = 8192, temperature = 0.7 }) {
+async function handleTextLarge(runtime, {
+  prompt,
+  stopSequences = [],
+  maxTokens = 8192,
+  temperature = 0.7,
+  attachments
+}) {
   const genAI = createGoogleGenAI(runtime);
   if (!genAI) {
     throw new Error("Google Generative AI client not initialized");
   }
-  const modelName = getLargeModel(runtime);
+  const modelName = getModelNameForType(runtime, TEXT_LARGE_MODEL_TYPE);
   logger6.log(`[TEXT_LARGE] Using model: ${modelName}`);
   try {
     const systemInstruction = runtime.character.system || undefined;
     const response = await genAI.models.generateContent({
       model: modelName,
-      contents: prompt,
+      contents: (attachments?.length ?? 0) > 0 ? [{ role: "user", parts: buildPromptParts(prompt, attachments) }] : prompt,
       config: {
         temperature,
         topK: 40,
@@ -351,7 +450,7 @@ async function handleTextLarge(runtime, { prompt, stopSequences = [], maxTokens
     const text = response.text || "";
     const promptTokens = await countTokens(prompt);
     const completionTokens = await countTokens(text);
-    emitModelUsageEvent(runtime, ModelType2.TEXT_LARGE, prompt, {
+    emitModelUsageEvent(runtime, TEXT_LARGE_MODEL_TYPE, prompt, {
       promptTokens,
       completionTokens,
       totalTokens: promptTokens + completionTokens
@@ -362,7 +461,76 @@ async function handleTextLarge(runtime, { prompt, stopSequences = [], maxTokens
     throw error;
   }
 }
+async function handleTextNano(runtime, params) {
+  return handleTextWithType(runtime, TEXT_NANO_MODEL_TYPE, params);
+}
+async function handleTextMedium(runtime, params) {
+  return handleTextWithType(runtime, TEXT_MEDIUM_MODEL_TYPE, params);
+}
+async function handleTextMega(runtime, params) {
+  return handleTextWithType(runtime, TEXT_MEGA_MODEL_TYPE, params);
+}
+async function handleResponseHandler(runtime, params) {
+  return handleTextWithType(runtime, RESPONSE_HANDLER_MODEL_TYPE, params);
+}
+async function handleActionPlanner(runtime, params) {
+  return handleTextWithType(runtime, ACTION_PLANNER_MODEL_TYPE, params);
+}
+async function handleTextWithType(runtime, modelType, {
+  prompt,
+  stopSequences = [],
+  maxTokens = 8192,
+  temperature = 0.7,
+  attachments
+}) {
+  const genAI = createGoogleGenAI(runtime);
+  if (!genAI) {
+    throw new Error("Google Generative AI client not initialized");
+  }
+  const modelName = getModelNameForType(runtime, modelType);
+  logger6.log(`[${modelType}] Using model: ${modelName}`);
+  try {
+    const systemInstruction = runtime.character.system || undefined;
+    const response = await genAI.models.generateContent({
+      model: modelName,
+      contents: (attachments?.length ?? 0) > 0 ? [{ role: "user", parts: buildPromptParts(prompt, attachments) }] : prompt,
+      config: {
+        temperature,
+        topK: 40,
+        topP: 0.95,
+        maxOutputTokens: maxTokens,
+        stopSequences,
+        safetySettings: getSafetySettings(),
+        ...systemInstruction && { systemInstruction }
+      }
+    });
+    const text = response.text || "";
+    const promptTokens = await countTokens(prompt);
+    const completionTokens = await countTokens(text);
+    emitModelUsageEvent(runtime, modelType, prompt, {
+      promptTokens,
+      completionTokens,
+      totalTokens: promptTokens + completionTokens
+    });
+    return text;
+  } catch (error) {
+    logger6.error(`[${modelType}] Error: ${error instanceof Error ? error.message : String(error)}`);
+    throw error;
+  }
+}
 // index.ts
+var CORE_MODEL_TYPES2 = ElizaCore3.ModelType ?? {};
+var TEXT_NANO_MODEL_TYPE2 = CORE_MODEL_TYPES2.TEXT_NANO ?? "TEXT_NANO";
+var TEXT_MEDIUM_MODEL_TYPE2 = CORE_MODEL_TYPES2.TEXT_MEDIUM ?? "TEXT_MEDIUM";
+var TEXT_SMALL_MODEL_TYPE2 = CORE_MODEL_TYPES2.TEXT_SMALL ?? "TEXT_SMALL";
+var TEXT_LARGE_MODEL_TYPE2 = CORE_MODEL_TYPES2.TEXT_LARGE ?? "TEXT_LARGE";
+var TEXT_EMBEDDING_MODEL_TYPE2 = CORE_MODEL_TYPES2.TEXT_EMBEDDING ?? "TEXT_EMBEDDING";
+var IMAGE_DESCRIPTION_MODEL_TYPE = CORE_MODEL_TYPES2.IMAGE_DESCRIPTION ?? "IMAGE_DESCRIPTION";
+var OBJECT_SMALL_MODEL_TYPE = CORE_MODEL_TYPES2.OBJECT_SMALL ?? "OBJECT_SMALL";
+var OBJECT_LARGE_MODEL_TYPE = CORE_MODEL_TYPES2.OBJECT_LARGE ?? "OBJECT_LARGE";
+var TEXT_MEGA_MODEL_TYPE2 = CORE_MODEL_TYPES2.TEXT_MEGA ?? "TEXT_MEGA";
+var RESPONSE_HANDLER_MODEL_TYPE2 = CORE_MODEL_TYPES2.RESPONSE_HANDLER ?? "RESPONSE_HANDLER";
+var ACTION_PLANNER_MODEL_TYPE2 = CORE_MODEL_TYPES2.ACTION_PLANNER ?? "ACTION_PLANNER";
 var pluginTests = [
   {
     name: "google_genai_plugin_tests",
@@ -387,7 +555,7 @@ var pluginTests = [
         name: "google_test_text_embedding",
         fn: async (runtime) => {
           try {
-            const embedding = await runtime.useModel(ModelType3.TEXT_EMBEDDING, {
+            const embedding = await runtime.useModel(ModelType4.TEXT_EMBEDDING, {
               text: "Hello, world!"
             });
             logger7.log(`Embedding dimension: ${embedding.length}`);
@@ -404,7 +572,7 @@ var pluginTests = [
         name: "google_test_text_small",
         fn: async (runtime) => {
           try {
-            const text = await runtime.useModel(ModelType3.TEXT_SMALL, {
+            const text = await runtime.useModel(ModelType4.TEXT_SMALL, {
               prompt: "What is the nature of reality in 10 words?"
             });
             if (text.length === 0) {
@@ -421,7 +589,7 @@ var pluginTests = [
         name: "google_test_text_large",
         fn: async (runtime) => {
           try {
-            const text = await runtime.useModel(ModelType3.TEXT_LARGE, {
+            const text = await runtime.useModel(ModelType4.TEXT_LARGE, {
               prompt: "Explain quantum mechanics in simple terms."
             });
             if (text.length === 0) {
@@ -438,8 +606,8 @@ var pluginTests = [
         name: "google_test_image_description",
         fn: async (runtime) => {
           try {
-            const result = await runtime.useModel(ModelType3.IMAGE_DESCRIPTION, "https://upload.wikimedia.org/wikipedia/commons/thumb/1/1c/Vitalik_Buterin_TechCrunch_London_2015_%28cropped%29.jpg/537px-Vitalik_Buterin_TechCrunch_London_2015_%28cropped%29.jpg");
-            if (result && typeof result === "object" && "title" in result && "description" in result) {
+            const result = await runtime.useModel(ModelType4.IMAGE_DESCRIPTION, "https://upload.wikimedia.org/wikipedia/commons/thumb/1/1c/Vitalik_Buterin_TechCrunch_London_2015_%28cropped%29.jpg/537px-Vitalik_Buterin_TechCrunch_London_2015_%28cropped%29.jpg");
+            if (result != null && typeof result === "object" && "title" in result && "description" in result) {
               logger7.log("Image description:", JSON.stringify(result));
             } else {
               logger7.error(`Invalid image description result format: ${JSON.stringify(result)}`);
@@ -463,12 +631,12 @@ var pluginTests = [
               },
               required: ["name", "age", "hobbies"]
             };
-            const result = await runtime.useModel(ModelType3.OBJECT_SMALL, {
+            const result = await runtime.useModel(ModelType4.OBJECT_SMALL, {
               prompt: "Generate a person profile with name, age, and hobbies.",
               schema
             });
             logger7.log("Generated object:", JSON.stringify(result));
-            if (!result.name || !result.age || !result.hobbies) {
+            if (!result?.name || !result?.age || !result?.hobbies) {
               throw new Error("Generated object missing required fields");
             }
           } catch (error) {
@@ -492,43 +660,75 @@ var googleGenAIPlugin = {
   description: "Google Generative AI plugin for Gemini models",
   config: {
     GOOGLE_GENERATIVE_AI_API_KEY: env.GOOGLE_GENERATIVE_AI_API_KEY ?? null,
+    GOOGLE_NANO_MODEL: env.GOOGLE_NANO_MODEL ?? null,
+    GOOGLE_MEDIUM_MODEL: env.GOOGLE_MEDIUM_MODEL ?? null,
     GOOGLE_SMALL_MODEL: env.GOOGLE_SMALL_MODEL ?? null,
     GOOGLE_LARGE_MODEL: env.GOOGLE_LARGE_MODEL ?? null,
+    GOOGLE_MEGA_MODEL: env.GOOGLE_MEGA_MODEL ?? null,
+    GOOGLE_RESPONSE_HANDLER_MODEL: env.GOOGLE_RESPONSE_HANDLER_MODEL ?? null,
+    GOOGLE_SHOULD_RESPOND_MODEL: env.GOOGLE_SHOULD_RESPOND_MODEL ?? null,
+    GOOGLE_ACTION_PLANNER_MODEL: env.GOOGLE_ACTION_PLANNER_MODEL ?? null,
+    GOOGLE_PLANNER_MODEL: env.GOOGLE_PLANNER_MODEL ?? null,
     GOOGLE_IMAGE_MODEL: env.GOOGLE_IMAGE_MODEL ?? null,
     GOOGLE_EMBEDDING_MODEL: env.GOOGLE_EMBEDDING_MODEL ?? null,
+    NANO_MODEL: env.NANO_MODEL ?? null,
+    MEDIUM_MODEL: env.MEDIUM_MODEL ?? null,
     SMALL_MODEL: env.SMALL_MODEL ?? null,
     LARGE_MODEL: env.LARGE_MODEL ?? null,
+    MEGA_MODEL: env.MEGA_MODEL ?? null,
+    RESPONSE_HANDLER_MODEL: env.RESPONSE_HANDLER_MODEL ?? null,
+    SHOULD_RESPOND_MODEL: env.SHOULD_RESPOND_MODEL ?? null,
+    ACTION_PLANNER_MODEL: env.ACTION_PLANNER_MODEL ?? null,
+    PLANNER_MODEL: env.PLANNER_MODEL ?? null,
     IMAGE_MODEL: env.IMAGE_MODEL ?? null
   },
   async init(config, runtime) {
     initializeGoogleGenAI(config, runtime);
   },
   models: {
-    [ModelType3.TEXT_SMALL]: async (runtime, params) => {
+    [TEXT_NANO_MODEL_TYPE2]: async (runtime, params) => {
+      return handleTextNano(runtime, params);
+    },
+    [TEXT_MEDIUM_MODEL_TYPE2]: async (runtime, params) => {
+      return handleTextMedium(runtime, params);
+    },
+    [TEXT_SMALL_MODEL_TYPE2]: async (runtime, params) => {
       return handleTextSmall(runtime, params);
     },
-    [ModelType3.TEXT_LARGE]: async (runtime, params) => {
+    [TEXT_LARGE_MODEL_TYPE2]: async (runtime, params) => {
       return handleTextLarge(runtime, params);
     },
-    [ModelType3.TEXT_EMBEDDING]: async (runtime, params) => {
+    [TEXT_MEGA_MODEL_TYPE2]: async (runtime, params) => {
+      return handleTextMega(runtime, params);
+    },
+    [RESPONSE_HANDLER_MODEL_TYPE2]: async (runtime, params) => {
+      return handleResponseHandler(runtime, params);
+    },
+    [ACTION_PLANNER_MODEL_TYPE2]: async (runtime, params) => {
+      return handleActionPlanner(runtime, params);
+    },
+    [TEXT_EMBEDDING_MODEL_TYPE2]: async (runtime, params) => {
       return handleTextEmbedding(runtime, params);
     },
-    [ModelType3.IMAGE_DESCRIPTION]: async (runtime, params) => {
+    [IMAGE_DESCRIPTION_MODEL_TYPE]: async (runtime, params) => {
       return handleImageDescription(runtime, params);
     },
-    [ModelType3.OBJECT_SMALL]: async (runtime, params) => {
+    [OBJECT_SMALL_MODEL_TYPE]: async (runtime, params) => {
       return handleObjectSmall(runtime, params);
     },
-    [ModelType3.OBJECT_LARGE]: async (runtime, params) => {
+    [OBJECT_LARGE_MODEL_TYPE]: async (runtime, params) => {
       return handleObjectLarge(runtime, params);
     }
   },
   tests: pluginTests
 };
-var typescript_default = googleGenAIPlugin;
+var plugin_google_genai_default = googleGenAIPlugin;
+
+// index.browser.ts
+var index_browser_default = plugin_google_genai_default;
 export {
   googleGenAIPlugin,
-  typescript_default as default
+  index_browser_default as default
 };
 
-//# debugId=B2AECF43CC82835464756E2164756E21
+//# debugId=DB1428E48B045C3E64756E2164756E21