fix logging cost flaky test (#2277)

2024-04-04 10:36:46 -04:00 · 2024-04-04 10:36:46 -04:00 · e83301ed18
parent 513b0b7884
commit e83301ed18
1 changed files with 1 additions and 1 deletions
--- 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"