Update test/agentchat/test_groupchat.py

microsoft · Oct 9, 2024 · 7de92b5 · 7de92b5
1 parent 81f29d4
commit 7de92b5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/agentchat/test_groupchat.py b/test/agentchat/test_groupchat.py
@@ -2101,7 +2101,9 @@ def get_usage(response):
         select_speaker_auto_model_client_cls=CustomModelClient
     )
 
-    result = group_chat._create_internal_agents(
+checking_agent, speaker_selection_agent = group_chat._create_internal_agents(agents=[], messages=[], max_attempts=3, validate_speaker_name=(True, "test"))
+
+assert isinstance(speaker_selection_agent.client._clients[0], CustomModelClient)
         agents=[], messages=[], max_attempts=3, validate_speaker_name=(True, "test")
     )