Skip to content

Commit 87e1eb3

Browse files
author
Radovan Fuchs
committed
fix
1 parent ac12850 commit 87e1eb3

1 file changed

Lines changed: 3 additions & 9 deletions

File tree

tests/e2e/features/steps/llm_query_response.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -256,9 +256,7 @@ def check_fragments_in_responses_output_text(context: Context) -> None:
256256

257257
for fragment in context.table:
258258
expected = fragment["Fragments in LLM response"]
259-
assert _response_contains_fragment(
260-
output_text, expected
261-
), (
259+
assert _response_contains_fragment(output_text, expected), (
262260
f"Fragment {expected!r} not found in output_text (case-insensitive): "
263261
f"{output_text!r}"
264262
)
@@ -297,9 +295,7 @@ def check_fragments_in_response(context: Context) -> None:
297295

298296
for fragment in context.table:
299297
expected = fragment["Fragments in LLM response"]
300-
assert _response_contains_fragment(
301-
response, expected
302-
), (
298+
assert _response_contains_fragment(response, expected), (
303299
f"Fragment {expected!r} not found in LLM response (case-insensitive): "
304300
f"{response!r}"
305301
)
@@ -323,9 +319,7 @@ def check_streamed_fragments_in_response(context: Context) -> None:
323319

324320
for fragment in context.table:
325321
expected = fragment["Fragments in LLM response"]
326-
assert _response_contains_fragment(
327-
response, expected
328-
), (
322+
assert _response_contains_fragment(response, expected), (
329323
f"Fragment {expected!r} not found in streamed LLM response "
330324
f"(case-insensitive): {response!r}"
331325
)

0 commit comments

Comments
 (0)