diff --git a/dreadnode/agent/agent.py b/dreadnode/agent/agent.py
index a680b87..80bf9f5 100644
--- a/dreadnode/agent/agent.py
+++ b/dreadnode/agent/agent.py
@@ -299,6 +299,8 @@ async def _generate(
             post_transforms.append(post_transform)
 
         try:
+            from dreadnode import log_output, task_and_run
+
             messages = rg.caching.apply_cache_mode_to_messages(self.caching, [messages])[0]
 
             logger.trace(f"Generating with model '{self.generator.model}'. Messages: {messages!r}")
@@ -307,6 +309,13 @@ async def _generate(
             if isinstance(generated, BaseException):
                 raise generated  # noqa: TRY301
 
+            with task_and_run(
+                name="Chat completion",
+                tags=["completion"],
+                label="Chat completion",
+            ):
+                log_output("messages", [messages[-1], generated.message])
+
             chat = rg.Chat(
                 messages,
                 [generated.message],