| """How SpellSpeak Tone reads a line (input format RENDER_VERSION). Shared by training and the runtime. |
| |
| Everyone in the scene comes first, as markers, so truncation can only cut the previous line: |
| |
| People: [Tomas] [Wren] [the player]. Mara to Tomas: "You still owe me ten coins." Before: Tomas: Lovely day. |
| """ |
| from __future__ import annotations |
|
|
| from contracts.schemas.line_tags import PLAYER |
|
|
| RENDER_VERSION = "linecls-input-0.1" |
|
|
|
|
| def who(n: str | None) -> str: |
| return "the player" if n == PLAYER else (n or "everyone") |
|
|
|
|
| def render(inp: dict) -> tuple[str, list[tuple[int, int]]]: |
| """The input text and, per target in the caller's order, the character span of its marker (brackets included). |
| |
| The addressee's marker always comes first, then the others in the caller's order. Training lines almost always |
| listed the addressee first, and the models learned that position (found in an early demo), so the |
| runtime renders every line the same way whatever order the harness passes.""" |
| targets = list(inp["targets"]) |
| to = inp.get("to") |
| order = ([to] + [t for t in targets if t != to]) if to in targets else targets |
| text, at = "People:", {} |
| for t in order: |
| start = len(text) + 1 |
| text += f" [{who(t)}]" |
| at[t] = (start, len(text)) |
| spans = [at[t] for t in targets] |
| text += f". {who(inp['speaker'])} to {who(inp.get('to'))}: \"{inp['text']}\"" |
| if inp.get("previous"): |
| prev = inp["previous"] |
| name, sep, rest = prev.partition(":") |
| text += f" Before: {who(name.strip()) if sep else ''}{':' if sep else ''}{rest if sep else prev}" |
| return text, spans |
|
|
|
|
| def span_mask(offsets: list[tuple[int, int]], spans: list[tuple[int, int]]) -> list[list[float]]: |
| """Per target, which tokens belong to its marker (special tokens have empty offsets and never do).""" |
| rows = [] |
| for a, b in spans: |
| row = [1.0 if (t > s and s >= a and t <= b) else 0.0 for s, t in offsets] |
| if not any(row): |
| row[0] = 1.0 |
| rows.append(row) |
| return rows |
|
|
