Gemini 3.5 Flash がデフォルトになった後、Flash と Pro をタスク単位で振り分ける

# .antigravity/agents.yaml
# タスク種別ごとにモデルを固定し、入口で振り分ける
agents:
  quick-edit:
    model: gemini-3.5-flash
    description: 整形・命名置換・コメント追記など、間違えても安全な軽作業
    temperature: 0.1
    max_context_files: 8
 
  bulk-refactor:
    model: gemini-3.5-flash
    description: 複数ファイルの機械的な置換。広い視野は要るが判断は単純
    temperature: 0.1
    max_context_files: 40
 
  design-change:
    model: gemini-3.5-pro
    description: 非同期境界・エラー設計・状態管理など、間違いが下流に伝播する変更
    temperature: 0.2
    max_context_files: 24
 
  architecture:
    model: gemini-3.5-pro
    description: データフロー再設計・モジュール整合。一度で通したい重い仕事
    temperature: 0.3
    max_context_files: 80

# route.py — タスク記述から行き先エージェントを決める最小ルーター
import re
 
# 「重い決定」を示唆する語。ここに当たれば Pro 側へ寄せる
HEAVY_SIGNALS = (
    "設計", "アーキテクチャ", "状態管理", "非同期", "エラーハンドリング",
    "データフロー", "リファクタ", "整合", "マイグレ", "並行",
)
 
# 「広い文脈」を示唆する語
WIDE_SIGNALS = ("複数ファイル", "全体", "横断", "依存関係", "モジュール間")
 
 
def route(task: str) -> str:
    heavy = any(s in task for s in HEAVY_SIGNALS)
    wide = any(s in task for s in WIDE_SIGNALS)
 
    if heavy and wide:
        return "architecture"
    if heavy:
        return "design-change"
    if wide:
        return "bulk-refactor"
    return "quick-edit"
 
 
if __name__ == "__main__":
    samples = [
        "変数名を userId から accountId に一括置換したい",
        "認証フローのエラーハンドリングを設計し直したい",
        "複数ファイルにまたがる import を整理したい",
        "状態管理をモジュール間で整合するよう再設計したい",
    ]
    for s in samples:
        print(f"{route(s):16} <- {s}")

# compare.py — 同じ入力を2モデルに渡し、出力の差分を並べる
import subprocess
import difflib
import sys
 
# 普段よく回す代表タスク。プロジェクトに合わせて数件選ぶ
CASES = [
    "この関数に型注釈を付けてください",
    "このコンポーネントのpropsをinterfaceに切り出してください",
    "このエラーハンドリングを早期returnに整理してください",
]
 
 
def run(model: str, prompt: str, fixture: str) -> str:
    # antigravity CLI を非対話で呼ぶ想定。環境に合わせて置き換える
    result = subprocess.run(
        ["antigravity", "run", "--model", model, "--input", fixture, "--prompt", prompt],
        capture_output=True, text=True, timeout=120,
    )
    return result.stdout
 
 
def main(fixture: str):
    for prompt in CASES:
        a = run("gemini-3.5-flash", prompt, fixture)
        b = run("gemini-3.5-pro", prompt, fixture)
        diff = list(difflib.unified_diff(
            a.splitlines(), b.splitlines(),
            fromfile="flash", tofile="pro", lineterm="",
        ))
        changed = len(diff)
        mark = "差分大" if changed > 40 else "差分小"
        print(f"[{mark}/{changed}行] {prompt}")
        if changed > 40:
            print("\n".join(diff[:30]))
            print("...")
 
 
if __name__ == "__main__":
    main(sys.argv[1] if len(sys.argv) > 1 else "fixtures/sample.ts")