File size: 1,932 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 | import { EModelEndpoint, isAssistantsEndpoint } from 'librechat-data-provider';
type TUseGenerations = {
error?: boolean;
endpoint?: string;
messageId?: string;
isEditing?: boolean;
isSubmitting: boolean;
searchResult?: boolean;
finish_reason?: string;
latestMessageId?: string;
isCreatedByUser?: boolean;
};
export default function useGenerationsByLatest({
error = false,
endpoint,
messageId,
isEditing = false,
isSubmitting,
searchResult = false,
finish_reason = '',
latestMessageId,
isCreatedByUser = false,
}: TUseGenerations) {
const isEditableEndpoint = Boolean(
[
EModelEndpoint.openAI,
EModelEndpoint.custom,
EModelEndpoint.google,
EModelEndpoint.agents,
EModelEndpoint.bedrock,
EModelEndpoint.anthropic,
EModelEndpoint.gptPlugins,
EModelEndpoint.azureOpenAI,
].find((e) => e === endpoint),
);
const continueSupported =
latestMessageId === messageId &&
finish_reason &&
finish_reason !== 'stop' &&
!isEditing &&
!searchResult &&
isEditableEndpoint;
const branchingSupported = Boolean(
[
EModelEndpoint.azureOpenAI,
EModelEndpoint.openAI,
EModelEndpoint.custom,
EModelEndpoint.agents,
EModelEndpoint.bedrock,
EModelEndpoint.chatGPTBrowser,
EModelEndpoint.google,
EModelEndpoint.gptPlugins,
EModelEndpoint.anthropic,
].find((e) => e === endpoint),
);
const regenerateEnabled =
!isCreatedByUser && !searchResult && !isEditing && !isSubmitting && branchingSupported;
const hideEditButton =
isSubmitting ||
error ||
searchResult ||
!branchingSupported ||
(!isEditableEndpoint && !isCreatedByUser);
const forkingSupported = !isAssistantsEndpoint(endpoint) && !searchResult;
return {
forkingSupported,
continueSupported,
regenerateEnabled,
isEditableEndpoint,
hideEditButton,
};
}
|