@core-ai/google-genai 0.5.1 → 0.6.1

package/dist/index.d.ts

@@ -1,5 +1,6 @@
 import { GoogleGenAI } from '@google/genai';
 import { ChatModel, EmbeddingModel, ImageModel } from '@core-ai/core-ai';
+import { z } from 'zod';
 
 type GoogleGenAIProviderOptions = {
     apiKey?: string;
@@ -14,4 +15,130 @@ type GoogleGenAIProvider = {
 };
 declare function createGoogleGenAI(options?: GoogleGenAIProviderOptions): GoogleGenAIProvider;
 
-export { type GoogleGenAIProvider, type GoogleGenAIProviderOptions, createGoogleGenAI };
+type GoogleReasoningMetadata = {
+    thoughtSignature?: string;
+};
+
+declare const googleGenerateProviderOptionsSchema: z.ZodObject<{
+    stopSequences: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    frequencyPenalty: z.ZodOptional<z.ZodNumber>;
+    presencePenalty: z.ZodOptional<z.ZodNumber>;
+    seed: z.ZodOptional<z.ZodNumber>;
+    topK: z.ZodOptional<z.ZodNumber>;
+}, "strict", z.ZodTypeAny, {
+    stopSequences?: string[] | undefined;
+    frequencyPenalty?: number | undefined;
+    presencePenalty?: number | undefined;
+    seed?: number | undefined;
+    topK?: number | undefined;
+}, {
+    stopSequences?: string[] | undefined;
+    frequencyPenalty?: number | undefined;
+    presencePenalty?: number | undefined;
+    seed?: number | undefined;
+    topK?: number | undefined;
+}>;
+type GoogleGenerateProviderOptions = z.infer<typeof googleGenerateProviderOptionsSchema>;
+declare const googleEmbedProviderOptionsSchema: z.ZodObject<{
+    taskType: z.ZodOptional<z.ZodString>;
+    title: z.ZodOptional<z.ZodString>;
+    mimeType: z.ZodOptional<z.ZodString>;
+    autoTruncate: z.ZodOptional<z.ZodBoolean>;
+}, "strict", z.ZodTypeAny, {
+    taskType?: string | undefined;
+    title?: string | undefined;
+    mimeType?: string | undefined;
+    autoTruncate?: boolean | undefined;
+}, {
+    taskType?: string | undefined;
+    title?: string | undefined;
+    mimeType?: string | undefined;
+    autoTruncate?: boolean | undefined;
+}>;
+type GoogleEmbedProviderOptions = z.infer<typeof googleEmbedProviderOptionsSchema>;
+declare const googleImageProviderOptionsSchema: z.ZodObject<{
+    outputGcsUri: z.ZodOptional<z.ZodString>;
+    negativePrompt: z.ZodOptional<z.ZodString>;
+    aspectRatio: z.ZodOptional<z.ZodString>;
+    guidanceScale: z.ZodOptional<z.ZodNumber>;
+    seed: z.ZodOptional<z.ZodNumber>;
+    safetyFilterLevel: z.ZodOptional<z.ZodEnum<["BLOCK_LOW_AND_ABOVE", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_ONLY_HIGH", "BLOCK_NONE"]>>;
+    personGeneration: z.ZodOptional<z.ZodEnum<["DONT_ALLOW", "ALLOW_ADULT", "ALLOW_ALL"]>>;
+    includeSafetyAttributes: z.ZodOptional<z.ZodBoolean>;
+    includeRaiReason: z.ZodOptional<z.ZodBoolean>;
+    language: z.ZodOptional<z.ZodString>;
+    outputMimeType: z.ZodOptional<z.ZodString>;
+    outputCompressionQuality: z.ZodOptional<z.ZodNumber>;
+    addWatermark: z.ZodOptional<z.ZodBoolean>;
+    labels: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodString>>;
+    imageSize: z.ZodOptional<z.ZodString>;
+    enhancePrompt: z.ZodOptional<z.ZodBoolean>;
+}, "strict", z.ZodTypeAny, {
+    seed?: number | undefined;
+    outputGcsUri?: string | undefined;
+    negativePrompt?: string | undefined;
+    aspectRatio?: string | undefined;
+    guidanceScale?: number | undefined;
+    safetyFilterLevel?: "BLOCK_LOW_AND_ABOVE" | "BLOCK_MEDIUM_AND_ABOVE" | "BLOCK_ONLY_HIGH" | "BLOCK_NONE" | undefined;
+    personGeneration?: "DONT_ALLOW" | "ALLOW_ADULT" | "ALLOW_ALL" | undefined;
+    includeSafetyAttributes?: boolean | undefined;
+    includeRaiReason?: boolean | undefined;
+    language?: string | undefined;
+    outputMimeType?: string | undefined;
+    outputCompressionQuality?: number | undefined;
+    addWatermark?: boolean | undefined;
+    labels?: Record<string, string> | undefined;
+    imageSize?: string | undefined;
+    enhancePrompt?: boolean | undefined;
+}, {
+    seed?: number | undefined;
+    outputGcsUri?: string | undefined;
+    negativePrompt?: string | undefined;
+    aspectRatio?: string | undefined;
+    guidanceScale?: number | undefined;
+    safetyFilterLevel?: "BLOCK_LOW_AND_ABOVE" | "BLOCK_MEDIUM_AND_ABOVE" | "BLOCK_ONLY_HIGH" | "BLOCK_NONE" | undefined;
+    personGeneration?: "DONT_ALLOW" | "ALLOW_ADULT" | "ALLOW_ALL" | undefined;
+    includeSafetyAttributes?: boolean | undefined;
+    includeRaiReason?: boolean | undefined;
+    language?: string | undefined;
+    outputMimeType?: string | undefined;
+    outputCompressionQuality?: number | undefined;
+    addWatermark?: boolean | undefined;
+    labels?: Record<string, string> | undefined;
+    imageSize?: string | undefined;
+    enhancePrompt?: boolean | undefined;
+}>;
+type GoogleImageProviderOptions = z.infer<typeof googleImageProviderOptionsSchema>;
+declare module '@core-ai/core-ai' {
+    interface GenerateProviderOptions {
+        google?: GoogleGenerateProviderOptions;
+    }
+    interface EmbedProviderOptions {
+        google?: GoogleEmbedProviderOptions;
+    }
+    interface ImageProviderOptions {
+        google?: GoogleImageProviderOptions;
+    }
+}
+declare const googleProviderOptionsSchema: z.ZodObject<{
+    stopSequences: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    frequencyPenalty: z.ZodOptional<z.ZodNumber>;
+    presencePenalty: z.ZodOptional<z.ZodNumber>;
+    seed: z.ZodOptional<z.ZodNumber>;
+    topK: z.ZodOptional<z.ZodNumber>;
+}, "strict", z.ZodTypeAny, {
+    stopSequences?: string[] | undefined;
+    frequencyPenalty?: number | undefined;
+    presencePenalty?: number | undefined;
+    seed?: number | undefined;
+    topK?: number | undefined;
+}, {
+    stopSequences?: string[] | undefined;
+    frequencyPenalty?: number | undefined;
+    presencePenalty?: number | undefined;
+    seed?: number | undefined;
+    topK?: number | undefined;
+}>;
+type GoogleProviderOptions = GoogleGenerateProviderOptions;
+
+export { type GoogleEmbedProviderOptions, type GoogleGenAIProvider, type GoogleGenAIProviderOptions, type GoogleGenerateProviderOptions, type GoogleImageProviderOptions, type GoogleProviderOptions as GoogleModelProviderOptions, type GoogleReasoningMetadata, createGoogleGenAI, googleEmbedProviderOptionsSchema, googleGenerateProviderOptionsSchema, googleImageProviderOptionsSchema, googleProviderOptionsSchema };

package/dist/index.js

@@ -6,8 +6,8 @@ import {
   StructuredOutputNoObjectGeneratedError,
   StructuredOutputParseError,
   StructuredOutputValidationError,
-  createObjectStreamResult,
-  createStreamResult
+  createObjectStream,
+  createChatStream
 } from "@core-ai/core-ai";
 
 // src/chat-adapter.ts
@@ -15,6 +15,7 @@ import {
   FunctionCallingConfigMode
 } from "@google/genai";
 import { zodToJsonSchema } from "zod-to-json-schema";
+import { getProviderMetadata } from "@core-ai/core-ai";
 
 // src/model-capabilities.ts
 var DEFAULT_CAPABILITIES = {
@@ -24,6 +25,18 @@ var DEFAULT_CAPABILITIES = {
   }
 };
 var MODEL_CAPABILITIES = {
+  "gemini-3.1-pro": {
+    reasoning: {
+      thinkingParam: "thinkingLevel",
+      canDisableThinking: false
+    }
+  },
+  "gemini-3.1-flash-lite-preview": {
+    reasoning: {
+      thinkingParam: "thinkingLevel",
+      canDisableThinking: false
+    }
+  },
   "gemini-3-pro": {
     reasoning: {
       thinkingParam: "thinkingLevel",
@@ -83,6 +96,67 @@ function asObject(value) {
   return {};
 }
 
+// src/provider-options.ts
+import { z } from "zod";
+var googleGenerateProviderOptionsSchema = z.object({
+  stopSequences: z.array(z.string()).optional(),
+  frequencyPenalty: z.number().optional(),
+  presencePenalty: z.number().optional(),
+  seed: z.number().int().optional(),
+  topK: z.number().int().optional()
+}).strict();
+var googleEmbedProviderOptionsSchema = z.object({
+  taskType: z.string().optional(),
+  title: z.string().optional(),
+  mimeType: z.string().optional(),
+  autoTruncate: z.boolean().optional()
+}).strict();
+var googleImageProviderOptionsSchema = z.object({
+  outputGcsUri: z.string().optional(),
+  negativePrompt: z.string().optional(),
+  aspectRatio: z.string().optional(),
+  guidanceScale: z.number().optional(),
+  seed: z.number().int().optional(),
+  safetyFilterLevel: z.enum([
+    "BLOCK_LOW_AND_ABOVE",
+    "BLOCK_MEDIUM_AND_ABOVE",
+    "BLOCK_ONLY_HIGH",
+    "BLOCK_NONE"
+  ]).optional(),
+  personGeneration: z.enum(["DONT_ALLOW", "ALLOW_ADULT", "ALLOW_ALL"]).optional(),
+  includeSafetyAttributes: z.boolean().optional(),
+  includeRaiReason: z.boolean().optional(),
+  language: z.string().optional(),
+  outputMimeType: z.string().optional(),
+  outputCompressionQuality: z.number().int().min(0).max(100).optional(),
+  addWatermark: z.boolean().optional(),
+  labels: z.record(z.string(), z.string()).optional(),
+  imageSize: z.string().optional(),
+  enhancePrompt: z.boolean().optional()
+}).strict();
+function parseGoogleGenerateProviderOptions(providerOptions) {
+  const rawOptions = providerOptions?.google;
+  if (rawOptions === void 0) {
+    return void 0;
+  }
+  return googleGenerateProviderOptionsSchema.parse(rawOptions);
+}
+function parseGoogleEmbedProviderOptions(providerOptions) {
+  const rawOptions = providerOptions?.google;
+  if (rawOptions === void 0) {
+    return void 0;
+  }
+  return googleEmbedProviderOptionsSchema.parse(rawOptions);
+}
+function parseGoogleImageProviderOptions(providerOptions) {
+  const rawOptions = providerOptions?.google;
+  if (rawOptions === void 0) {
+    return void 0;
+  }
+  return googleImageProviderOptionsSchema.parse(rawOptions);
+}
+var googleProviderOptionsSchema = googleGenerateProviderOptionsSchema;
+
 // src/chat-adapter.ts
 var DEFAULT_STRUCTURED_OUTPUT_TOOL_NAME = "core_ai_generate_object";
 var DEFAULT_STRUCTURED_OUTPUT_TOOL_DESCRIPTION = "Return a JSON object that matches the requested schema.";
@@ -121,13 +195,19 @@ function convertMessages(messages) {
           });
           continue;
         }
+        const googleMeta = getProviderMetadata(
+          part.providerMetadata,
+          "google"
+        );
+        if (part.text.length === 0) {
+          continue;
+        }
         const thoughtPart = {
           text: part.text,
           thought: true
         };
-        const thoughtSignature = part.providerMetadata?.["thoughtSignature"];
-        if (typeof thoughtSignature === "string") {
-          thoughtPart["thoughtSignature"] = thoughtSignature;
+        if (typeof googleMeta?.thoughtSignature === "string") {
+          thoughtPart["thoughtSignature"] = googleMeta.thoughtSignature;
         }
         assistantParts.push(thoughtPart);
       }
@@ -239,11 +319,7 @@ function convertToolChoice(choice) {
   };
 }
 function getStructuredOutputToolName(options) {
-  const trimmedName = options.schemaName?.trim();
-  if (trimmedName && trimmedName.length > 0) {
-    return trimmedName;
-  }
-  return DEFAULT_STRUCTURED_OUTPUT_TOOL_NAME;
+  return options.schemaName?.trim() || DEFAULT_STRUCTURED_OUTPUT_TOOL_NAME;
 }
 function createStructuredOutputOptions(options) {
   const toolName = getStructuredOutputToolName(options);
@@ -261,7 +337,9 @@ function createStructuredOutputOptions(options) {
       type: "tool",
       toolName
     },
-    config: options.config,
+    temperature: options.temperature,
+    maxTokens: options.maxTokens,
+    topP: options.topP,
     providerOptions: options.providerOptions,
     signal: options.signal
   };
@@ -292,35 +370,39 @@ function inferMimeTypeFromUrl(url) {
   return "application/octet-stream";
 }
 function createGenerateRequest(modelId, options) {
+  const googleOptions = parseGoogleGenerateProviderOptions(
+    options.providerOptions
+  );
   const convertedMessages = convertMessages(options.messages);
-  const baseRequest = {
+  const requestConfig = {
+    ...convertedMessages.systemInstruction ? { systemInstruction: convertedMessages.systemInstruction } : {},
+    ...options.tools && Object.keys(options.tools).length > 0 ? { tools: convertTools(options.tools) } : {},
+    ...options.toolChoice ? { toolConfig: convertToolChoice(options.toolChoice) } : {},
+    ...mapSamplingToConfig(options),
+    ...mapReasoningToConfig(modelId, options, googleOptions),
+    ...mapGoogleProviderOptionsToConfig(googleOptions),
+    ...options.signal ? { abortSignal: options.signal } : {}
+  };
+  return {
     model: modelId,
     contents: convertedMessages.contents,
-    config: {
-      ...convertedMessages.systemInstruction ? { systemInstruction: convertedMessages.systemInstruction } : {},
-      ...options.tools && Object.keys(options.tools).length > 0 ? { tools: convertTools(options.tools) } : {},
-      ...options.toolChoice ? { toolConfig: convertToolChoice(options.toolChoice) } : {},
-      ...options.config?.temperature !== void 0 ? { temperature: options.config.temperature } : {},
-      ...options.config?.maxTokens !== void 0 ? { maxOutputTokens: options.config.maxTokens } : {},
-      ...options.config?.topP !== void 0 ? { topP: options.config.topP } : {},
-      ...options.config?.stopSequences ? { stopSequences: options.config.stopSequences } : {},
-      ...options.config?.frequencyPenalty !== void 0 ? { frequencyPenalty: options.config.frequencyPenalty } : {},
-      ...options.config?.presencePenalty !== void 0 ? { presencePenalty: options.config.presencePenalty } : {},
-      ...mapReasoningToConfig(modelId, options)
-    }
+    config: requestConfig
   };
-  const providerOptions = options.providerOptions;
-  if (!providerOptions) {
-    return baseRequest;
-  }
-  const providerConfig = asObject(providerOptions["config"]);
+}
+function mapSamplingToConfig(options) {
   return {
-    ...baseRequest,
-    ...providerOptions,
-    config: {
-      ...baseRequest.config,
-      ...providerConfig
-    }
+    ...options.temperature !== void 0 ? { temperature: options.temperature } : {},
+    ...options.maxTokens !== void 0 ? { maxOutputTokens: options.maxTokens } : {},
+    ...options.topP !== void 0 ? { topP: options.topP } : {}
+  };
+}
+function mapGoogleProviderOptionsToConfig(options) {
+  return {
+    ...options?.stopSequences ? { stopSequences: options.stopSequences } : {},
+    ...options?.frequencyPenalty !== void 0 ? { frequencyPenalty: options.frequencyPenalty } : {},
+    ...options?.presencePenalty !== void 0 ? { presencePenalty: options.presencePenalty } : {},
+    ...options?.seed !== void 0 ? { seed: options.seed } : {},
+    ...options?.topK !== void 0 ? { topK: options.topK } : {}
   };
 }
 function mapGenerateResponse(response) {
@@ -333,16 +415,6 @@ function mapGenerateResponse(response) {
   const finishReason = mapFinishReason(
     response.candidates?.[0]?.finishReason ?? void 0
   );
-  if (!response.candidates?.[0]) {
-    return {
-      parts,
-      content: content.length > 0 ? content : null,
-      reasoning: reasoning.length > 0 ? reasoning : null,
-      toolCalls,
-      finishReason: toolCalls.length > 0 ? "tool-calls" : finishReason,
-      usage: mapUsage(response)
-    };
-  }
   return {
     parts,
     content: content.length > 0 ? content : null,
@@ -385,6 +457,16 @@ async function* transformStream(stream) {
     },
     outputTokenDetails: {}
   };
+  const closeReasoning = () => {
+    if (!reasoningOpen) {
+      return null;
+    }
+    reasoningOpen = false;
+    return {
+      type: "reasoning-end",
+      providerMetadata: { google: {} }
+    };
+  };
   for await (const chunk of stream) {
     usage = mapUsage(chunk, usage);
     const reasoningDeltas = extractReasoningDeltas(chunk);
@@ -403,11 +485,9 @@ async function* transformStream(stream) {
       }
     }
     if (chunk.text) {
-      if (reasoningOpen) {
-        reasoningOpen = false;
-        yield {
-          type: "reasoning-end"
-        };
+      const reasoningEnd2 = closeReasoning();
+      if (reasoningEnd2) {
+        yield reasoningEnd2;
       }
       yield {
         type: "text-delta",
@@ -416,11 +496,9 @@ async function* transformStream(stream) {
     }
     const functionCalls = chunk.functionCalls ?? [];
     if (functionCalls.length > 0) {
-      if (reasoningOpen) {
-        reasoningOpen = false;
-        yield {
-          type: "reasoning-end"
-        };
+      const reasoningEnd2 = closeReasoning();
+      if (reasoningEnd2) {
+        yield reasoningEnd2;
       }
       sawToolCalls = true;
       for (const [index, functionCall] of functionCalls.entries()) {
@@ -464,10 +542,9 @@ async function* transformStream(stream) {
       finishReason = candidateFinishReason;
     }
   }
-  if (reasoningOpen) {
-    yield {
-      type: "reasoning-end"
-    };
+  const reasoningEnd = closeReasoning();
+  if (reasoningEnd) {
+    yield reasoningEnd;
   }
   for (const toolCall of bufferedToolCalls.values()) {
     yield {
@@ -484,16 +561,11 @@ async function* transformStream(stream) {
     usage
   };
 }
-function mapReasoningToConfig(modelId, options) {
+function mapReasoningToConfig(modelId, options, _googleProviderOptions) {
   if (!options.reasoning) {
     return {};
   }
   const capabilities = getGoogleModelCapabilities(modelId);
-  const providerConfig = asObject(options.providerOptions?.["config"]);
-  const providerThinkingConfig = asObject(providerConfig["thinkingConfig"]);
-  if (Object.keys(providerThinkingConfig).length > 0) {
-    return {};
-  }
   if (capabilities.reasoning.thinkingParam === "thinkingLevel") {
     return {
       thinkingConfig: {
@@ -523,11 +595,11 @@ function extractAssistantParts(response) {
       parts.push({
         type: "reasoning",
         text: thoughtText,
-        ...thoughtSignature ? {
-          providerMetadata: {
-            thoughtSignature
+        providerMetadata: {
+          google: {
+            ...thoughtSignature ? { thoughtSignature } : {}
           }
-        } : {}
+        }
       });
       continue;
     }
@@ -637,8 +709,10 @@ function createGoogleGenAIChatModel(client, modelId) {
   }
   async function streamChat(options) {
     const request = createGenerateRequest(modelId, options);
-    const stream = await callGenerateContentStreamApi(request);
-    return createStreamResult(transformStream(stream));
+    return createChatStream(
+      async () => transformStream(await callGenerateContentStreamApi(request)),
+      { signal: options.signal }
+    );
   }
   return {
     provider,
@@ -665,13 +739,16 @@ function createGoogleGenAIChatModel(client, modelId) {
       const structuredOptions = createStructuredOutputOptions(options);
       const stream = await streamChat(structuredOptions);
       const toolName = getStructuredOutputToolName(options);
-      return createObjectStreamResult(
+      return createObjectStream(
         transformStructuredOutputStream(
           stream,
           options.schema,
           provider,
           toolName
-        )
+        ),
+        {
+          signal: options.signal
+        }
       );
     }
   };
@@ -786,7 +863,12 @@ function validateStructuredToolArguments(schema, toolArguments, provider) {
 }
 function parseAndValidateStructuredPayload(schema, rawPayload, provider) {
   const parsedPayload = parseJson(rawPayload, provider);
-  return validateStructuredObject(schema, parsedPayload, provider, rawPayload);
+  return validateStructuredObject(
+    schema,
+    parsedPayload,
+    provider,
+    rawPayload
+  );
 }
 function parseJson(rawOutput, provider) {
   try {
@@ -839,13 +921,15 @@ function createGoogleGenAIEmbeddingModel(client, modelId) {
             }
           } : {}
         };
-        const providerOptions = options.providerOptions;
-        const request = providerOptions ? {
+        const googleOptions = parseGoogleEmbedProviderOptions(
+          options.providerOptions
+        );
+        const providerConfig = mapGoogleEmbedProviderOptionsToConfig(googleOptions);
+        const request = Object.keys(providerConfig).length > 0 ? {
           ...baseRequest,
-          ...providerOptions,
           config: {
             ...baseRequest.config,
-            ...asObject(providerOptions["config"])
+            ...providerConfig
           }
         } : baseRequest;
         const response = await client.models.embedContent(request);
@@ -870,6 +954,14 @@ function createGoogleGenAIEmbeddingModel(client, modelId) {
     }
   };
 }
+function mapGoogleEmbedProviderOptionsToConfig(options) {
+  return {
+    ...options?.taskType !== void 0 ? { taskType: options.taskType } : {},
+    ...options?.title !== void 0 ? { title: options.title } : {},
+    ...options?.mimeType !== void 0 ? { mimeType: options.mimeType } : {},
+    ...options?.autoTruncate !== void 0 ? { autoTruncate: options.autoTruncate } : {}
+  };
+}
 
 // src/image-model.ts
 function createGoogleGenAIImageModel(client, modelId) {
@@ -886,13 +978,15 @@ function createGoogleGenAIImageModel(client, modelId) {
             ...mapSizeToImageConfig(options.size)
           }
         };
-        const providerOptions = options.providerOptions;
-        const request = providerOptions ? {
+        const googleOptions = parseGoogleImageProviderOptions(
+          options.providerOptions
+        );
+        const providerConfig = mapGoogleImageProviderOptionsToConfig(googleOptions);
+        const request = Object.keys(providerConfig).length > 0 ? {
           ...baseRequest,
-          ...providerOptions,
           config: {
             ...baseRequest.config,
-            ...asObject(providerOptions["config"])
+            ...providerConfig
           }
         } : baseRequest;
         const response = await client.models.generateImages(request);
@@ -909,6 +1003,26 @@ function createGoogleGenAIImageModel(client, modelId) {
     }
   };
 }
+function mapGoogleImageProviderOptionsToConfig(options) {
+  return {
+    ...options?.outputGcsUri !== void 0 ? { outputGcsUri: options.outputGcsUri } : {},
+    ...options?.negativePrompt !== void 0 ? { negativePrompt: options.negativePrompt } : {},
+    ...options?.aspectRatio !== void 0 ? { aspectRatio: options.aspectRatio } : {},
+    ...options?.guidanceScale !== void 0 ? { guidanceScale: options.guidanceScale } : {},
+    ...options?.seed !== void 0 ? { seed: options.seed } : {},
+    ...options?.safetyFilterLevel !== void 0 ? { safetyFilterLevel: options.safetyFilterLevel } : {},
+    ...options?.personGeneration !== void 0 ? { personGeneration: options.personGeneration } : {},
+    ...options?.includeSafetyAttributes !== void 0 ? { includeSafetyAttributes: options.includeSafetyAttributes } : {},
+    ...options?.includeRaiReason !== void 0 ? { includeRaiReason: options.includeRaiReason } : {},
+    ...options?.language !== void 0 ? { language: options.language } : {},
+    ...options?.outputMimeType !== void 0 ? { outputMimeType: options.outputMimeType } : {},
+    ...options?.outputCompressionQuality !== void 0 ? { outputCompressionQuality: options.outputCompressionQuality } : {},
+    ...options?.addWatermark !== void 0 ? { addWatermark: options.addWatermark } : {},
+    ...options?.labels !== void 0 ? { labels: options.labels } : {},
+    ...options?.imageSize !== void 0 ? { imageSize: options.imageSize } : {},
+    ...options?.enhancePrompt !== void 0 ? { enhancePrompt: options.enhancePrompt } : {}
+  };
+}
 function mapSizeToImageConfig(size) {
   if (!size) {
     return {};
@@ -962,5 +1076,9 @@ function createGoogleGenAI(options = {}) {
   };
 }
 export {
-  createGoogleGenAI
+  createGoogleGenAI,
+  googleEmbedProviderOptionsSchema,
+  googleGenerateProviderOptionsSchema,
+  googleImageProviderOptionsSchema,
+  googleProviderOptionsSchema
 };

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@core-ai/google-genai",
-    "version": "0.5.1",
+    "version": "0.6.1",
     "description": "Google GenAI provider package for @core-ai/core-ai",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",
@@ -43,7 +43,7 @@
         "test:watch": "vitest"
     },
     "dependencies": {
-        "@core-ai/core-ai": "^0.5.1",
+        "@core-ai/core-ai": "^0.6.1",
         "@google/genai": "^1.42.0",
         "zod-to-json-schema": "^3.25.1"
     },
@@ -51,8 +51,9 @@
         "zod": "^3.25.0 || ^4.0.0"
     },
     "devDependencies": {
-        "@core-ai/eslint-config": "^0.0.0",
-        "@core-ai/typescript-config": "^0.0.0",
+        "@core-ai/eslint-config": "*",
+        "@core-ai/testing": "*",
+        "@core-ai/typescript-config": "*",
         "typescript": "^5.7.3",
         "vitest": "^3.2.4"
     }