SurfSense/surfsense_web/lib/query-client/cache-keys.ts

import type { GetDocumentsRequest } from "@/contracts/types/document.types";
import type { GetLLMConfigsRequest } from "@/contracts/types/llm-config.types";
import type { GetSearchSpacesRequest } from "@/contracts/types/search-space.types";

export const cacheKeys = {
	// New chat threads (assistant-ui)
	threads: {
		list: (searchSpaceId: number) => ["threads", searchSpaceId] as const,
		detail: (threadId: number) => ["threads", "detail", threadId] as const,
		search: (searchSpaceId: number, query: string) =>
			["threads", "search", searchSpaceId, query] as const,
	},
	documents: {
		globalQueryParams: (queries: GetDocumentsRequest["queryParams"]) =>
			["documents", ...(queries ? Object.values(queries) : [])] as const,
		withQueryParams: (queries: GetDocumentsRequest["queryParams"]) =>
			["documents-with-queries", ...(queries ? Object.values(queries) : [])] as const,
		document: (documentId: string) => ["document", documentId] as const,
		typeCounts: (searchSpaceId?: string) => ["documents", "type-counts", searchSpaceId] as const,
		byChunk: (chunkId: string) => ["documents", "by-chunk", chunkId] as const,
	},
	llmConfigs: {
		global: () => ["llm-configs", "global"] as const,
		all: (searchSpaceId: string) => ["llm-configs", searchSpaceId] as const,
		withQueryParams: (queries: GetLLMConfigsRequest["queryParams"]) =>
			["llm-configs", ...(queries ? Object.values(queries) : [])] as const,
		byId: (llmConfigId: string) => ["llm-config", llmConfigId] as const,
		preferences: (searchSpaceId: string) => ["llm-preferences", searchSpaceId] as const,
	},
	auth: {
		user: ["auth", "user"] as const,
	},
	searchSpaces: {
		all: ["search-spaces"] as const,
		withQueryParams: (queries: GetSearchSpacesRequest["queryParams"]) =>
			["search-spaces", ...(queries ? Object.values(queries) : [])] as const,
		detail: (searchSpaceId: string) => ["search-spaces", searchSpaceId] as const,
		communityPrompts: ["search-spaces", "community-prompts"] as const,
	},
	user: {
		current: () => ["user", "me"] as const,
	},
	roles: {
		all: (searchSpaceId: string) => ["roles", searchSpaceId] as const,
		byId: (searchSpaceId: string, roleId: string) => ["roles", searchSpaceId, roleId] as const,
	},
	permissions: {
		all: () => ["permissions"] as const,
	},
	members: {
		all: (searchSpaceId: string) => ["members", searchSpaceId] as const,
		myAccess: (searchSpaceId: string) => ["members", "my-access", searchSpaceId] as const,
	},
	invites: {
		all: (searchSpaceId: string) => ["invites", searchSpaceId] as const,
		info: (inviteCode: string) => ["invites", "info", inviteCode] as const,
	},
};