diff --git a/scripts/generate_ai_samples.py b/scripts/generate_ai_samples.py
index 679587e..a691be1 100644
--- a/scripts/generate_ai_samples.py
+++ b/scripts/generate_ai_samples.py
@@ -388,6 +388,9 @@ def main() -> int:
                             "source_group": source_group,
                             "generation_type": "pure_ai",
                             "provenance": "ai_generated_controlled",
+                            "temperature": args.temperature,
+                            "shard_index": args.shard_index,
+                            "shard_count": args.shard_count,
                             "meta": meta,
                         }, ensure_ascii=False) + "\n")
                         sink.flush()