@elizaos/plugin-google-genai 2.0.0-alpha.8 → 2.0.0-beta.1

package/dist/init.d.ts

@@ -0,0 +1,16 @@
+import { type IAgentRuntime } from "@elizaos/core";
+export interface PluginConfig {
+	readonly GOOGLE_GENERATIVE_AI_API_KEY?: string;
+	readonly GOOGLE_SMALL_MODEL?: string;
+	readonly GOOGLE_LARGE_MODEL?: string;
+	readonly GOOGLE_IMAGE_MODEL?: string;
+	readonly GOOGLE_EMBEDDING_MODEL?: string;
+	readonly SMALL_MODEL?: string;
+	readonly LARGE_MODEL?: string;
+	readonly IMAGE_MODEL?: string;
+}
+export declare function initializeGoogleGenAI(
+	_config: PluginConfig,
+	runtime: IAgentRuntime,
+): void;
+//# sourceMappingURL=init.d.ts.map

package/dist/init.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"init.d.ts","sourceRoot":"","sources":["../typescript/init.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,aAAa,EAAU,MAAM,eAAe,CAAC;AAI3D,MAAM,WAAW,YAAY;IAC3B,QAAQ,CAAC,4BAA4B,CAAC,EAAE,MAAM,CAAC;IAC/C,QAAQ,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IACrC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IACrC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,MAAM,CAAC;IACrC,QAAQ,CAAC,sBAAsB,CAAC,EAAE,MAAM,CAAC;IACzC,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,wBAAgB,qBAAqB,CAAC,OAAO,EAAE,YAAY,EAAE,OAAO,EAAE,aAAa,GAAG,IAAI,CAsBzF"}

package/dist/init.js

@@ -0,0 +1,27 @@
+import { logger } from "@elizaos/core";
+import { GoogleGenAI } from "@google/genai";
+import { getApiKey } from "./utils/config";
+export function initializeGoogleGenAI(_config, runtime) {
+	(async () => {
+		try {
+			const apiKey = getApiKey(runtime);
+			if (!apiKey) {
+				logger.warn("GOOGLE_GENERATIVE_AI_API_KEY is not set");
+				return;
+			}
+			const genAI = new GoogleGenAI({ apiKey });
+			const modelList = await genAI.models.list();
+			const models = [];
+			for await (const model of modelList) {
+				models.push(model);
+			}
+			logger.log(
+				`Google AI API key validated. Available models: ${models.length}`,
+			);
+		} catch (error) {
+			logger.warn(
+				`Google AI configuration error: ${error instanceof Error ? error.message : String(error)}`,
+			);
+		}
+	})();
+}

package/dist/models/embedding.d.ts

@@ -0,0 +1,6 @@
+import type { IAgentRuntime, TextEmbeddingParams } from "@elizaos/core";
+export declare function handleTextEmbedding(
+	runtime: IAgentRuntime,
+	params: TextEmbeddingParams | string | null,
+): Promise<number[]>;
+//# sourceMappingURL=embedding.d.ts.map

package/dist/models/embedding.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.d.ts","sourceRoot":"","sources":["../../typescript/models/embedding.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAUxE,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,mBAAmB,GAAG,MAAM,GAAG,IAAI,GAC1C,OAAO,CAAC,MAAM,EAAE,CAAC,CA0DnB"}

package/dist/models/embedding.js

