mirror of
https://github.com/All-Hands-AI/OpenHands.git
synced 2026-01-10 07:18:10 -05:00
Integration test: exit if no prompt match (#1772)
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user