Fix test

VRSEN · Apr 26, 2024 · 0bca6f6 · 0bca6f6
1 parent 364c59d
commit 0bca6f6
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_agency.py b/tests/test_agency.py
@@ -357,12 +357,11 @@ def test_7_init_async_agency(self):
         # reset loaded thread ids
         self.__class__.loaded_thread_ids = {}
 
-        self.__class__.agent1.instructions = "Your task is to say 'success'"
+        self.__class__.agent1.instructions = "Your task is to say 'success' and nothing else."
 
         self.__class__.agency = Agency([
             self.__class__.ceo,
-            [self.__class__.ceo, self.__class__.agent1],
-            [self.__class__.agent1, self.__class__.agent2]],
+            [self.__class__.ceo, self.__class__.agent1]],
             shared_instructions="This is a shared instruction",
             settings_callbacks=self.__class__.settings_callbacks,
             threads_callbacks=self.__class__.threads_callbacks,