@@ -0,0 +1,57 @@
+import * as ElizaCore from "@elizaos/core";
+import { logger } from "@elizaos/core";
+import { createGoogleGenAI, getEmbeddingModel } from "../utils/config";
+import { emitModelUsageEvent } from "../utils/events";
+import { countTokens } from "../utils/tokenization";
+
+const TEXT_EMBEDDING_MODEL_TYPE =
+	ElizaCore.ModelType?.TEXT_EMBEDDING ?? "TEXT_EMBEDDING";
+export async function handleTextEmbedding(runtime, params) {
+	const genAI = createGoogleGenAI(runtime);
+	if (!genAI) {
+		throw new Error("Google Generative AI client not initialized");
+	}
+	const embeddingModelName = getEmbeddingModel(runtime);
+	logger.debug(`[TEXT_EMBEDDING] Using model: ${embeddingModelName}`);
+	if (params === null) {
+		return Array(768).fill(0);
+	}
+	let text =
+		typeof params === "string"
+			? params
+			: typeof params === "object" && params.text
+				? params.text
+				: "";
+	if (!text.trim()) {
+		logger.warn("Empty text for embedding");
+		return Array(768).fill(0);
+	}
+	// Truncate to stay within embedding model token limits (~4 chars per token)
+	const maxChars = 8_192 * 4;
+	if (text.length > maxChars) {
+		logger.warn(
+			`[Google GenAI] Embedding input too long (~${Math.ceil(text.length / 4)} tokens), truncating to ~8192 tokens`,
+		);
+		text = text.slice(0, maxChars);
+	}
+	try {
+		const response = await genAI.models.embedContent({
+			model: embeddingModelName,
+			contents: text,
+		});
+		const embedding = response.embeddings?.[0]?.values || [];
+		const promptTokens = await countTokens(text);
+		emitModelUsageEvent(runtime, TEXT_EMBEDDING_MODEL_TYPE, text, {
+			promptTokens,
+			completionTokens: 0,
+			totalTokens: promptTokens,
+		});
+		logger.log(`Got embedding with length ${embedding.length}`);
+		return embedding;
+	} catch (error) {
+		logger.error(
+			`Error generating embedding: ${error instanceof Error ? error.message : String(error)}`,
+		);
+		return Array(768).fill(0);
+	}
+}

package/dist/models/image.d.ts

@@ -0,0 +1,7 @@
+import type { IAgentRuntime, ImageDescriptionParams } from "@elizaos/core";
+import type { ImageDescriptionResponse } from "../types";
+export declare function handleImageDescription(
+	runtime: IAgentRuntime,
+	params: ImageDescriptionParams | string,
+): Promise<ImageDescriptionResponse>;
+//# sourceMappingURL=image.d.ts.map

package/dist/models/image.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"image.d.ts","sourceRoot":"","sources":["../../typescript/models/image.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AAE3E,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,UAAU,CAAC;AAKzD,wBAAsB,sBAAsB,CAC1C,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,sBAAsB,GAAG,MAAM,GACtC,OAAO,CAAC,wBAAwB,CAAC,CAoFnC"}

package/dist/models/image.js

@@ -0,0 +1,91 @@
+import { logger } from "@elizaos/core";
+import {
+	createGoogleGenAI,
+	getImageModel,
+	getSafetySettings,
+} from "../utils/config";
+
+const crossFetch =
+	typeof globalThis.fetch === "function" ? globalThis.fetch : fetch;
+export async function handleImageDescription(runtime, params) {
+	const genAI = createGoogleGenAI(runtime);
+	if (!genAI) {
+		throw new Error("Google Generative AI client not initialized");
+	}
+	let imageUrl;
+	let promptText;
+	const modelName = getImageModel(runtime);
+	logger.log(`[IMAGE_DESCRIPTION] Using model: ${modelName}`);
+	if (typeof params === "string") {
+		imageUrl = params;
+		promptText =
+			"Please analyze this image and provide a title and detailed description.";
+	} else {
+		imageUrl = params.imageUrl;
+		promptText =
+			params.prompt ||
+			"Please analyze this image and provide a title and detailed description.";
+	}
+	try {
+		const imageResponse = await crossFetch(imageUrl);
+		if (!imageResponse.ok) {
+			throw new Error(`Failed to fetch image: ${imageResponse.statusText}`);
+		}
+		const imageData = await imageResponse.arrayBuffer();
+		const base64Image = Buffer.from(imageData).toString("base64");
+		const contentType =
+			imageResponse.headers.get("content-type") || "image/jpeg";
+		const response = await genAI.models.generateContent({
+			model: modelName,
+			contents: [
+				{
+					role: "user",
+					parts: [
+						{ text: promptText },
+						{
+							inlineData: {
+								mimeType: contentType,
+								data: base64Image,
+							},
+						},
+					],
+				},
+			],
+			config: {
+				temperature: 0.7,
+				topK: 40,
+				topP: 0.95,
+				maxOutputTokens: 8192,
+				safetySettings: getSafetySettings(),
+			},
+		});
+		const responseText = response.text || "";
+		try {
+			const jsonResponse = JSON.parse(responseText);
+			if (
+				typeof jsonResponse.title === "string" &&
+				typeof jsonResponse.description === "string"
+			) {
+				return {
+					title: jsonResponse.title,
+					description: jsonResponse.description,
+				};
+			}
+		} catch {
+			// Fall through to text parsing
+		}
+		const titleMatch = responseText.match(/title[:\s]+(.+?)(?:\n|$)/i);
+		const title = titleMatch?.[1]?.trim() || "Image Analysis";
+		const description = titleMatch
+			? responseText.replace(/title[:\s]+(.+?)(?:\n|$)/i, "").trim()
+			: responseText.trim();
+		return { title, description };
+	} catch (error) {
+		const message = error instanceof Error ? error.message : String(error);
+		logger.error(`Error analyzing image: ${message}`);
+		return {
+			title: "Failed to analyze image",
+			description: `Error: ${message}`,
+		};
+	}
+}

