diff --git a/llm/models.py b/llm/models.py index c4a00db6..5bf9f11c 100644 --- a/llm/models.py +++ b/llm/models.py @@ -297,7 +297,7 @@ def log_to_db(self, db): "duration_ms": self.duration_ms(), "datetime_utc": self.datetime_utc(), "input_tokens": self.input_tokens, - "output_tokens": self.input_tokens, + "output_tokens": self.output_tokens, "token_details": ( json.dumps(self.token_details) if self.token_details else None ), diff --git a/tests/test_chat.py b/tests/test_chat.py index d8decde6..1a31f290 100644 --- a/tests/test_chat.py +++ b/tests/test_chat.py @@ -79,7 +79,7 @@ def test_chat_basic(mock_model, logs_db): "duration_ms": ANY, "datetime_utc": ANY, "input_tokens": 2, - "output_tokens": 2, + "output_tokens": 1, "token_details": None, }, ]