bug fix: fix agent_logging test (microsoft#2036)

* fix test * Update test/agentchat/test_agent_logging.py --------- Co-authored-by: Chi Wang <[email protected]>
whiskyboy · Mar 17, 2024 · a433a94 · a433a94
1 parent 7d61410
commit a433a94
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/test/agentchat/test_agent_logging.py b/test/agentchat/test_agent_logging.py
@@ -110,7 +110,11 @@ def test_two_agents_logging(db_connection):
         assert first_request_role == "system"
 
         response = json.loads(row["response"])
-        assert "choices" in response and len(response["choices"]) > 0
+
+        if "response" in response:  # config failed or response was empty
+            assert response["response"] is None or "error_code" in response["response"]
+        else:
+            assert "choices" in response and len(response["choices"]) > 0
 
         assert row["cost"] > 0
         assert row["start_time"], "start timestamp is empty"