package/dist/models/index.d.ts

@@ -0,0 +1,13 @@
+export { handleTextEmbedding } from "./embedding";
+export { handleImageDescription } from "./image";
+export { handleObjectLarge, handleObjectSmall } from "./object";
+export {
+	handleActionPlanner,
+	handleResponseHandler,
+	handleTextLarge,
+	handleTextMedium,
+	handleTextMega,
+	handleTextNano,
+	handleTextSmall,
+} from "./text";
+//# sourceMappingURL=index.d.ts.map

package/dist/models/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../typescript/models/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AACjD,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,UAAU,CAAC;AAChE,OAAO,EACL,mBAAmB,EACnB,qBAAqB,EACrB,eAAe,EACf,gBAAgB,EAChB,cAAc,EACd,cAAc,EACd,eAAe,GAChB,MAAM,QAAQ,CAAC"}

package/dist/models/index.js

@@ -0,0 +1,12 @@
+export { handleTextEmbedding } from "./embedding";
+export { handleImageDescription } from "./image";
+export { handleObjectLarge, handleObjectSmall } from "./object";
+export {
+	handleActionPlanner,
+	handleResponseHandler,
+	handleTextLarge,
+	handleTextMedium,
+	handleTextMega,
+	handleTextNano,
+	handleTextSmall,
+} from "./text";

package/dist/models/object.d.ts

@@ -0,0 +1,10 @@
+import type { IAgentRuntime, ObjectGenerationParams } from "@elizaos/core";
+export declare function handleObjectSmall(
+	runtime: IAgentRuntime,
+	params: ObjectGenerationParams,
+): Promise<Record<string, string | number | boolean | null>>;
+export declare function handleObjectLarge(
+	runtime: IAgentRuntime,
+	params: ObjectGenerationParams,
+): Promise<Record<string, string | number | boolean | null>>;
+//# sourceMappingURL=object.d.ts.map

package/dist/models/object.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"object.d.ts","sourceRoot":"","sources":["../../typescript/models/object.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAiB,sBAAsB,EAAE,MAAM,eAAe,CAAC;AA8E1F,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,sBAAsB,GAC7B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC,CAE3D;AAED,wBAAsB,iBAAiB,CACrC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,sBAAsB,GAC7B,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,GAAG,IAAI,CAAC,CAAC,CAE3D"}

package/dist/models/object.js

