From e83301ed187fa1695934069118e856c2331e7c19 Mon Sep 17 00:00:00 2001 From: cheng-tan Date: Thu, 4 Apr 2024 10:36:46 -0400 Subject: [PATCH] fix logging cost flaky test (#2277) --- test/agentchat/test_agent_logging.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/agentchat/test_agent_logging.py b/test/agentchat/test_agent_logging.py index 0a974ff8d..8de4712de 100644 --- a/test/agentchat/test_agent_logging.py +++ b/test/agentchat/test_agent_logging.py @@ -116,7 +116,7 @@ def test_two_agents_logging(db_connection): else: assert "choices" in response and len(response["choices"]) > 0 - assert row["cost"] > 0 + assert row["cost"] >= 0.0 assert row["start_time"], "start timestamp is empty" assert row["end_time"], "end timestamp is empty"