mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-08-11 10:07:50 +00:00
13 lines
429 B
Python
13 lines
429 B
Python
|
from unittest.mock import Mock
|
||
|
|
|
||
|
|
|
||
|
|
def create_mock_layer_that_supports(model_name, response=["fake_response"]):
|
||
|
|
"""
|
||
|
|
Create a mock invocation layer that supports the model_name and returns response.
|
||
|
|
"""
|
||
|
|
|
||
|
|
def mock_supports(model_name_or_path, **kwargs):
|
||
|
|
return model_name_or_path == model_name
|
||
|
|
|
||
|
|
return Mock(**{"model_name_or_path": model_name, "supports": mock_supports, "invoke.return_value": response})
|