engine/tools/state/mark_beat.py

"""mark_beat — 목표 비트 달성 기록 (1파일 = 1함수, FunctionTool)."""

from google.adk.tools import ToolContext


def mark_beat(beat: str, tool_context: ToolContext) -> dict:
    """장면 목표 비트가 달성되었음을 기록한다.

    Args:
        beat: 달성된 비트의 짧은 식별 문구
    """
    hits = list(tool_context.state.get("pocket_beats_hit", []))
    if beat not in hits:
        hits.append(beat)
    tool_context.state["pocket_beats_hit"] = hits
    return {"status": "ok", "beats_hit": hits}