test: OpenAIChatGenerator - fix flaky test (#9075)

Co-authored-by: Amna Mubashar <amnahkhan.ak@gmail.com>
This commit is contained in:
Stefano Fiorucci 2025-03-20 15:29:07 +01:00 committed by GitHub
parent f3fc6894fd
commit 67ab3788ea
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -886,7 +886,7 @@ class TestOpenAIChatGenerator:
# check that the completion_start_time is set and valid ISO format
assert "completion_start_time" in message.meta
assert datetime.fromisoformat(message.meta["completion_start_time"]) < datetime.now()
assert datetime.fromisoformat(message.meta["completion_start_time"]) <= datetime.now()
assert isinstance(message.meta["usage"], dict)
assert message.meta["usage"]["prompt_tokens"] > 0