diff --git a/weave/trace_server/clickhouse_trace_server_batched.py b/weave/trace_server/clickhouse_trace_server_batched.py index d100fac3a6b3..8bbae5bc45b7 100644 --- a/weave/trace_server/clickhouse_trace_server_batched.py +++ b/weave/trace_server/clickhouse_trace_server_batched.py @@ -1865,9 +1865,6 @@ def _insert_call(self, ch_call: CallCHInsertable) -> None: def _get_matched_calls_for_filters( self, project_id: str, call_ids: list[str] ) -> list[tuple[ActionDispatchFilter, str, list[tsi.CallSchema]]]: - # TODO: this can be lifted to the top once core's deps are updated - import xxhash - """Helper function to get calls that match action filters. Returns a list of tuples containing (filter, matched_calls) pairs. @@ -1938,6 +1935,9 @@ def _get_matched_calls_for_filters( ) calls_res = self.calls_query(calls_query_req) + # TODO: this can be lifted to the top once core's deps are updated + import xxhash + # Match calls to filters matched_filters_and_calls = [] for filter, filter_ref in filters: