Fix update_cached_state typing

posthog/caching/calculate_results.py

@@ -146,12 +146,16 @@ def calculate_for_query_based_insight(
     if isinstance(response, BaseModel):
         response = response.model_dump()
 
-    update_cached_state(  # Updating the relevant InsightCachingState
-        insight.team_id,
-        response.get("cache_key"),
-        response.get("last_refresh"),
-        result=None,  # Not caching the result here, since in HogQL this is the query runner's responsibility
-    )
+    cache_key = response.get("cache_key")
+    last_refresh = response.get("last_refresh")
+    if isinstance(cache_key, str) and isinstance(last_refresh, str):
+        update_cached_state(  # Updating the relevant InsightCachingState
+            insight.team_id,
+            cache_key,
+            last_refresh,
+            result=None,  # Not caching the result here, since in HogQL this is the query runner's responsibility
+        )
+
     return InsightResult(
         # Translating `QueryResponse` to legacy insights shape
         # The response may not be conformant with that, hence these are all `.get()`s

posthog/caching/insight_cache.py

@@ -171,7 +171,7 @@ def update_cache(caching_state_id: UUID):
 def update_cached_state(
     team_id: int,
     cache_key: str,
-    timestamp: datetime,
+    timestamp: datetime | str,
     result: Any,
     ttl: Optional[int] = None,
 ):