diff --git a/surfsense_web/lib/query-client/cache-keys.ts b/surfsense_web/lib/query-client/cache-keys.ts
index 54f411ad1..8c1316aa4 100644
--- a/surfsense_web/lib/query-client/cache-keys.ts
+++ b/surfsense_web/lib/query-client/cache-keys.ts
@@ -72,4 +72,11 @@ export const cacheKeys = {
 				["connectors", "google-drive", connectorId, "folders", parentId] as const,
 		},
 	},
+	comments: {
+		byMessage: (messageId: number) => ["comments", "message", messageId] as const,
+	},
+	mentions: {
+		all: (searchSpaceId?: number) => ["mentions", searchSpaceId] as const,
+		unreadOnly: (searchSpaceId?: number) => ["mentions", "unread", searchSpaceId] as const,
+	},
 };