Function _compose_deterministic

uncommon_route/composition.py:346–405

(
    messages: list[dict[str, Any]],
    artifact_store: ArtifactStore,
    policy: CompositionPolicy,
)

def _compose_deterministic(
    messages: list[dict[str, Any]],
    artifact_store: ArtifactStore,
    policy: CompositionPolicy,
) -> CompositionResult:
    rewritten: list[dict[str, Any]] = []
    artifact_ids: list[str] = []
    compacted_messages = 0
    offloaded_messages = 0
    input_before = 0

    for idx, msg in enumerate(messages):
        if not isinstance(msg, dict):
            rewritten.append(msg)
            continue

        content = msg.get("content")
        text = _content_to_text(content)
        if text is None:
            rewritten.append(dict(msg))
            continue

        input_before += estimate_tokens(text)
        new_msg = dict(msg)
        compacted = _safe_compact_text(text)

        if msg.get("role") == "tool" and estimate_tokens(compacted) >= policy.tool_offload_threshold_tokens:
            tool_call_id = str(msg.get("tool_call_id", ""))
            tool_name = _infer_tool_name(messages, idx, tool_call_id)
            content_type = "application/json" if _looks_like_json(compacted) else "text/plain"
            record = artifact_store.store_text(
                text,
                kind="tool-result",
                role="tool",
                tool_name=tool_name,
                tool_call_id=tool_call_id,
                content_type=content_type,
            )
            new_msg["content"] = _build_artifact_stub(compacted, record, policy)
            artifact_ids.append(record.id)
            offloaded_messages += 1
        elif msg.get("role") == "tool" and compacted != text:
            new_msg["content"] = compacted
            compacted_messages += 1
        elif isinstance(content, str):
            if len(content) >= policy.inline_compact_threshold_chars and compacted != content:
                new_msg["content"] = compacted
                compacted_messages += 1

        rewritten.append(new_msg)

    input_after = _estimate_messages_tokens(rewritten)
    return CompositionResult(
        messages=rewritten,
        input_tokens_before=input_before,
        input_tokens_after=input_after,
        artifact_ids=artifact_ids,
        compacted_messages=compacted_messages,