feat(auto-routing): benchmark-driven decision engine and kilo-auto/efficient

.github/workflows/deploy-workers.yml

@@ -49,6 +49,10 @@ jobs:
         with:
           apiToken: ${{ secrets.CLOUDFLARE_API_TOKEN }}
           workingDirectory: ${{ inputs.worker }}
+          # Workers that define a `predeploy` script (e.g. D1 migrations) run it
+          # right before deploy; all other workers are unaffected.
+          preCommands: |
+            if [ "$(jq -r '.scripts.predeploy // empty' package.json)" != "" ]; then pnpm run predeploy; fi
           command: deploy
 
   detect-changes:
@@ -150,4 +154,8 @@ jobs:
         with:
           apiToken: ${{ secrets.CLOUDFLARE_API_TOKEN }}
           workingDirectory: ${{ matrix.worker }}
+          # Workers that define a `predeploy` script (e.g. D1 migrations) run it
+          # right before deploy; all other workers are unaffected.
+          preCommands: |
+            if [ "$(jq -r '.scripts.predeploy // empty' package.json)" != "" ]; then pnpm run predeploy; fi
           command: deploy

apps/web/.env.development.local.example

@@ -19,6 +19,9 @@ AUTO_TRIAGE_URL=http://localhost:8791
 # @url auto-routing
 AUTO_ROUTING_WORKER_URL=http://localhost:8810
 
+# @url auto-routing-benchmark
+AUTO_ROUTING_BENCHMARK_WORKER_URL=http://localhost:8814
+
 # @url cloudflare-security-sync
 SECURITY_SYNC_WORKER_URL=http://localhost:8812
 

apps/web/src/app/admin/api/auto-routing/benchmark-config/route.test.ts

