通知文の長さ指示が英語版だけ効かなかった理由と、生成後に幅で詰める実装

from google import genai
from google.genai import types
 
client = genai.Client(api_key="YOUR_API_KEY")
 
NOTIFICATION_SCHEMA = {
    "type": "object",
    "properties": {
        # 通知として最低限成立する一文。これは絶対に落とさない
        "lead": {"type": "string"},
        # 余白があれば載せる補足。丸ごと落としても意味が壊れないこと
        "detail": {"type": "string"},
    },
    "required": ["lead", "detail"],
    "propertyOrdering": ["lead", "detail"],
}
 
PROMPT = """次の配信内容から、プッシュ通知の文面を作ってください。
 
lead: 何が起きたかだけを述べる一文。固有名詞は入れない。
detail: lead を補う情報。作品名や作者名はここに入れる。
detail は削除されても lead だけで意味が通るように書くこと。
 
言語: {lang}
配信内容: {payload}
"""
 
def generate_notification(lang: str, payload: str) -> dict:
    res = client.models.generate_content(
        model="gemini-3.7-flash",
        contents=PROMPT.format(lang=lang, payload=payload),
        config=types.GenerateContentConfig(
            response_mime_type="application/json",
            response_schema=NOTIFICATION_SCHEMA,
            temperature=0.4,
        ),
    )
    import json
    return json.loads(res.text)

import re
import unicodedata
 
# 前の文字に吸着させる修飾子: ZWJ・異体字セレクタ・キーキャップ・肌の色・結合記号
_MODIFIER = re.compile(
    r"[‍️︎⃣\U0001F3FB-\U0001F3FF̀-ͯ]"
)
 
# Ambiguous を幅2で数えるロケール（CJK フォント環境）
_WIDE_AMBIGUOUS_LANGS = {"ja", "zh", "zh-TW", "zh-CN", "ko"}
 
 
def grapheme_clusters(text: str) -> list[str]:
    """結合文字を直前の文字に吸着させて、切ってはいけない単位にまとめる。"""
    out: list[str] = []
    for ch in text:
        if out and _MODIFIER.match(ch):
            out[-1] += ch
        elif out and _MODIFIER.match(out[-1][-1]):
            # ZWJ の直後の文字は前のクラスタの一部
            out[-1] += ch
        else:
            out.append(ch)
    return out
 
 
def display_width(text: str, lang: str) -> int:
    wide_ambiguous = lang.split("-")[0] in _WIDE_AMBIGUOUS_LANGS
    total = 0
    for cluster in grapheme_clusters(text):
        head = cluster[0]
        eaw = unicodedata.east_asian_width(head)
        if eaw in ("W", "F"):
            total += 2
        elif eaw == "A":
            total += 2 if wide_ambiguous else 1
        elif ord(head) >= 0x1F000:  # 絵文字は Neutral 判定でも 2 幅で描画される
            total += 2
        else:
            total += 1
    return total
 
 
def cut_to_width(text: str, limit: int, lang: str) -> str:
    """書記素クラスタ単位で、表示幅が limit を超えない位置まで切る。"""
    used = 0
    out: list[str] = []
    for cluster in grapheme_clusters(text):
        w = display_width(cluster, lang)
        if used + w > limit:
            break
        out.append(cluster)
        used += w
    return "".join(out)
 
 
def fit(text: str, limit: int, lang: str) -> str:
    """収まっていればそのまま、超えていれば語境界を尊重して詰める。"""
    if display_width(text, lang) <= limit:
        return text
 
    ellipsis = "…"
    body = cut_to_width(text, limit - display_width(ellipsis, lang), lang)
 
    # 空白区切りの言語では、語の途中で切れた場合だけ手前の語境界へ戻す
    if " " in body and len(text) > len(body) and not text[len(body)].isspace():
        head = body.rsplit(" ", 1)[0]
        if head:
            body = head
 
    return body.rstrip(" ,.;:、。") + ellipsis
 
 
def build_notification(parts: dict, limit: int, lang: str) -> str:
    """lead は必ず残し、detail は入るときだけ足す。"""
    joiner = "" if lang.split("-")[0] in _WIDE_AMBIGUOUS_LANGS else " "
    terminator = "。" if lang.split("-")[0] == "ja" else "."
 
    lead = parts["lead"].rstrip("。.")
    detail = parts.get("detail", "").rstrip("。.")
 
    full = f"{lead}{terminator}{joiner}{detail}{terminator}"
    if display_width(full, lang) <= limit:
        return full
 
    # detail を丸ごと落とす。これでも収まらないときだけ lead を詰める
    lead_only = f"{lead}{terminator}"
    if display_width(lead_only, lang) <= limit:
        return lead_only
 
    return fit(lead_only, limit, lang)

def assert_notification_safe(text: str, limit: int, lang: str) -> None:
    w = display_width(text, lang)
    assert w <= limit                          # 1. 幅予算に収まっている
    assert not _MODIFIER.match(text[-1])       # 2. 不可視の制御文字で終わっていない
    if text.endswith("…"):
        prev = text[-2]
        assert not (prev.isascii() and prev.isalpha())  # 3. 語の途中で切れていない

def assert_notification_safe(text: str, original: str, limit: int, lang: str) -> None:
    # 1. 幅予算に収まっていること
    w = display_width(text, lang)
    assert w <= limit, f"[{lang}] width {w} > {limit}: {text!r}"
 
    # 2. 不可視の制御文字で終わっていないこと（ZWJ 割れの検出）
    assert not _MODIFIER.match(text[-1]), f"[{lang}] broken grapheme at tail: {text!r}"
 
    # 3. 切っているなら、切断点が元の文字列の語境界であること
    if text.endswith("…"):
        body = text[:-1]
        assert original.startswith(body), f"[{lang}] not a prefix of source: {text!r}"
        nxt = original[len(body):len(body) + 1]
        cut_inside_word = (
            nxt.isascii() and nxt.isalpha()
            and body[-1].isascii() and body[-1].isalpha()
        )
        assert not cut_inside_word, f"[{lang}] cut inside a word: {text!r}"