@@ -0,0 +1,84 @@
+import { logger } from "@elizaos/core";
+import {
+	createGoogleGenAI,
+	getLargeModel,
+	getSafetySettings,
+	getSmallModel,
+} from "../utils/config";
+import { emitModelUsageEvent } from "../utils/events";
+import { countTokens } from "../utils/tokenization";
+
+async function generateObjectByModelType(
+	runtime,
+	params,
+	modelType,
+	getModelFn,
+) {
+	const genAI = createGoogleGenAI(runtime);
+	if (!genAI) {
+		throw new Error("Google Generative AI client not initialized");
+	}
+	const modelName = getModelFn(runtime);
+	const temperature = params.temperature ?? 0.1;
+	logger.info(`Using ${modelType} model: ${modelName}`);
+	try {
+		let enhancedPrompt = params.prompt;
+		if (params.schema) {
+			enhancedPrompt += `\n\nPlease respond with a JSON object that follows this schema:\n${JSON.stringify(params.schema, null, 2)}`;
+		}
+		const response = await genAI.models.generateContent({
+			model: modelName,
+			contents: enhancedPrompt,
+			config: {
+				temperature,
+				topK: 40,
+				topP: 0.95,
+				maxOutputTokens: 8192,
+				responseMimeType: "application/json",
+				safetySettings: getSafetySettings(),
+			},
+		});
+		const text = response.text || "";
+		const promptTokens = await countTokens(enhancedPrompt);
+		const completionTokens = await countTokens(text);
+		emitModelUsageEvent(runtime, modelType, params.prompt, {
+			promptTokens,
+			completionTokens,
+			totalTokens: promptTokens + completionTokens,
+		});
+		try {
+			return JSON.parse(text);
+		} catch {
+			const jsonMatch = text.match(/\{[\s\S]*\}/);
+			if (jsonMatch) {
+				try {
+					return JSON.parse(jsonMatch[0]);
+				} catch {
+					throw new Error("Failed to parse JSON from response");
+				}
+			}
+			throw new Error("Failed to parse JSON from response");
+		}
+	} catch (error) {
+		logger.error(
+			`[generateObject] Error: ${error instanceof Error ? error.message : String(error)}`,
+		);
+		throw error;
+	}
+}
+export async function handleObjectSmall(runtime, params) {
+	return generateObjectByModelType(
+		runtime,
+		params,
+		"OBJECT_SMALL",
+		getSmallModel,
+	);
+}
+export async function handleObjectLarge(runtime, params) {
+	return generateObjectByModelType(
+		runtime,
+		params,
+		"OBJECT_LARGE",
+		getLargeModel,
+	);
+}

package/dist/models/text.d.ts

@@ -0,0 +1,51 @@
+import type { GenerateTextParams, IAgentRuntime } from "@elizaos/core";
+
+type ChatAttachment = {
+	data: string | Uint8Array | URL;
+	mediaType: string;
+	filename?: string;
+};
+type GenerateTextParamsWithAttachments = GenerateTextParams & {
+	attachments?: ChatAttachment[];
+};
+export declare function handleTextSmall(
+	runtime: IAgentRuntime,
+	{
+		prompt,
+		stopSequences,
+		maxTokens,
+		temperature,
+		attachments,
+	}: GenerateTextParamsWithAttachments,
+): Promise<string>;
+export declare function handleTextLarge(
+	runtime: IAgentRuntime,
+	{
+		prompt,
+		stopSequences,
+		maxTokens,
+		temperature,
+		attachments,
+	}: GenerateTextParamsWithAttachments,
+): Promise<string>;
+export declare function handleTextNano(
+	runtime: IAgentRuntime,
+	params: GenerateTextParamsWithAttachments,
+): Promise<string>;
+export declare function handleTextMedium(
+	runtime: IAgentRuntime,
+	params: GenerateTextParamsWithAttachments,
+): Promise<string>;
+export declare function handleTextMega(
+	runtime: IAgentRuntime,
+	params: GenerateTextParamsWithAttachments,
+): Promise<string>;
+export declare function handleResponseHandler(
+	runtime: IAgentRuntime,
+	params: GenerateTextParamsWithAttachments,
+): Promise<string>;
+export declare function handleActionPlanner(
+	runtime: IAgentRuntime,
+	params: GenerateTextParamsWithAttachments,
+): Promise<string>;
+//# sourceMappingURL=text.d.ts.map

