From 667c818ef840699eac5055b459501672c20d08f7 Mon Sep 17 00:00:00 2001 From: LJ Date: Wed, 26 Mar 2025 08:59:03 -0700 Subject: [PATCH] Minor improvement: also allow `collect()` taking constant. --- python/cocoindex/flow.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/python/cocoindex/flow.py b/python/cocoindex/flow.py index 2ace32df..b5925a3e 100644 --- a/python/cocoindex/flow.py +++ b/python/cocoindex/flow.py @@ -254,7 +254,7 @@ def __init__(self, flow_builder_state: _FlowBuilderState, self._flow_builder_state = flow_builder_state self._engine_data_collector = data_collector - def collect(self, **kwargs: DataSlice | GeneratedField): + def collect(self, **kwargs): """ Collect data into the collector. """ @@ -269,7 +269,8 @@ def collect(self, **kwargs: DataSlice | GeneratedField): else: raise ValueError(f"Unexpected generated field: {v}") else: - regular_kwargs.append((k, _data_slice_state(v).engine_data_slice)) + regular_kwargs.append( + (k, self._flow_builder_state.get_data_slice(v))) self._flow_builder_state.engine_flow_builder.collect( self._engine_data_collector, regular_kwargs, auto_uuid_field)