From 23ea930cab9ecf73a4e310f80e4f4e414e826a28 Mon Sep 17 00:00:00 2001 From: Raja Sekhar Rao Dheekonda <43563047+rdheekonda@users.noreply.github.com> Date: Tue, 3 Dec 2024 14:03:25 -0800 Subject: [PATCH] Fix Test Prompt Response Error Values (#591) Co-authored-by: rdheekonda --- tests/test_literals.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_literals.py b/tests/test_literals.py index f68b1b0ad..362b8fad0 100644 --- a/tests/test_literals.py +++ b/tests/test_literals.py @@ -16,7 +16,7 @@ def test_prompt_data_type(): def test_prompt_response_error(): assert get_origin(PromptResponseError) is Literal - expected_literals = {"blocked", "none", "processing", "unknown"} + expected_literals = {"blocked", "none", "processing", "unknown", "empty"} assert set(get_args(PromptResponseError)) == expected_literals