@@ -0,0 +1,123 @@
+import { NextRequest } from 'next/server';
+import type { User } from '@kilocode/db';
+import {
+  getBenchmarkConfig,
+  updateBenchmarkConfig,
+} from '@/lib/ai-gateway/auto-routing-benchmark-admin-client';
+import { getUserFromAuth } from '@/lib/user/server';
+import type { KiloExclusiveModel } from '@/lib/ai-gateway/providers/kilo-exclusive-model';
+import type * as ModelsModule from '@/lib/ai-gateway/models';
+
+jest.mock('@/lib/user/server', () => ({
+  getUserFromAuth: jest.fn(),
+}));
+
+jest.mock('@/lib/ai-gateway/auto-routing-benchmark-admin-client', () => ({
+  getBenchmarkConfig: jest.fn(),
+  updateBenchmarkConfig: jest.fn(),
+}));
+
+// Stub the catalog so tests don't depend on any specific provider file.
+// 'test-exclusive/alibaba-only' maps to the alibaba gateway (chat_completions only).
+jest.mock('@/lib/ai-gateway/models', () => {
+  const actual = jest.requireActual<typeof ModelsModule>('@/lib/ai-gateway/models');
+  const stubModel: KiloExclusiveModel = {
+    public_id: 'test-exclusive/alibaba-only',
+    display_name: 'Test Alibaba-only',
+    description: 'stub for unit tests',
+    context_length: 8192,
+    max_completion_tokens: 4096,
+    status: 'public',
+    flags: [],
+    gateway: 'alibaba',
+    internal_id: 'stub-internal',
+    pricing: null,
+    exclusive_to: [],
+    inference_provider_restriction: [],
+  };
+  return {
+    ...actual,
+    findKiloExclusiveModel: (id: string) =>
+      id === 'test-exclusive/alibaba-only' ? stubModel : actual.findKiloExclusiveModel(id),
+  };
+});
+
+import { PUT } from './route';
+
+const mockGetUserFromAuth = jest.mocked(getUserFromAuth);
+const mockGetBenchmarkConfig = jest.mocked(getBenchmarkConfig);
+const mockUpdateBenchmarkConfig = jest.mocked(updateBenchmarkConfig);
+
+// Test-fixture boundary: only the fields the route actually reads.
+function adminUserFixture(): User {
+  return { id: 'admin_123', google_user_email: 'admin@kilocode.ai' } as Partial<User> as User;
+}
+
+function putRequest(body: unknown) {
+  return new NextRequest('http://localhost:3000/admin/api/auto-routing/benchmark-config', {
+    method: 'PUT',
+    body: JSON.stringify(body),
+    headers: { 'content-type': 'application/json' },
+  });
+}
+
+const validConfig = {
+  classifierModels: ['google/gemini-2.5-flash-lite'],
+  deciderModels: [{ id: 'openai/gpt-5-mini', reasoningEffort: null }],
+  minAccuracy: 0.7,
+  switchCostFactor: 3,
+  maxConcurrency: 4,
+  benchmarkUserId: null,
+  classifierRepetitions: 1,
+  deciderRepetitions: 1,
+  classifierMaxP95LatencyMs: 1000,
+  updatedAt: null,
+  updatedBy: null,
+};
+
+describe('PUT /admin/api/auto-routing/benchmark-config', () => {
+  beforeEach(() => {
+    jest.clearAllMocks();
+    mockGetUserFromAuth.mockResolvedValue({
+      user: adminUserFixture(),
+      authFailedResponse: null,
+    });
+    mockUpdateBenchmarkConfig.mockResolvedValue({
+      status: 200,
+      body: { config: validConfig },
+    });
+    mockGetBenchmarkConfig.mockResolvedValue({ status: 200, body: { config: null } });
+  });
+
+  it('forwards a config whose decider models all serve every gateway chat API', async () => {
+    const response = await PUT(putRequest(validConfig));
+    expect(response.status).toBe(200);
+    expect(mockUpdateBenchmarkConfig).toHaveBeenCalledWith(validConfig, 'admin@kilocode.ai');
+  });
+
+  it('rejects with 400 listing decider models not servable on all gateway chat APIs', async () => {
+    const response = await PUT(
+      putRequest({
+        ...validConfig,
+        deciderModels: [
+          { id: 'openai/gpt-5-mini', reasoningEffort: null },
+          { id: 'test-exclusive/alibaba-only', reasoningEffort: null },
+        ],
+      })
+    );
+
+    expect(response.status).toBe(400);
+    const body = (await response.json()) as { error: string };
+    expect(body.error).toContain('test-exclusive/alibaba-only');
+    expect(body.error).toContain('chat_completions');
+    expect(body.error).not.toContain('openai/gpt-5-mini (');
+    expect(mockUpdateBenchmarkConfig).not.toHaveBeenCalled();
+  });
+
+  it('rejects a schema-invalid config with 400', async () => {
+    const response = await PUT(putRequest({ classifierModels: 'oops' }));
+    expect(response.status).toBe(400);
+    await expect(response.json()).resolves.toEqual({ error: 'Invalid benchmark config' });
+    expect(mockUpdateBenchmarkConfig).not.toHaveBeenCalled();
+  });
+});

apps/web/src/app/admin/api/auto-routing/benchmark-config/route.ts

