Forward active from PipelineWorker through to BaseWorker
PipelineWorker.__init__ was only forwarding `name` to BaseWorker, so the `active` flag (the other BaseWorker constructor arg) wasn't reachable from PipelineWorker callers. Add `active: bool = True` to the signature and pass it through.
This commit is contained in:
@@ -204,6 +204,7 @@ class PipelineWorker(BaseWorker):
|
|||||||
self,
|
self,
|
||||||
pipeline: BasePipeline,
|
pipeline: BasePipeline,
|
||||||
*,
|
*,
|
||||||
|
active: bool = True,
|
||||||
additional_span_attributes: dict | None = None,
|
additional_span_attributes: dict | None = None,
|
||||||
app_resources: Any = None,
|
app_resources: Any = None,
|
||||||
bridged: tuple[str, ...] | None = None,
|
bridged: tuple[str, ...] | None = None,
|
||||||
@@ -230,6 +231,8 @@ class PipelineWorker(BaseWorker):
|
|||||||
|
|
||||||
Args:
|
Args:
|
||||||
pipeline: The pipeline to execute.
|
pipeline: The pipeline to execute.
|
||||||
|
active: Whether the worker starts active. Forwarded to
|
||||||
|
:class:`BaseWorker`.
|
||||||
additional_span_attributes: Optional dictionary of attributes to propagate as
|
additional_span_attributes: Optional dictionary of attributes to propagate as
|
||||||
OpenTelemetry conversation span attributes.
|
OpenTelemetry conversation span attributes.
|
||||||
app_resources: Optional application-defined bag of anything your
|
app_resources: Optional application-defined bag of anything your
|
||||||
@@ -276,7 +279,7 @@ class PipelineWorker(BaseWorker):
|
|||||||
Use ``app_resources`` instead. ``tool_resources`` will be
|
Use ``app_resources`` instead. ``tool_resources`` will be
|
||||||
removed in a future version.
|
removed in a future version.
|
||||||
"""
|
"""
|
||||||
super().__init__(name=name)
|
super().__init__(name=name, active=active)
|
||||||
self._bridged = bridged
|
self._bridged = bridged
|
||||||
if tool_resources is not None:
|
if tool_resources is not None:
|
||||||
with warnings.catch_warnings():
|
with warnings.catch_warnings():
|
||||||
|
|||||||
Reference in New Issue
Block a user