chore(hogql): Add retention queries in HogQL

frontend/src/queries/schema.json

@@ -2518,7 +2518,7 @@
                     "type": "object"
                 },
                 "total_intervals": {
-                    "type": "number"
+                    "type": "integer"
                 }
             },
             "type": "object"
@@ -2560,6 +2560,9 @@
                     ],
                     "description": "Property filters for all series"
                 },
+                "response": {
+                    "$ref": "#/definitions/RetentionQueryResponse"
+                },
                 "retentionFilter": {
                     "$ref": "#/definitions/RetentionFilter",
                     "description": "Properties specific to the retention insight"
@@ -2572,6 +2575,35 @@
             "required": ["kind"],
             "type": "object"
         },
+        "RetentionQueryResponse": {
+            "additionalProperties": false,
+            "properties": {
+                "hogql": {
+                    "type": "string"
+                },
+                "is_cached": {
+                    "type": "boolean"
+                },
+                "last_refresh": {
+                    "type": "string"
+                },
+                "next_allowed_client_refresh": {
+                    "type": "string"
+                },
+                "results": {
+                    "items": {},
+                    "type": "array"
+                },
+                "timings": {
+                    "items": {
+                        "$ref": "#/definitions/QueryTiming"
+                    },
+                    "type": "array"
+                }
+            },
+            "required": ["results"],
+            "type": "object"
+        },
         "RetentionType": {
             "enum": ["retention_recurring", "retention_first_time"],
             "type": "string"

frontend/src/queries/schema.ts

@@ -498,8 +498,13 @@ export interface FunnelsQuery extends InsightsQueryBase {
 
 /** `RetentionFilterType` minus everything inherited from `FilterType` */
 export type RetentionFilter = Omit<RetentionFilterType, keyof FilterType>
+
+export interface RetentionQueryResponse extends QueryResponse {
+    results: any[]
+}
 export interface RetentionQuery extends InsightsQueryBase {
     kind: NodeKind.RetentionQuery
+    response?: RetentionQueryResponse
     /** Properties specific to the retention insight */
     retentionFilter?: RetentionFilter
 }

frontend/src/queries/utils.ts

@@ -160,7 +160,7 @@ export function isLifecycleQuery(node?: Node | null): node is LifecycleQuery {
 }
 
 export function isQueryWithHogQLSupport(node?: Node | null): node is LifecycleQuery {
-    return isLifecycleQuery(node) || isTrendsQuery(node)
+    return isLifecycleQuery(node) || isTrendsQuery(node) || isRetentionQuery(node)
 }
 
 export function isInsightQueryWithDisplay(node?: Node | null): node is TrendsQuery | StickinessQuery {

frontend/src/types.ts

@@ -1825,6 +1825,7 @@ export interface PathsFilterType extends FilterType {
 export interface RetentionFilterType extends FilterType {
     retention_type?: RetentionType
     retention_reference?: 'total' | 'previous' // retention wrt cohort size or previous period
+    /** @asType integer */
     total_intervals?: number // retention total intervals
     returning_entity?: Record<string, any>
     target_entity?: Record<string, any>

posthog/api/services/query.py

@@ -18,6 +18,7 @@
 
 QUERY_WITH_RUNNER = [
     "LifecycleQuery",
+    "RetentionQuery",
     "TrendsQuery",
     "WebOverviewQuery",
     "WebTopSourcesQuery",