apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-07-22 23:31:12 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat(model-connections): add llmSetupStatusAtomFamily and invalidate on mutations

Browse source
This commit is contained in:
Anish Sarkar 2026-07-12 20:29:59 +05:30
parent ad28bd912c
commit 5efd4291bf
2 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

6
surfsense_web/atoms/model-connections/model-connections-mutation.atoms.ts
View file

@ -25,6 +25,9 @@ function invalidateModelConnections(workspaceId: number) {
queryClient.invalidateQueries({ queryClient.invalidateQueries({
queryKey: cacheKeys.modelConnections.roles(workspaceId), queryKey: cacheKeys.modelConnections.roles(workspaceId),
}); });
queryClient.invalidateQueries({
queryKey: cacheKeys.modelConnections.setupStatus(workspaceId),
});
} }
function upsertModelConnection(workspaceId: number, connection: ConnectionRead) { function upsertModelConnection(workspaceId: number, connection: ConnectionRead) {
@ -208,6 +211,9 @@ export const updateModelRolesMutationAtom = atomWithMutation((get) => {
queryClient.invalidateQueries({ queryClient.invalidateQueries({
queryKey: cacheKeys.modelConnections.roles(workspaceId), queryKey: cacheKeys.modelConnections.roles(workspaceId),
}); });
queryClient.invalidateQueries({
queryKey: cacheKeys.modelConnections.setupStatus(workspaceId),
});
}, },
onError: (error: Error) => toast.error(error.message || "Failed to update model roles"), onError: (error: Error) => toast.error(error.message || "Failed to update model roles"),
}; };

13
surfsense_web/atoms/model-connections/model-connections-query.atoms.ts
View file

@ -1,4 +1,5 @@
import { atomWithQuery } from "jotai-tanstack-query"; import { atomWithQuery } from "jotai-tanstack-query";
import { atomFamily } from "jotai/utils";
import { modelConnectionsApiService } from "@/lib/apis/model-connections-api.service"; import { modelConnectionsApiService } from "@/lib/apis/model-connections-api.service";
import { isAuthenticated } from "@/lib/auth-utils"; import { isAuthenticated } from "@/lib/auth-utils";
import { cacheKeys } from "@/lib/query-client/cache-keys"; import { cacheKeys } from "@/lib/query-client/cache-keys";
@ -44,3 +45,15 @@ export const modelRolesAtom = atomWithQuery((get) => {
queryFn: () => modelConnectionsApiService.getModelRoles(workspaceId), queryFn: () => modelConnectionsApiService.getModelRoles(workspaceId),
}; };
}); });
// Keyed by the route workspaceId (not activeWorkspaceIdAtom) so the onboarding
// gate's verdict can never be computed against a stale workspace during a
// workspace switch.
export const llmSetupStatusAtomFamily = atomFamily((workspaceId: number) =>
atomWithQuery(() => ({
queryKey: cacheKeys.modelConnections.setupStatus(workspaceId),
enabled: workspaceId > 0 && isAuthenticated(),
staleTime: 5 * 60 * 1000,
queryFn: () => modelConnectionsApiService.getLlmSetupStatus(workspaceId),
}))
);