fix: Status is required

rungalileo · Jul 29, 2024 · b3c7a81 · b3c7a81
1 parent 86c186d
commit b3c7a81
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/test_langchain.py b/tests/test_langchain.py
@@ -28,8 +28,8 @@ def _call(
 @mark.parametrize(
     ["output", "ignore_trigger", "expected_return", "expected_call_count"],
     [
-        [dumps({"text": "foo", **A_TRACE_METADATA_DICT}), False, "foo", 1],
-        [dumps({"text": "foo", **A_TRACE_METADATA_DICT}), True, "foo", 1],
+        [dumps({"text": "foo", "status": "NOT_TRIGGERED", **A_TRACE_METADATA_DICT}), False, "foo", 1],
+        [dumps({"text": "foo", "status": "NOT_TRIGGERED", **A_TRACE_METADATA_DICT}), True, "foo", 1],
         [dumps({"text": "timeout", "status": "TIMEOUT", **A_TRACE_METADATA_DICT}), False, "timeout", 1],
         [dumps({"text": "timeout", "status": "TIMEOUT", **A_TRACE_METADATA_DICT}), True, "timeout", 1],
         [dumps({"text": "success", "status": "SUCCESS", **A_TRACE_METADATA_DICT}), False, "success", 1],
@@ -49,6 +49,6 @@ def test_parser(output: str, ignore_trigger: bool, expected_return: str, expecte
 @mark.parametrize(["echo_output", "expected_output"], [[True, "> Raw response: foo\n"], [False, ""]])
 def test_echo(echo_output: bool, expected_output: str, capsys: CaptureFixture) -> None:
     parser = ProtectParser(chain=ProtectLLM(), echo_output=echo_output)
-    parser.parser(dumps({"text": "foo", **A_TRACE_METADATA_DICT}))
+    parser.parser(dumps({"text": "foo", "status": "NOT_TRIGGERED", **A_TRACE_METADATA_DICT}))
     captured = capsys.readouterr()
     assert captured.out == expected_output