Function _jsonl_response

tests/extraction/llm/test_codex_provider.py:24–38

Return a bash snippet that emits JSONL events mimicking codex output.

(text: str, input_tokens: int = 0, output_tokens: int = 0)

def _jsonl_response(text: str, input_tokens: int = 0, output_tokens: int = 0) -> str:
    """Return a bash snippet that emits JSONL events mimicking codex output."""
    item_event = json.dumps(
        {
            "type": "item.completed",
            "item": {"id": "item_0", "type": "agent_message", "text": text},
        }
    )
    usage_event = json.dumps(
        {
            "type": "turn.completed",
            "usage": {"input_tokens": input_tokens, "output_tokens": output_tokens},
        }
    )
    return f"echo '{item_event}'\necho '{usage_event}'\n"


class TestParseJsonl(unittest.TestCase):