Spaces:
Runtime error
Runtime error
import pytest | |
from swarms.worker.omni_worker import OmniWorkerAgent | |
def omni_worker(): | |
api_key = 'test-key' | |
api_endpoint = 'test-endpoint' | |
api_type = 'test-type' | |
return OmniWorkerAgent(api_key, api_endpoint, api_type) | |
def test_chat_valid_data(mocker, omni_worker, data, expected_response): | |
mocker.patch('yourmodule.chat_huggingface', return_value=expected_response) # replace 'yourmodule' with actual module name | |
assert omni_worker.chat(data) == expected_response | |
def test_chat_invalid_data(omni_worker, invalid_data): | |
with pytest.raises(ValueError): | |
omni_worker.chat(invalid_data) | |