Merge branch 'main' into feat/kilo-auto-model-cache-prices

Author: chrarnoldus

src/app/api/internal/code-review-status/[reviewId]/route.test.ts

@@ -599,5 +599,27 @@ describe('POST /api/internal/code-review-status/[reviewId]', () => {
         expect.stringContaining('https://app.kilo.ai/')
       );
     });
+
+    it('suggests switching to a free model in the billing notice', async () => {
+      mockGetCodeReviewById.mockResolvedValue(makeReview());
+      mockHasPRCommentWithMarker.mockResolvedValue(false);
+
+      await POST(
+        makeRequest({
+          status: 'failed',
+          errorMessage: 'Insufficient credits',
+          terminalReason: 'billing',
+        }),
+        makeParams(REVIEW_ID)
+      );
+
+      expect(mockCreatePRComment).toHaveBeenCalledWith(
+        'inst-1',
+        'owner',
+        'repo',
+        1,
+        expect.stringContaining('switch to a free model')
+      );
+    });
   });
 });

src/app/api/internal/code-review-status/[reviewId]/route.ts

@@ -166,7 +166,7 @@ const BILLING_NOTICE_MARKER = '<!-- kilo-billing-notice -->';
 const BILLING_NOTICE_BODY = `${BILLING_NOTICE_MARKER}
 **Kilo Code Review could not run — your account is out of credits.**
 
-Add credits at [app.kilo.ai](https://app.kilo.ai/) to enable reviews on this change.`;
+[Add credits](https://app.kilo.ai/) or [switch to a free model](https://app.kilo.ai/code-reviews) to enable reviews on this change.`;
 
 /**
  * Read a review's usage data.

src/lib/kilo-auto-model.ts

@@ -14,6 +14,7 @@ import type {
   OpenRouterChatCompletionRequest,
   OpenRouterReasoningConfig,
 } from '@/lib/providers/openrouter/types';
+import { requestContainsImages } from '@/lib/providers/openrouter/request-helpers';
 import type { ModelSettings, OpenCodeSettings, Verbosity } from '@kilocode/db/schema-types';
 import type OpenAI from 'openai';
 
@@ -106,6 +107,11 @@ const BALANCED_CODE_MODEL: ResolvedAutoModel = {
   model: MINIMAX_CURRENT_MODEL_ID,
 };
 
+const BALANCED_IMAGE_MODEL: ResolvedAutoModel = {
+  model: KIMI_CURRENT_MODEL_ID,
+  reasoning: { enabled: true },
+};
+
 const BALANCED_MODE_TO_MODEL: Record<string, ResolvedAutoModel> = {
   plan: { model: KIMI_CURRENT_MODEL_ID, reasoning: { enabled: true } },
   general: { model: KIMI_CURRENT_MODEL_ID, reasoning: { enabled: true } },
@@ -164,7 +170,7 @@ export const KILO_AUTO_BALANCED_MODEL: AutoModel = {
   completion_price: '0.000003',
   input_cache_read_price: '0.000000225',
   input_cache_write_price: undefined,
-  supports_images: false,
+  supports_images: true,
   roocode_settings: {
     included_tools: ['edit_file'],
     excluded_tools: ['apply_diff'],
@@ -212,7 +218,8 @@ const legacyMapping: Record<string, AutoModel | undefined> = {
 export async function resolveAutoModel(
   model: string,
   modeHeader: string | null,
-  balancePromise: Promise<number>
+  balancePromise: Promise<number>,
+  hasImages: boolean
 ): Promise<ResolvedAutoModel> {
   const mappedModel =
     (Object.hasOwn(legacyMapping, model) ? legacyMapping[model] : null)?.id ?? model;
@@ -226,6 +233,9 @@ export async function resolveAutoModel(
   }
   const mode = modeHeader?.trim().toLowerCase() ?? '';
   if (mappedModel === KILO_AUTO_BALANCED_MODEL.id) {
+    if (hasImages) {
+      return BALANCED_IMAGE_MODEL;
+    }
     return (
       (Object.hasOwn(BALANCED_MODE_TO_MODEL, mode) ? BALANCED_MODE_TO_MODEL[mode] : null) ??
       BALANCED_CODE_MODEL
@@ -244,10 +254,12 @@ export async function applyResolvedAutoModel(
   featureHeader: FeatureValue | null,
   balancePromise: Promise<number>
 ) {
+  const hasImages = requestContainsImages(request);
   const resolved = await resolveAutoModel(
     model,
     featureHeader === 'kiloclaw' ? 'plan' : modeHeader,
-    balancePromise
+    balancePromise,
+    hasImages
   );
   request.body.model = resolved.model;
   if (resolved.reasoning) {

src/lib/models.ts

@@ -48,7 +48,7 @@ export async function getMonitoredModels() {
   const set = new Set<string>();
   for (const model of preferredModels) {
     if (isKiloAutoModel(model)) {
-      set.add((await resolveAutoModel(model, null, Promise.resolve(0))).model);
+      set.add((await resolveAutoModel(model, null, Promise.resolve(0), false)).model);
     } else {
       set.add(model);
     }

src/lib/providers/openrouter/request-helpers.ts

@@ -152,3 +152,44 @@ export function scrubOpenCodeSpecificProperties(request: OpenRouterChatCompletio
   delete body.usage;
   delete body.reasoningEffort;
 }
+
+export function requestContainsImages(request: GatewayRequest): boolean {
+  switch (request.kind) {
+    case 'chat_completions':
+      return request.body.messages.some(
+        msg =>
+          (msg.role === 'user' || msg.role === 'tool') &&
+          Array.isArray(msg.content) &&
+          msg.content.some(part => part.type === 'image_url')
+      );
+    case 'responses': {
+      if (!Array.isArray(request.body.input)) return false;
+      return request.body.input.some(item => {
+        if (typeof item === 'string') return false;
+        if (item.type === 'message') {
+          return (
+            Array.isArray(item.content) && item.content.some(part => part.type === 'input_image')
+          );
+        }
+        if (item.type === 'function_call_output') {
+          return (
+            Array.isArray(item.output) && item.output.some(part => part.type === 'input_image')
+          );
+        }
+        return false;
+      });
+    }
+    case 'messages':
+      return request.body.messages.some(
+        msg =>
+          Array.isArray(msg.content) &&
+          msg.content.some(
+            block =>
+              block.type === 'image' ||
+              (block.type === 'tool_result' &&
+                Array.isArray(block.content) &&
+                block.content.some(inner => inner.type === 'image'))
+          )
+      );
+  }
+}

src/tests/openrouter-models-sorting.approved.json

@@ -52,7 +52,8 @@
       "description": "Great balance of price and capability. Uses Kimi K2.5 for plan, general, architect, orchestrator, ask, debug; MiniMax M2.7 for build, explore, code.",
       "architecture": {
         "input_modalities": [
-          "text"
+          "text",
+          "image"
         ],
         "output_modalities": [
           "text"