@@ -0,0 +1,57 @@
+import { BenchmarkConfigSchema } from '@kilocode/auto-routing-contracts';
+import type { NextRequest } from 'next/server';
+import { NextResponse } from 'next/server';
+import {
+  getBenchmarkConfig,
+  updateBenchmarkConfig,
+} from '@/lib/ai-gateway/auto-routing-benchmark-admin-client';
+import {
+  gatewayChatApisForModel,
+  modelServesAllGatewayChatApis,
+} from '@/lib/ai-gateway/model-api-kinds';
+import { getUserFromAuth } from '@/lib/user/server';
+
+export async function GET() {
+  const { authFailedResponse } = await getUserFromAuth({ adminOnly: true });
+  if (authFailedResponse) return authFailedResponse;
+
+  const result = await getBenchmarkConfig();
+  return NextResponse.json(result.body, { status: result.status });
+}
+
+export async function PUT(request: NextRequest) {
+  const { authFailedResponse, user } = await getUserFromAuth({ adminOnly: true });
+  if (authFailedResponse) return authFailedResponse;
+
+  let rawBody: unknown;
+  try {
+    rawBody = await request.json();
+  } catch {
+    return NextResponse.json({ error: 'Invalid JSON body' }, { status: 400 });
+  }
+
+  const parsed = BenchmarkConfigSchema.safeParse(rawBody);
+  if (!parsed.success) {
+    return NextResponse.json({ error: 'Invalid benchmark config' }, { status: 400 });
+  }
+
+  // Routing-table candidates carry no per-protocol metadata, so every decider
+  // model must be servable on ALL gateway chat API kinds by the provider the
+  // gateway would route it to.
+  const unsupported = parsed.data.deciderModels
+    .map(m => m.id)
+    .filter(id => !modelServesAllGatewayChatApis(id))
+    .map(id => `${id} (supports: ${gatewayChatApisForModel(id).join(', ') || 'none'})`);
+  if (unsupported.length > 0) {
+    return NextResponse.json(
+      {
+        error: `Decider models must support all gateway chat APIs (chat_completions, responses, messages): ${unsupported.join('; ')}`,
+      },
+      { status: 400 }
+    );
+  }
+
+  const email = user?.google_user_email ?? '';
+  const result = await updateBenchmarkConfig(parsed.data, email);
+  return NextResponse.json(result.body, { status: result.status });
+}

apps/web/src/app/admin/api/auto-routing/benchmark-routing-table/route.ts

@@ -0,0 +1,11 @@
+import { NextResponse } from 'next/server';
+import { getBenchmarkRoutingTable } from '@/lib/ai-gateway/auto-routing-benchmark-admin-client';
+import { getUserFromAuth } from '@/lib/user/server';
+
+export async function GET() {
+  const { authFailedResponse } = await getUserFromAuth({ adminOnly: true });
+  if (authFailedResponse) return authFailedResponse;
+
+  const result = await getBenchmarkRoutingTable();
+  return NextResponse.json(result.body, { status: result.status });
+}

apps/web/src/app/admin/api/auto-routing/benchmark-runs/route.ts

@@ -0,0 +1,36 @@
+import { StartBenchmarkRunRequestSchema } from '@kilocode/auto-routing-contracts';
+import type { NextRequest } from 'next/server';
+import { NextResponse } from 'next/server';
+import {
+  listBenchmarkRuns,
+  startBenchmarkRun,
+} from '@/lib/ai-gateway/auto-routing-benchmark-admin-client';
+import { getUserFromAuth } from '@/lib/user/server';
+
+export async function GET() {
+  const { authFailedResponse } = await getUserFromAuth({ adminOnly: true });
+  if (authFailedResponse) return authFailedResponse;
+
+  const result = await listBenchmarkRuns();
+  return NextResponse.json(result.body, { status: result.status });
+}
+
+export async function POST(request: NextRequest) {
+  const { authFailedResponse } = await getUserFromAuth({ adminOnly: true });
+  if (authFailedResponse) return authFailedResponse;
+
+  let rawBody: unknown;
+  try {
+    rawBody = await request.json();
+  } catch {
+    return NextResponse.json({ error: 'Invalid JSON body' }, { status: 400 });
+  }
+
+  const parsed = StartBenchmarkRunRequestSchema.safeParse(rawBody);
+  if (!parsed.success) {
+    return NextResponse.json({ error: 'Invalid start benchmark run request' }, { status: 400 });
+  }
+
+  const result = await startBenchmarkRun(parsed.data.kind, parsed.data.force);
+  return NextResponse.json(result.body, { status: result.status });
+}