update guestbook

PrefectHQ · Apr 17, 2024 · 01685d6 · 01685d6
1 parent c0579b1
commit 01685d6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/flows/test_sign_guestbook.py b/tests/flows/test_sign_guestbook.py
@@ -1,5 +1,4 @@
-from control_flow import Agent, run_ai
-from control_flow.core.flow import ai_flow
+from control_flow import Agent, ai_flow, run_ai
 
 # define assistants
 
@@ -34,7 +33,7 @@ def guestbook_flow():
         sign their names for the task to be complete. You can read the sign to
         see if that has happened yet. You can not sign for another assistant.
         """,
-        assistants=[a, b, c],
+        agents=[a, b, c],
         tools=[sign, view_guestbook],
     )