廃止予定の Gemini モデルを CI で検知する ― 連続するデッドラインを取りこぼさない仕組み

#!/usr/bin/env python3
"""廃止予定の Gemini モデル指定をリポジトリから検出する。
残り日数に応じて warning（情報）か error（CIを落とす）かを切り替える。"""
 
import json
import os
import re
import sys
from datetime import date, datetime
 
# 残りこの日数を切ったら error として扱う（CIを失敗させる）
FAIL_WITHIN_DAYS = 30
# 走査対象の拡張子
SCAN_EXT = {".py", ".ts", ".tsx", ".js", ".jsx", ".mjs", ".json", ".yaml", ".yml", ".md", ".mdx", ".env", ".sh"}
# 走査から除外するディレクトリ
SKIP_DIRS = {".git", "node_modules", ".next", "dist", "build", "vendor", ".venv"}
 
 
def load_registry(path):
    with open(path, encoding="utf-8") as f:
        data = json.load(f)
    items = []
    for d in data["deprecations"]:
        shutdown = datetime.strptime(d["shutdown"], "%Y-%m-%d").date()
        # 単語境界で囲み、'gemini-2.0-flash' が 'gemini-2.0-flash-lite' に
        # 誤マッチしないようにする
        pattern = re.compile(r"(?<![\w.-])" + re.escape(d["model"]) + r"(?![\w-])")
        items.append({**d, "shutdown_date": shutdown, "pattern": pattern})
    return items
 
 
def iter_files(root):
    for dirpath, dirnames, filenames in os.walk(root):
        dirnames[:] = [d for d in dirnames if d not in SKIP_DIRS]
        for name in filenames:
            ext = os.path.splitext(name)[1].lower()
            if ext in SCAN_EXT or name.startswith(".env"):
                yield os.path.join(dirpath, name)
 
 
def scan(root, registry, today):
    hits = []
    for path in iter_files(root):
        try:
            with open(path, encoding="utf-8", errors="ignore") as f:
                lines = f.readlines()
        except OSError:
            continue
        for lineno, line in enumerate(lines, 1):
            for item in registry:
                if item["pattern"].search(line):
                    remaining = (item["shutdown_date"] - today).days
                    hits.append({
                        "path": os.path.relpath(path, root),
                        "line": lineno,
                        "model": item["model"],
                        "replacement": item["replacement"],
                        "remaining": remaining,
                        "text": line.strip()[:120],
                    })
    return hits
 
 
def main():
    registry_path = os.environ.get("DEPRECATION_REGISTRY", "deprecations.json")
    root = sys.argv[1] if len(sys.argv) > 1 else "."
    today = date.today()
 
    registry = load_registry(registry_path)
    hits = scan(root, registry, today)
 
    if not hits:
        print("OK: 廃止予定のモデル指定は見つかりませんでした。")
        return 0
 
    # 残り日数が少ない順に並べる
    hits.sort(key=lambda h: h["remaining"])
    has_error = False
    for h in hits:
        level = "ERROR" if h["remaining"] <= FAIL_WITHIN_DAYS else "WARN"
        if level == "ERROR":
            has_error = True
        when = f"残り{h['remaining']}日" if h["remaining"] >= 0 else f"{abs(h['remaining'])}日超過"
        print(f"[{level}] {h['path']}:{h['line']} "
              f"{h['model']} → {h['replacement']} ({when})")
        print(f"        {h['text']}")
 
    print(f"\n検出 {len(hits)} 件 / error 閾値 {FAIL_WITHIN_DAYS} 日以内")
    return 1 if has_error else 0
 
 
if __name__ == "__main__":
    sys.exit(main())

[ERROR] scripts/make_thumbnail.py:42 gemini-3.1-flash-image-preview → gemini-3.1-flash-image (残り8日)
        model="gemini-3.1-flash-image-preview",
[WARN] docs/sample.md:88 gemini-2.0-flash → gemini-3.5-flash (残り105日)
        response = client.generate(model="gemini-2.0-flash", ...)
 
検出 2 件 / error 閾値 30 日以内

# 前: 直前が \w . - のいずれでもない / 後: 直後が \w - のいずれでもない
pattern = re.compile(r"(?<![\w.-])" + re.escape(d["model"]) + r"(?![\w-])")

name: gemini-deprecation-guard
on: [push, pull_request]
 
jobs:
  guard:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - uses: actions/setup-python@v5
        with:
          python-version: "3.12"
      - name: 廃止モデル検知
        env:
          DEPRECATION_REGISTRY: ci/deprecations.json
        run: python ci/check_deprecated_models.py .

# .pre-commit-config.yaml
- repo: local
  hooks:
    - id: gemini-deprecation-guard
      name: Gemini 廃止モデル検知
      entry: python ci/check_deprecated_models.py
      language: system
      pass_filenames: false

import os
from google import genai
 
client = genai.Client(api_key=os.environ["YOUR_GEMINI_API_KEY"])
 
# API が返すモデル一覧を取得し、説明文に deprecation の手がかりがあるものを拾う
for m in client.models.list():
    name = m.name.split("/")[-1]
    desc = (getattr(m, "description", "") or "").lower()
    if "deprecat" in desc or "preview" in name:
        print(f"要確認: {name} — {desc[:80]}")