@ljoukov/llm 7.0.4 → 7.0.5

package/dist/index.js

@@ -7135,25 +7135,16 @@ function toGeminiThinkingLevel(thinkingLevel) {
       return ThinkingLevel.HIGH;
   }
 }
-function toGemini25ProThinkingBudget(thinkingLevel) {
-  switch (thinkingLevel) {
-    case "low":
-      return 256;
-    case "medium":
-      return 4096;
-    case "high":
-      return 32768;
-  }
-}
 function resolveGeminiThinkingConfig(modelId, thinkingLevel) {
   if (isGeminiImageModelId(modelId) || modelId === "gemini-flash-lite-latest") {
     return void 0;
   }
   if (thinkingLevel) {
-    if (modelId === "gemini-2.5-pro") {
+    const thinkingBudget = resolveGeminiThinkingBudget(modelId, thinkingLevel);
+    if (thinkingBudget !== void 0) {
       return {
         includeThoughts: true,
-        thinkingBudget: toGemini25ProThinkingBudget(thinkingLevel)
+        thinkingBudget
       };
     }
     return {
@@ -7175,6 +7166,39 @@ function resolveGeminiThinkingConfig(modelId, thinkingLevel) {
       return { includeThoughts: true };
   }
 }
+function resolveGeminiThinkingBudget(modelId, thinkingLevel) {
+  if (modelId === "gemini-2.5-pro") {
+    switch (thinkingLevel) {
+      case "low":
+        return 256;
+      case "medium":
+        return 4096;
+      case "high":
+        return 32768;
+    }
+  }
+  if (modelId === "gemini-flash-latest") {
+    switch (thinkingLevel) {
+      case "low":
+        return 256;
+      case "medium":
+        return 8192;
+      case "high":
+        return 24576;
+    }
+  }
+  if (modelId === "gemini-3-flash-preview") {
+    switch (thinkingLevel) {
+      case "low":
+        return 256;
+      case "medium":
+        return 8192;
+      case "high":
+        return 16384;
+    }
+  }
+  return void 0;
+}
 function decodeInlineDataBuffer(base64) {
   try {
     return Buffer5.from(base64, "base64");