mirror of
https://github.com/deepset-ai/haystack.git
synced 2025-07-15 21:10:46 +00:00
a7f36fdd32
* Enhance Pipeline.draw() to show image directly in Jupyter notebook * Add util method to check if we're in a Jupyter notebook * Split Pipeline.draw() in two methods * Update tests * Update releasenotes
31 lines
814 B
Python
31 lines
814 B
Python
# SPDX-FileCopyrightText: 2022-present deepset GmbH <info@deepset.ai>
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
import logging
|
|
from pathlib import Path
|
|
|
|
from haystack.core.component import component
|
|
from haystack.core.pipeline import Pipeline
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
|
|
|
|
@component
|
|
class WithDefault:
|
|
@component.output_types(b=int)
|
|
def run(self, a: int, b: int = 2):
|
|
return {"c": a + b}
|
|
|
|
|
|
def test_pipeline():
|
|
pipeline = Pipeline()
|
|
pipeline.add_component("with_defaults", WithDefault())
|
|
|
|
# Pass all the inputs
|
|
results = pipeline.run({"with_defaults": {"a": 40, "b": 30}})
|
|
assert results == {"with_defaults": {"c": 70}}
|
|
|
|
# Rely on default value for 'b'
|
|
results = pipeline.run({"with_defaults": {"a": 40}})
|
|
assert results == {"with_defaults": {"c": 42}}
|