diff --git a/llm/cli.py b/llm/cli.py index 18b5db2..8fb3f9e 100644 --- a/llm/cli.py +++ b/llm/cli.py @@ -2106,7 +2106,7 @@ def logs_list( json.loads(row["token_details"]) if row["token_details"] else None, ) if token_usage: - click.echo("## Token usage:\n\n{}\n".format(token_usage)) + click.echo("## Token usage\n\n{}\n".format(token_usage)) @cli.group( diff --git a/tests/test_llm_logs.py b/tests/test_llm_logs.py index 92a36a8..312c5de 100644 --- a/tests/test_llm_logs.py +++ b/tests/test_llm_logs.py @@ -112,7 +112,7 @@ def test_logs_text(log_path, usage): "## Response\n\n" 'response\n```python\nprint("hello word")\n```\n\n' ) - + ("## Token usage:\n\n2 input, 5 output\n\n" if usage else "") + + ("## Token usage\n\n2 input, 5 output\n\n" if usage else "") + ( "# YYYY-MM-DDTHH:MM:SS conversation: abc123 id: xxx\n\n" "Model: **davinci**\n\n" @@ -121,7 +121,7 @@ def test_logs_text(log_path, usage): "## Response\n\n" 'response\n```python\nprint("hello word")\n```\n\n' ) - + ("## Token usage:\n\n2 input, 5 output\n\n" if usage else "") + + ("## Token usage\n\n2 input, 5 output\n\n" if usage else "") + ( "# YYYY-MM-DDTHH:MM:SS conversation: abc123 id: xxx\n\n" "Model: **davinci**\n\n" @@ -130,7 +130,7 @@ def test_logs_text(log_path, usage): "## Response\n\n" 'response\n```python\nprint("hello word")\n```\n\n' ) - + ("## Token usage:\n\n2 input, 5 output\n\n" if usage else "") + + ("## Token usage\n\n2 input, 5 output\n\n" if usage else "") ) assert output == expected