package/dist/models/text.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"text.d.ts","sourceRoot":"","sources":["../../typescript/models/text.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,kBAAkB,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AA2BvE,KAAK,cAAc,GAAG;IACpB,IAAI,EAAE,MAAM,GAAG,UAAU,GAAG,GAAG,CAAC;IAChC,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB,CAAC;AAEF,KAAK,iCAAiC,GAAG,kBAAkB,GAAG;IAC5D,WAAW,CAAC,EAAE,cAAc,EAAE,CAAC;CAChC,CAAC;AAyEF,wBAAsB,eAAe,CACnC,OAAO,EAAE,aAAa,EACtB,EACE,MAAM,EACN,aAAkB,EAClB,SAAgB,EAChB,WAAiB,EACjB,WAAW,GACZ,EAAE,iCAAiC,GACnC,OAAO,CAAC,MAAM,CAAC,CA6CjB;AAED,wBAAsB,eAAe,CACnC,OAAO,EAAE,aAAa,EACtB,EACE,MAAM,EACN,aAAkB,EAClB,SAAgB,EAChB,WAAiB,EACjB,WAAW,GACZ,EAAE,iCAAiC,GACnC,OAAO,CAAC,MAAM,CAAC,CA6CjB;AAED,wBAAsB,cAAc,CAClC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAEjB;AAED,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAEjB;AAED,wBAAsB,cAAc,CAClC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAEjB;AAED,wBAAsB,qBAAqB,CACzC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAEjB;AAED,wBAAsB,mBAAmB,CACvC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,iCAAiC,GACxC,OAAO,CAAC,MAAM,CAAC,CAEjB"}

package/dist/models/text.js

@@ -0,0 +1,257 @@
+import * as ElizaCore from "@elizaos/core";
+import { logger } from "@elizaos/core";
+import {
+	createGoogleGenAI,
+	getActionPlannerModel,
+	getLargeModel,
+	getMediumModel,
+	getMegaModel,
+	getNanoModel,
+	getResponseHandlerModel,
+	getSafetySettings,
+	getSmallModel,
+} from "../utils/config";
+import { emitModelUsageEvent } from "../utils/events";
+import { countTokens } from "../utils/tokenization";
+
+const CORE_MODEL_TYPES = ElizaCore.ModelType ?? {};
+const TEXT_NANO_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_NANO ?? "TEXT_NANO";
+const TEXT_MEDIUM_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_MEDIUM ?? "TEXT_MEDIUM";
+const TEXT_SMALL_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_SMALL ?? "TEXT_SMALL";
+const TEXT_LARGE_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_LARGE ?? "TEXT_LARGE";
+const TEXT_MEGA_MODEL_TYPE = CORE_MODEL_TYPES.TEXT_MEGA ?? "TEXT_MEGA";
+const RESPONSE_HANDLER_MODEL_TYPE =
+	CORE_MODEL_TYPES.RESPONSE_HANDLER ?? "RESPONSE_HANDLER";
+const 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);
+	}
+}
+export 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 = getModelNameForType(runtime, TEXT_SMALL_MODEL_TYPE);
+	logger.log(`[TEXT_SMALL] 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, TEXT_SMALL_MODEL_TYPE, prompt, {
+			promptTokens,
+			completionTokens,
+			totalTokens: promptTokens + completionTokens,
+		});
+		return text;
+	} catch (error) {
+		logger.error(
+			`[TEXT_SMALL] Error: ${error instanceof Error ? error.message : String(error)}`,
+		);
+		throw error;
+	}
+}
+export 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 = getModelNameForType(runtime, TEXT_LARGE_MODEL_TYPE);
+	logger.log(`[TEXT_LARGE] 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, TEXT_LARGE_MODEL_TYPE, prompt, {
+			promptTokens,
+			completionTokens,
+			totalTokens: promptTokens + completionTokens,
+		});
+		return text;
+	} catch (error) {
+		logger.error(
+			`[TEXT_LARGE] Error: ${error instanceof Error ? error.message : String(error)}`,
+		);
+		throw error;
+	}
+}
+export async function handleTextNano(runtime, params) {
+	return handleTextWithType(runtime, TEXT_NANO_MODEL_TYPE, params);
+}
+export async function handleTextMedium(runtime, params) {
+	return handleTextWithType(runtime, TEXT_MEDIUM_MODEL_TYPE, params);
+}
+export async function handleTextMega(runtime, params) {
+	return handleTextWithType(runtime, TEXT_MEGA_MODEL_TYPE, params);
+}
+export async function handleResponseHandler(runtime, params) {
+	return handleTextWithType(runtime, RESPONSE_HANDLER_MODEL_TYPE, params);
+}
+export 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);
+	logger.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) {
+		logger.error(
+			`[${modelType}] Error: ${error instanceof Error ? error.message : String(error)}`,
+		);
+		throw error;
+	}
+}

package/dist/node/index.node.d.ts

@@ -0,0 +1,2 @@
+export * from "../index.node";
+export { default } from "../index.node";