Integration test: exit if no prompt match (#1772)

This commit is contained in:
Boxuan Li
2024-05-13 20:03:09 -07:00
committed by GitHub
parent 2771328036
commit b84f25ab35

View File

@@ -1,6 +1,7 @@
import io
import os
import re
import sys
from functools import partial
import pytest
@@ -79,9 +80,10 @@ def mock_completion(*args, test_name, **kwargs):
for message in messages:
message_str += message['content']
mock_response = get_mock_response(test_name, message_str)
assert mock_response is not None, (
'Mock response for prompt is not found:\n\n' + message_str
)
if mock_response is None:
print('Mock response for prompt is not found:\n\n' + message_str)
print('Exiting...')
sys.exit(1)
response = completion(**kwargs, mock_response=mock_response)
return response