Skip to content

Commit ea233b3

Browse files
committed
feat: respect HAYSTACK_PROGRESS_BARS in LLMEvaluator
1 parent 6e549a6 commit ea233b3

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

haystack/components/evaluators/llm_evaluator.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from haystack.components.generators.chat.types import ChatGenerator
1414
from haystack.core.serialization import component_to_dict
1515
from haystack.dataclasses.chat_message import ChatMessage
16-
from haystack.utils import deserialize_chatgenerator_inplace, deserialize_type, serialize_type
16+
from haystack.utils import deserialize_chatgenerator_inplace, deserialize_type, get_progress_bar_setting, serialize_type
1717

1818
logger = logging.getLogger(__name__)
1919

@@ -97,7 +97,8 @@ def __init__( # pylint: disable=too-many-positional-arguments
9797
self.inputs = inputs
9898
self.outputs = outputs
9999
self.examples = examples
100-
self.progress_bar = progress_bar
100+
self._progress_bar_param = progress_bar
101+
self.progress_bar = get_progress_bar_setting(progress_bar)
101102

102103
template = self.prepare_template()
103104
self.builder = PromptBuilder(template=template)
@@ -298,7 +299,7 @@ def to_dict(self) -> dict[str, Any]:
298299
outputs=self.outputs,
299300
examples=self.examples,
300301
chat_generator=component_to_dict(obj=self._chat_generator, name="chat_generator"),
301-
progress_bar=self.progress_bar,
302+
progress_bar=self._progress_bar_param,
302303
)
303304

304305
@classmethod

0 commit comments

Comments
 (0)