File size: 3,023 Bytes
f0743f4 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | import { InfiniteCollections, QueryKeys } from 'librechat-data-provider';
import type { InfiniteData, QueryClient } from '@tanstack/react-query';
import type {
PromptGroupListResponse,
PromptGroupListData,
TPromptGroup,
} from 'librechat-data-provider';
import {
addData,
deleteData,
updateData,
updateFields,
addToCacheList,
updateCacheList,
removeFromCacheList,
getRecordByProperty,
} from './collection';
export const addPromptGroup = (
data: InfiniteData<PromptGroupListResponse>,
newPromptGroup: TPromptGroup,
): PromptGroupListData => {
return addData<PromptGroupListResponse, TPromptGroup>(
data,
InfiniteCollections.PROMPT_GROUPS,
newPromptGroup,
(page) => page.promptGroups.findIndex((group) => group._id === newPromptGroup._id),
);
};
export const updatePromptGroup = (
data: InfiniteData<PromptGroupListResponse>,
updatedPromptGroup: TPromptGroup,
): PromptGroupListData => {
return updateData<PromptGroupListResponse, TPromptGroup>(
data,
InfiniteCollections.PROMPT_GROUPS,
updatedPromptGroup,
(page) => page.promptGroups.findIndex((group) => group._id === updatedPromptGroup._id),
);
};
export const deletePromptGroup = (
data: InfiniteData<PromptGroupListResponse>,
groupId: string,
): PromptGroupListData => {
return deleteData<PromptGroupListResponse, PromptGroupListData>(
data,
InfiniteCollections.PROMPT_GROUPS,
(page) => page.promptGroups.findIndex((group) => group._id === groupId),
);
};
export const updateGroupFields = (
data: InfiniteData<PromptGroupListResponse>,
updatedGroup: Partial<TPromptGroup>,
callback?: (group: TPromptGroup) => void,
): InfiniteData<PromptGroupListResponse> => {
return updateFields<PromptGroupListResponse, TPromptGroup>(
data,
updatedGroup,
InfiniteCollections.PROMPT_GROUPS,
'_id',
callback,
);
};
export const getSnippet = (promptText: string, length = 56) => {
return promptText.length > length ? `${promptText.slice(0, length - 3)}...` : promptText;
};
export const findPromptGroup = (
data: InfiniteData<PromptGroupListResponse>,
findProperty: (group: TPromptGroup) => boolean,
): TPromptGroup | undefined => {
return getRecordByProperty<PromptGroupListResponse, TPromptGroup>(
data,
InfiniteCollections.PROMPT_GROUPS,
findProperty,
);
};
export const addGroupToAll = (queryClient: QueryClient, newGroup: TPromptGroup) => {
addToCacheList<TPromptGroup>(queryClient, [QueryKeys.allPromptGroups], newGroup);
};
export const updateGroupInAll = (
queryClient: QueryClient,
updatedGroup: Partial<TPromptGroup> & { _id: string },
) => {
updateCacheList<TPromptGroup>({
queryClient,
queryKey: [QueryKeys.allPromptGroups],
searchProperty: '_id',
updateData: updatedGroup,
searchValue: updatedGroup._id,
});
};
export const removeGroupFromAll = (queryClient: QueryClient, groupId: string) => {
removeFromCacheList<TPromptGroup>(queryClient, [QueryKeys.allPromptGroups], '_id', groupId);
};
|