@core-ai/google-genai 0.11.1 → 0.13.0

package/dist/index.d.ts

@@ -1,5 +1,5 @@
 import { GoogleGenAI } from '@google/genai';
-import { ChatModel, EmbeddingModel, ImageModel } from '@core-ai/core-ai';
+import { ChatModel, EmbeddingModel, ImageModel, ModelCapabilities } from '@core-ai/core-ai';
 import { z } from 'zod';
 
 type GoogleGenAIProviderOptions = {
@@ -19,6 +19,14 @@ type GoogleReasoningMetadata = {
     thoughtSignature?: string;
 };
 
+type GoogleModelCapabilities = {
+    reasoning: ModelCapabilities['reasoning'] & {
+        thinkingParam: 'thinkingLevel' | 'thinkingBudget';
+        canDisableThinking: boolean;
+    };
+};
+declare function getGoogleModelCapabilities(modelId: string): GoogleModelCapabilities;
+
 declare const googleGenerateProviderOptionsSchema: z.ZodObject<{
     stopSequences: z.ZodOptional<z.ZodArray<z.ZodString>>;
     frequencyPenalty: z.ZodOptional<z.ZodNumber>;
@@ -82,4 +90,4 @@ declare const googleProviderOptionsSchema: z.ZodObject<{
 }, z.core.$strict>;
 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 };
+export { type GoogleEmbedProviderOptions, type GoogleGenAIProvider, type GoogleGenAIProviderOptions, type GoogleGenerateProviderOptions, type GoogleImageProviderOptions, type GoogleModelCapabilities, type GoogleProviderOptions as GoogleModelProviderOptions, type GoogleReasoningMetadata, createGoogleGenAI, getGoogleModelCapabilities, googleEmbedProviderOptionsSchema, googleGenerateProviderOptionsSchema, googleImageProviderOptionsSchema, googleProviderOptionsSchema };

package/dist/index.js

@@ -20,49 +20,43 @@ import { getProviderMetadata, zodSchemaToJsonSchema } from "@core-ai/core-ai";
 import {
   stripModelDateSuffix
 } from "@core-ai/core-ai";
-var DEFAULT_CAPABILITIES = {
-  reasoning: {
-    thinkingParam: "thinkingBudget",
-    canDisableThinking: true
-  }
-};
-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",
-      canDisableThinking: false
-    }
-  },
-  "gemini-2.5-pro": {
-    reasoning: {
-      thinkingParam: "thinkingBudget",
-      canDisableThinking: false
-    }
-  },
-  "gemini-2.5-flash": {
-    reasoning: {
-      thinkingParam: "thinkingBudget",
-      canDisableThinking: true
-    }
-  },
-  "gemini-2.5-flash-lite": {
+var ALL_EFFORTS = [
+  "minimal",
+  "low",
+  "medium",
+  "high",
+  "max"
+];
+function createCapabilities(config) {
+  return {
     reasoning: {
-      thinkingParam: "thinkingBudget",
-      canDisableThinking: true
+      supported: true,
+      supportedEfforts: ALL_EFFORTS,
+      restrictsSamplingParams: false,
+      thinkingParam: config.thinkingParam,
+      canDisableThinking: config.canDisableThinking
     }
-  }
+  };
+}
+var DEFAULT_CAPABILITIES = createCapabilities({
+  thinkingParam: "thinkingBudget",
+  canDisableThinking: true
+});
+var REQUIRED_THINKING_BUDGET_CAPABILITIES = createCapabilities({
+  thinkingParam: "thinkingBudget",
+  canDisableThinking: false
+});
+var THINKING_LEVEL_CAPABILITIES = createCapabilities({
+  thinkingParam: "thinkingLevel",
+  canDisableThinking: false
+});
+var MODEL_CAPABILITIES = {
+  "gemini-3.1-pro": THINKING_LEVEL_CAPABILITIES,
+  "gemini-3.1-flash-lite-preview": THINKING_LEVEL_CAPABILITIES,
+  "gemini-3-pro": THINKING_LEVEL_CAPABILITIES,
+  "gemini-2.5-pro": REQUIRED_THINKING_BUDGET_CAPABILITIES,
+  "gemini-2.5-flash": DEFAULT_CAPABILITIES,
+  "gemini-2.5-flash-lite": DEFAULT_CAPABILITIES
 };
 var GOOGLE_THINKING_LEVEL_MAP = {
   minimal: "LOW",
@@ -715,6 +709,7 @@ function createGoogleGenAIChatModel(client, modelId) {
   return {
     provider,
     modelId,
+    capabilities: getGoogleModelCapabilities(modelId),
     generate: generateChat,
     stream: streamChat,
     async generateObject(options) {
@@ -1075,6 +1070,7 @@ function createGoogleGenAI(options = {}) {
 }
 export {
   createGoogleGenAI,
+  getGoogleModelCapabilities,
   googleEmbedProviderOptionsSchema,
   googleGenerateProviderOptionsSchema,
   googleImageProviderOptionsSchema,

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@core-ai/google-genai",
-    "version": "0.11.1",
+    "version": "0.13.0",
     "description": "Google GenAI provider package for @core-ai/core-ai",
     "license": "MIT",
     "author": "Omnifact (https://omnifact.ai)",
@@ -45,7 +45,7 @@
         "test:watch": "vitest"
     },
     "dependencies": {
-        "@core-ai/core-ai": "^0.11.1",
+        "@core-ai/core-ai": "^0.13.0",
         "@google/genai": "^1.42.0"
     },
     "peerDependencies": {