From 8e0558a1d1196fd9a700690eddadd689c2c2c1f3 Mon Sep 17 00:00:00 2001
From: Pawel Szczur <pawel@posthog.com>
Date: Wed, 11 Dec 2024 19:43:41 +0000
Subject: [PATCH] Makes log_comment,is_initial_query and type not visible in UI

---
 posthog/hogql/database/schema/query_log.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/posthog/hogql/database/schema/query_log.py b/posthog/hogql/database/schema/query_log.py
index 3bd2372897b97..873ebfa50a239 100644
--- a/posthog/hogql/database/schema/query_log.py
+++ b/posthog/hogql/database/schema/query_log.py
@@ -14,7 +14,7 @@
 )
 
 QUERY_LOG_FIELDS: dict[str, FieldOrTable] = {
-    "query_id": StringDatabaseField(name="query_id"),  #
+    "query_id": StringDatabaseField(name="query_id"),
     "query": StringDatabaseField(name="query"),  #
     "query_start_time": DateTimeDatabaseField(name="event_time"),  #
     "query_duration_ms": FloatDatabaseField(name="query_duration_ms"),  #
@@ -25,18 +25,19 @@
     "result_bytes": IntegerDatabaseField(name="result_bytes"),
     "memory_usage": IntegerDatabaseField(name="memory_usage"),
     "status": StringDatabaseField(name="type"),
-    "log_comment": StringDatabaseField(name="log_comment"),
     "kind": StringDatabaseField(name="kind"),
     "query_type": StringDatabaseField(name="query_type"),
     "is_personal_api_key_request": BooleanDatabaseField(name="is_personal_api_key_request"),
-    # below fields are used in condition and cannot be removed
+}
+
+RAW_QUERY_LOG_FIELDS: dict[str, FieldOrTable] = QUERY_LOG_FIELDS | {
+    # below fields are necessary to compute some of the resulting fields
     "type": StringDatabaseField(name="type"),
     "is_initial_query": BooleanDatabaseField(name="is_initial_query"),
-    # "query_1": ExpressionField(name="query_1", ),
+    "log_comment": StringDatabaseField(name="log_comment"),
 }
 
 STRING_FIELDS = {
-    "cache_key": ["cache_key"],
     "query_type": ["query_type"],
     "query_id": ["client_query_id"],
     "query": ["query", "query"],
@@ -119,7 +120,7 @@ def get_alias(name, chain):
 
 
 class RawQueryLogTable(FunctionCallTable):
-    fields: dict[str, FieldOrTable] = QUERY_LOG_FIELDS
+    fields: dict[str, FieldOrTable] = RAW_QUERY_LOG_FIELDS
 
     name: str = "raw_query_log"