Function _completion

tests/test_rl_logging.py:54–70

(*, content: str | None = "hi there", tool_calls: list | None = None,
                choices: list | None = None)

def _completion(*, content: str | None = "hi there", tool_calls: list | None = None,
                choices: list | None = None) -> ChatResponse:
    message: dict = {"role": "assistant"}
    if content is not None:
        message["content"] = content
    if tool_calls is not None:
        message["tool_calls"] = tool_calls
    if choices is None:
        choices = [{"index": 0, "message": message, "finish_reason": "stop"}]
    return ChatResponse.openai_completion({
        "id": "chatcmpl-test",
        "object": "chat.completion",
        "created": 1700000000,
        "model": "gpt-test",
        "choices": choices,
        "usage": {"prompt_tokens": 5, "completion_tokens": 3, "total_tokens": 8},
    })


def _read_only_trace(log_dir: Path) -> dict: