mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-09-22 14:43:07 +00:00
5497ca2a45
* Adapt GPTGenerator to string input/output * Finishing touches * punctuation upd * PR feedback * Small naming fixes * Update haystack/preview/components/generators/openai.py Co-authored-by: Stefano Fiorucci <44616784+anakin87@users.noreply.github.com> * Update class pydoc with a printed response --------- Co-authored-by: Stefano Fiorucci <44616784+anakin87@users.noreply.github.com> Co-authored-by: Daria Fokina <daria.fokina@deepset.ai>
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import pytest
|
|
|
|
from haystack.preview.components.generators.utils import default_streaming_callback
|
|
from haystack.preview.components.generators.utils import serialize_callback_handler, deserialize_callback_handler
|
|
|
|
|
|
# streaming callback needs to be on module level
|
|
def streaming_callback(chunk):
|
|
pass
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_callback_handler_serialization():
|
|
result = serialize_callback_handler(streaming_callback)
|
|
assert result == "test_utils.streaming_callback"
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_callback_handler_serialization_non_local():
|
|
result = serialize_callback_handler(default_streaming_callback)
|
|
assert result == "haystack.preview.components.generators.utils.default_streaming_callback"
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_callback_handler_deserialization():
|
|
result = serialize_callback_handler(streaming_callback)
|
|
fn = deserialize_callback_handler(result)
|
|
|
|
assert fn is streaming_callback
|
|
|
|
|
|
@pytest.mark.unit
|
|
def test_callback_handler_deserialization_non_local():
|
|
result = serialize_callback_handler(default_streaming_callback)
|
|
fn = deserialize_callback_handler(result)
|
|
|
|
assert fn is default_streaming_callback
|