mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-04 22:02:16 +02:00
feat: add LLM config cache keys and create mutation atom
This commit is contained in:
parent
03279a2836
commit
2200d7b63c
3 changed files with 38 additions and 1 deletions
29
surfsense_web/atoms/llm-config/llm-config-mutation.atoms.ts
Normal file
29
surfsense_web/atoms/llm-config/llm-config-mutation.atoms.ts
Normal file
|
|
@ -0,0 +1,29 @@
|
||||||
|
import { atomWithMutation } from "jotai-tanstack-query";
|
||||||
|
import { toast } from "sonner";
|
||||||
|
import { activeSearchSpaceIdAtom } from "@/atoms/seach-spaces/seach-space-queries.atom";
|
||||||
|
import type { CreateLLMConfigRequest } from "@/contracts/types/llm-config.types";
|
||||||
|
import { llmConfigApiService } from "@/lib/apis/llm-config-api.service";
|
||||||
|
import { cacheKeys } from "@/lib/query-client/cache-keys";
|
||||||
|
import { queryClient } from "@/lib/query-client/client";
|
||||||
|
|
||||||
|
export const createLLMConfigMutationAtom = atomWithMutation((get) => {
|
||||||
|
const searchSpaceId = get(activeSearchSpaceIdAtom);
|
||||||
|
|
||||||
|
return {
|
||||||
|
mutationKey: cacheKeys.llmConfigs.all(searchSpaceId!),
|
||||||
|
enabled: !!searchSpaceId,
|
||||||
|
mutationFn: async (request: CreateLLMConfigRequest) => {
|
||||||
|
return llmConfigApiService.createLLMConfig(request);
|
||||||
|
},
|
||||||
|
|
||||||
|
onSuccess: () => {
|
||||||
|
toast.success("LLM configuration created successfully");
|
||||||
|
queryClient.invalidateQueries({
|
||||||
|
queryKey: cacheKeys.llmConfigs.all(searchSpaceId!),
|
||||||
|
});
|
||||||
|
queryClient.invalidateQueries({
|
||||||
|
queryKey: cacheKeys.llmConfigs.global(),
|
||||||
|
});
|
||||||
|
},
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
@ -5,7 +5,6 @@ import {
|
||||||
type DeleteLLMConfigRequest,
|
type DeleteLLMConfigRequest,
|
||||||
deleteLLMConfigRequest,
|
deleteLLMConfigRequest,
|
||||||
deleteLLMConfigResponse,
|
deleteLLMConfigResponse,
|
||||||
type GetGlobalLLMConfigsResponse,
|
|
||||||
type GetLLMConfigRequest,
|
type GetLLMConfigRequest,
|
||||||
type GetLLMConfigsRequest,
|
type GetLLMConfigsRequest,
|
||||||
type GetLLMPreferencesRequest,
|
type GetLLMPreferencesRequest,
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,7 @@
|
||||||
import type { GetChatsRequest } from "@/contracts/types/chat.types";
|
import type { GetChatsRequest } from "@/contracts/types/chat.types";
|
||||||
import type { GetDocumentsRequest } from "@/contracts/types/document.types";
|
import type { GetDocumentsRequest } from "@/contracts/types/document.types";
|
||||||
import type { GetPodcastsRequest } from "@/contracts/types/podcast.types";
|
import type { GetPodcastsRequest } from "@/contracts/types/podcast.types";
|
||||||
|
import type { GetLLMConfigsRequest } from "@/contracts/types/llm-config.types";
|
||||||
|
|
||||||
export const cacheKeys = {
|
export const cacheKeys = {
|
||||||
chats: {
|
chats: {
|
||||||
|
|
@ -21,6 +22,14 @@ export const cacheKeys = {
|
||||||
typeCounts: (searchSpaceId?: string) => ["documents", "type-counts", searchSpaceId] as const,
|
typeCounts: (searchSpaceId?: string) => ["documents", "type-counts", searchSpaceId] as const,
|
||||||
byChunk: (chunkId: string) => ["documents", "by-chunk", chunkId] 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: {
|
auth: {
|
||||||
user: ["auth", "user"] as const,
|
user: ["auth", "user"] as const,
|
||||||
},
|
},
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue