mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-18 06:22:31 +00:00
d4f6531c52
* First rough implementation of refactored run * Further improve run logic * Properly handle variadic input in run * Further work * Enhance names and add more documentation * Fix issue with output distribution * This works * Enhance run comments * Mark Multiplexer as greedy * Remove MergeLoop in favour of Multiplexer in tests * Remove FirstIntSelector in favour of Multiplexer * Handle corner when waiting for input is stuck * Remove unused import * Handle mutable input data in run and misbehaving components * Handle run input validation * Test validation * Fix pylint * Fix mypy * Call warm_up in run to fix tests
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
import logging
|
|
|
|
from haystack.core.pipeline import Pipeline
|
|
from haystack.testing.sample_components import Hello, StringJoiner, StringListJoiner, TextSplitter
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
|
|
def test_joiner():
|
|
pipeline = Pipeline()
|
|
pipeline.add_component("hello_one", Hello())
|
|
pipeline.add_component("hello_two", Hello())
|
|
pipeline.add_component("hello_three", Hello())
|
|
pipeline.add_component("joiner", StringJoiner())
|
|
|
|
pipeline.connect("hello_one", "hello_two")
|
|
pipeline.connect("hello_two", "joiner")
|
|
pipeline.connect("hello_three", "joiner")
|
|
|
|
results = pipeline.run({"hello_one": {"word": "world"}, "hello_three": {"word": "my friend"}})
|
|
assert results == {"joiner": {"output": "Hello, my friend! Hello, Hello, world!!"}}
|
|
|
|
|
|
def test_joiner_with_lists():
|
|
pipeline = Pipeline()
|
|
pipeline.add_component("first", TextSplitter())
|
|
pipeline.add_component("second", TextSplitter())
|
|
pipeline.add_component("joiner", StringListJoiner())
|
|
|
|
pipeline.connect("first", "joiner")
|
|
pipeline.connect("second", "joiner")
|
|
|
|
results = pipeline.run({"first": {"sentence": "Hello world!"}, "second": {"sentence": "How are you?"}})
|
|
assert results == {"joiner": {"output": ["Hello", "world!", "How", "are", "you?"]}}
|
|
|
|
|
|
def test_joiner_with_pipeline_run():
|
|
pipeline = Pipeline()
|
|
pipeline.add_component("hello", Hello())
|
|
pipeline.add_component("joiner", StringJoiner())
|
|
pipeline.connect("hello", "joiner")
|
|
|
|
results = pipeline.run({"hello": {"word": "world"}, "joiner": {"input_str": "another string!"}})
|
|
assert results == {"joiner": {"output": "another string! Hello, world!"}}
|