bug fix: logging test may fail if some config fails (#1695)

This commit is contained in:
cheng-tan 2024-02-15 13:51:38 -05:00 committed by GitHub
parent a52f52a1b5
commit 5ca69da8e5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -88,7 +88,7 @@ def test_two_agents_logging(db_connection):
cur.execute(CHAT_COMPLETIONS_QUERY) cur.execute(CHAT_COMPLETIONS_QUERY)
rows = cur.fetchall() rows = cur.fetchall()
assert len(rows) == 3 assert len(rows) >= 3 # some config may fail
session_id = rows[0]["session_id"] session_id = rows[0]["session_id"]
for idx, row in enumerate(rows): for idx, row in enumerate(rows):