llm logs --json for attachments, refs #587

simonw · Oct 28, 2024 · a9bc1c7 · a9bc1c7
1 parent 2384fd5
commit a9bc1c7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/llm/cli.py b/llm/cli.py
@@ -854,6 +854,11 @@ def logs_list(
 
     if json_output:
         # Output as JSON if requested
+        for row in rows:
+            row["attachments"] = [
+                {k: v for k, v in attachment.items() if k != "response_id"}
+                for attachment in attachments_by_id.get(row["id"], [])
+            ]
         click.echo(json.dumps(list(rows), indent=2))
     elif response:
         # Just output the last response