From c30ef524273559a23d31257d33a036d4a7e7c11e Mon Sep 17 00:00:00 2001 From: gtarpenning Date: Thu, 12 Dec 2024 13:09:01 -0800 Subject: [PATCH] add back importand val_dump step --- weave/trace_server/clickhouse_trace_server_batched.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/weave/trace_server/clickhouse_trace_server_batched.py b/weave/trace_server/clickhouse_trace_server_batched.py index 38896d9b104..bbe7a5ccabe 100644 --- a/weave/trace_server/clickhouse_trace_server_batched.py +++ b/weave/trace_server/clickhouse_trace_server_batched.py @@ -1869,6 +1869,11 @@ def _select_objs_query( for row in query_result: (object_id, digest, val_dump) = row object_values[(object_id, digest)] = val_dump + + # update the val_dump for each object + for obj in result: + obj.val_dump = object_values.get((obj.object_id, obj.digest), "{}") + return result def _run_migrations(self) -> None: