timing

AbanteAI · Aug 26, 2024 · bbf95b5 · bbf95b5
1 parent 4eb8d19
commit bbf95b5
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/scripts/timing.py b/scripts/timing.py
@@ -14,10 +14,11 @@
 async def speed_compare():
     client = Spice()
 
-    messages = [
-        {"role": "system", "content": "You are a helpful assistant."},
-        {"role": "user", "content": "list 100 random words"},
-    ]
+    messages = (
+        client.new_messages()
+        .add_system_message("You are a helpful assistant.")
+        .add_user_message("list 100 random words")
+    )
 
     models = [GPT_4o_2024_08_06, SONNET_3_5]
     runs = 3