librechat-data-provider 0.1.3 → 0.1.5

package/dist/index.es.js

@@ -1,4 +1,5 @@
 import axios from 'axios';
+import { z } from 'zod';
 import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
 
 /******************************************************************************
@@ -336,19 +337,301 @@ var EModelEndpoint;
     EModelEndpoint["azureOpenAI"] = "azureOpenAI";
     EModelEndpoint["openAI"] = "openAI";
     EModelEndpoint["bingAI"] = "bingAI";
-    EModelEndpoint["chatGPT"] = "chatGPT";
     EModelEndpoint["chatGPTBrowser"] = "chatGPTBrowser";
     EModelEndpoint["google"] = "google";
     EModelEndpoint["gptPlugins"] = "gptPlugins";
     EModelEndpoint["anthropic"] = "anthropic";
 })(EModelEndpoint || (EModelEndpoint = {}));
-var Side;
-(function (Side) {
-    Side["Top"] = "top";
-    Side["Right"] = "right";
-    Side["Bottom"] = "bottom";
-    Side["Left"] = "left";
-})(Side || (Side = {}));
+var eModelEndpointSchema = z.nativeEnum(EModelEndpoint);
+var tMessageSchema = z.object({
+    messageId: z.string(),
+    clientId: z.string().nullable().optional(),
+    conversationId: z.string().nullable(),
+    parentMessageId: z.string().nullable(),
+    sender: z.string(),
+    text: z.string(),
+    isCreatedByUser: z.boolean(),
+    error: z.boolean(),
+    createdAt: z
+        .string()
+        .optional()
+        .default(function () { return new Date().toISOString(); }),
+    updatedAt: z
+        .string()
+        .optional()
+        .default(function () { return new Date().toISOString(); }),
+    current: z.boolean().optional(),
+    unfinished: z.boolean().optional(),
+    submitting: z.boolean().optional(),
+    searchResult: z.boolean().optional(),
+    finish_reason: z.string().optional(),
+});
+var tPluginAuthConfigSchema = z.object({
+    authField: z.string(),
+    label: z.string(),
+    description: z.string(),
+});
+var tPluginSchema = z.object({
+    name: z.string(),
+    pluginKey: z.string(),
+    description: z.string(),
+    icon: z.string(),
+    authConfig: z.array(tPluginAuthConfigSchema),
+    authenticated: z.boolean().optional(),
+    isButton: z.boolean().optional(),
+});
+var tExampleSchema = z.object({
+    input: z.object({
+        content: z.string(),
+    }),
+    output: z.object({
+        content: z.string(),
+    }),
+});
+var tAgentOptionsSchema = z.object({
+    agent: z.string(),
+    skipCompletion: z.boolean(),
+    model: z.string(),
+    temperature: z.number(),
+});
+var tConversationSchema = z.object({
+    conversationId: z.string().nullable(),
+    title: z.string(),
+    user: z.string().optional(),
+    endpoint: eModelEndpointSchema.nullable(),
+    suggestions: z.array(z.string()).optional(),
+    messages: z.array(z.string()).optional(),
+    tools: z.array(tPluginSchema).optional(),
+    createdAt: z.string(),
+    updatedAt: z.string(),
+    systemMessage: z.string().nullable().optional(),
+    modelLabel: z.string().nullable().optional(),
+    examples: z.array(tExampleSchema).optional(),
+    chatGptLabel: z.string().nullable().optional(),
+    userLabel: z.string().optional(),
+    model: z.string().nullable().optional(),
+    promptPrefix: z.string().nullable().optional(),
+    temperature: z.number().optional(),
+    topP: z.number().optional(),
+    topK: z.number().optional(),
+    context: z.string().nullable().optional(),
+    top_p: z.number().optional(),
+    frequency_penalty: z.number().optional(),
+    presence_penalty: z.number().optional(),
+    jailbreak: z.boolean().optional(),
+    jailbreakConversationId: z.string().nullable().optional(),
+    conversationSignature: z.string().nullable().optional(),
+    parentMessageId: z.string().optional(),
+    clientId: z.string().nullable().optional(),
+    invocationId: z.number().nullable().optional(),
+    toneStyle: z.string().nullable().optional(),
+    maxOutputTokens: z.number().optional(),
+    agentOptions: tAgentOptionsSchema.nullable().optional(),
+});
+var tPresetSchema = tConversationSchema
+    .omit({
+    conversationId: true,
+    createdAt: true,
+    updatedAt: true,
+    title: true,
+})
+    .merge(z.object({
+    conversationId: z.string().optional(),
+    presetId: z.string().nullable().optional(),
+    title: z.string().nullable().optional(),
+}));
+var openAISchema = tConversationSchema
+    .pick({
+    model: true,
+    chatGptLabel: true,
+    promptPrefix: true,
+    temperature: true,
+    top_p: true,
+    presence_penalty: true,
+    frequency_penalty: true,
+})
+    .transform(function (obj) {
+    var _a, _b, _c, _d, _e, _f, _g;
+    return (__assign(__assign({}, obj), { model: (_a = obj.model) !== null && _a !== void 0 ? _a : 'gpt-3.5-turbo', chatGptLabel: (_b = obj.chatGptLabel) !== null && _b !== void 0 ? _b : null, promptPrefix: (_c = obj.promptPrefix) !== null && _c !== void 0 ? _c : null, temperature: (_d = obj.temperature) !== null && _d !== void 0 ? _d : 1, top_p: (_e = obj.top_p) !== null && _e !== void 0 ? _e : 1, presence_penalty: (_f = obj.presence_penalty) !== null && _f !== void 0 ? _f : 0, frequency_penalty: (_g = obj.frequency_penalty) !== null && _g !== void 0 ? _g : 0 }));
+})
+    .catch(function () { return ({
+    model: 'gpt-3.5-turbo',
+    chatGptLabel: null,
+    promptPrefix: null,
+    temperature: 1,
+    top_p: 1,
+    presence_penalty: 0,
+    frequency_penalty: 0,
+}); });
+var googleSchema = tConversationSchema
+    .pick({
+    model: true,
+    modelLabel: true,
+    promptPrefix: true,
+    examples: true,
+    temperature: true,
+    maxOutputTokens: true,
+    topP: true,
+    topK: true,
+})
+    .transform(function (obj) {
+    var _a, _b, _c, _d, _e, _f, _g;
+    return (__assign(__assign({}, obj), { model: (_a = obj.model) !== null && _a !== void 0 ? _a : 'chat-bison', modelLabel: (_b = obj.modelLabel) !== null && _b !== void 0 ? _b : null, promptPrefix: (_c = obj.promptPrefix) !== null && _c !== void 0 ? _c : null, temperature: (_d = obj.temperature) !== null && _d !== void 0 ? _d : 0.2, maxOutputTokens: (_e = obj.maxOutputTokens) !== null && _e !== void 0 ? _e : 1024, topP: (_f = obj.topP) !== null && _f !== void 0 ? _f : 0.95, topK: (_g = obj.topK) !== null && _g !== void 0 ? _g : 40 }));
+})
+    .catch(function () { return ({
+    model: 'chat-bison',
+    modelLabel: null,
+    promptPrefix: null,
+    temperature: 0.2,
+    maxOutputTokens: 1024,
+    topP: 0.95,
+    topK: 40,
+}); });
+var bingAISchema = tConversationSchema
+    .pick({
+    jailbreak: true,
+    systemMessage: true,
+    context: true,
+    toneStyle: true,
+    jailbreakConversationId: true,
+    conversationSignature: true,
+    clientId: true,
+    invocationId: true,
+})
+    .transform(function (obj) {
+    var _a, _b, _c, _d, _e, _f, _g, _h;
+    return (__assign(__assign({}, obj), { model: '', jailbreak: (_a = obj.jailbreak) !== null && _a !== void 0 ? _a : false, systemMessage: (_b = obj.systemMessage) !== null && _b !== void 0 ? _b : null, context: (_c = obj.context) !== null && _c !== void 0 ? _c : null, toneStyle: (_d = obj.toneStyle) !== null && _d !== void 0 ? _d : 'creative', jailbreakConversationId: (_e = obj.jailbreakConversationId) !== null && _e !== void 0 ? _e : null, conversationSignature: (_f = obj.conversationSignature) !== null && _f !== void 0 ? _f : null, clientId: (_g = obj.clientId) !== null && _g !== void 0 ? _g : null, invocationId: (_h = obj.invocationId) !== null && _h !== void 0 ? _h : 1 }));
+})
+    .catch(function () { return ({
+    model: '',
+    jailbreak: false,
+    systemMessage: null,
+    context: null,
+    toneStyle: 'creative',
+    jailbreakConversationId: null,
+    conversationSignature: null,
+    clientId: null,
+    invocationId: 1,
+}); });
+var anthropicSchema = tConversationSchema
+    .pick({
+    model: true,
+    modelLabel: true,
+    promptPrefix: true,
+    temperature: true,
+    maxOutputTokens: true,
+    topP: true,
+    topK: true,
+})
+    .transform(function (obj) {
+    var _a, _b, _c, _d, _e, _f, _g;
+    return (__assign(__assign({}, obj), { model: (_a = obj.model) !== null && _a !== void 0 ? _a : 'claude-1', modelLabel: (_b = obj.modelLabel) !== null && _b !== void 0 ? _b : null, promptPrefix: (_c = obj.promptPrefix) !== null && _c !== void 0 ? _c : null, temperature: (_d = obj.temperature) !== null && _d !== void 0 ? _d : 1, maxOutputTokens: (_e = obj.maxOutputTokens) !== null && _e !== void 0 ? _e : 1024, topP: (_f = obj.topP) !== null && _f !== void 0 ? _f : 0.7, topK: (_g = obj.topK) !== null && _g !== void 0 ? _g : 5 }));
+})
+    .catch(function () { return ({
+    model: 'claude-1',
+    modelLabel: null,
+    promptPrefix: null,
+    temperature: 1,
+    maxOutputTokens: 1024,
+    topP: 0.7,
+    topK: 5,
+}); });
+var chatGPTBrowserSchema = tConversationSchema
+    .pick({
+    model: true,
+})
+    .transform(function (obj) {
+    var _a;
+    return (__assign(__assign({}, obj), { model: (_a = obj.model) !== null && _a !== void 0 ? _a : 'text-davinci-002-render-sha' }));
+})
+    .catch(function () { return ({
+    model: 'text-davinci-002-render-sha',
+}); });
+var gptPluginsSchema = tConversationSchema
+    .pick({
+    model: true,
+    chatGptLabel: true,
+    promptPrefix: true,
+    temperature: true,
+    top_p: true,
+    presence_penalty: true,
+    frequency_penalty: true,
+    tools: true,
+    agentOptions: true,
+})
+    .transform(function (obj) {
+    var _a, _b, _c, _d, _e, _f, _g, _h, _j;
+    return (__assign(__assign({}, obj), { model: (_a = obj.model) !== null && _a !== void 0 ? _a : 'gpt-3.5-turbo', chatGptLabel: (_b = obj.chatGptLabel) !== null && _b !== void 0 ? _b : null, promptPrefix: (_c = obj.promptPrefix) !== null && _c !== void 0 ? _c : null, temperature: (_d = obj.temperature) !== null && _d !== void 0 ? _d : 0.8, top_p: (_e = obj.top_p) !== null && _e !== void 0 ? _e : 1, presence_penalty: (_f = obj.presence_penalty) !== null && _f !== void 0 ? _f : 0, frequency_penalty: (_g = obj.frequency_penalty) !== null && _g !== void 0 ? _g : 0, tools: (_h = obj.tools) !== null && _h !== void 0 ? _h : [], agentOptions: (_j = obj.agentOptions) !== null && _j !== void 0 ? _j : {
+            agent: 'functions',
+            skipCompletion: true,
+            model: 'gpt-3.5-turbo',
+            temperature: 0,
+        } }));
+})
+    .catch(function () { return ({
+    model: 'gpt-3.5-turbo',
+    chatGptLabel: null,
+    promptPrefix: null,
+    temperature: 0.8,
+    top_p: 1,
+    presence_penalty: 0,
+    frequency_penalty: 0,
+    tools: [],
+    agentOptions: {
+        agent: 'functions',
+        skipCompletion: true,
+        model: 'gpt-3.5-turbo',
+        temperature: 0,
+    },
+}); });
+var endpointSchemas = {
+    openAI: openAISchema,
+    azureOpenAI: openAISchema,
+    google: googleSchema,
+    bingAI: bingAISchema,
+    anthropic: anthropicSchema,
+    chatGPTBrowser: chatGPTBrowserSchema,
+    gptPlugins: gptPluginsSchema,
+};
+function getFirstDefinedValue(possibleValues) {
+    var returnValue;
+    for (var _i = 0, possibleValues_1 = possibleValues; _i < possibleValues_1.length; _i++) {
+        var value = possibleValues_1[_i];
+        if (value) {
+            returnValue = value;
+            break;
+        }
+    }
+    return returnValue;
+}
+var parseConvo = function (endpoint, conversation, possibleValues) {
+    var _a;
+    var schema = endpointSchemas[endpoint];
+    if (!schema) {
+        throw new Error("Unknown endpoint: ".concat(endpoint));
+    }
+    var convo = schema.parse(conversation);
+    if (possibleValues && convo) {
+        convo.model = (_a = getFirstDefinedValue(possibleValues.model)) !== null && _a !== void 0 ? _a : convo.model;
+    }
+    return convo;
+};
+var getResponseSender = function (endpointOption) {
+    var endpoint = endpointOption.endpoint, chatGptLabel = endpointOption.chatGptLabel, modelLabel = endpointOption.modelLabel, jailbreak = endpointOption.jailbreak;
+    if (['openAI', 'azureOpenAI', 'gptPlugins', 'chatGPTBrowser'].includes(endpoint)) {
+        return chatGptLabel !== null && chatGptLabel !== void 0 ? chatGptLabel : 'ChatGPT';
+    }
+    if (endpoint === 'bingAI') {
+        return jailbreak ? 'Sydney' : 'BingAI';
+    }
+    if (endpoint === 'anthropic') {
+        return modelLabel !== null && modelLabel !== void 0 ? modelLabel : 'Anthropic';
+    }
+    if (endpoint === 'google') {
+        return modelLabel !== null && modelLabel !== void 0 ? modelLabel : 'PaLM2';
+    }
+    return '';
+};
 
 var QueryKeys;
 (function (QueryKeys) {
@@ -384,7 +667,7 @@ var useGetConversationByIdQuery = function (id, config) {
 var useGetConversationByIdMutation = function (id) {
     var queryClient = useQueryClient();
     return useMutation(function () { return getConversationById(id); }, {
-        // onSuccess: (res: t.TConversation) => {
+        // onSuccess: (res: s.TConversation) => {
         onSuccess: function () {
             queryClient.invalidateQueries([QueryKeys.conversation, id]);
         },
@@ -759,8 +1042,8 @@ var SSE = function (url, options) {
 // }
 
 function createPayload(submission) {
-    var conversation = submission.conversation, message = submission.message, endpointOption = submission.endpointOption;
-    var conversationId = conversation.conversationId;
+    var conversation = submission.conversation, message = submission.message, endpointOption = submission.endpointOption, isEdited = submission.isEdited;
+    var conversationId = tConversationSchema.parse(conversation).conversationId;
     var endpoint = endpointOption.endpoint;
     var endpointUrlMap = {
         azureOpenAI: '/api/ask/azureOpenAI',
@@ -773,8 +1056,11 @@ function createPayload(submission) {
         anthropic: '/api/ask/anthropic',
     };
     var server = endpointUrlMap[endpoint];
+    if (isEdited) {
+        server = server.replace('/ask/', '/edit/');
+    }
     var payload = __assign(__assign(__assign({}, message), endpointOption), { conversationId: conversationId });
     return { server: server, payload: payload };
 }
 
-export { EModelEndpoint, QueryKeys, SSE, Side, abortRequestWithMessage, clearAllConversations, createPayload, createPreset, deleteConversation, deletePreset, getAIEndpoints, getAvailablePlugins, getConversationById, getConversations, getLoginGoogle, getMessagesByConvoId, getPresets, getSearchEnabled, getStartupConfig, getUser, login, logout, refreshToken, register, requestPasswordReset, resetPassword, searchConversations, setAcceptLanguageHeader, setTokenHeader, updateConversation, updatePreset, updateTokenCount, updateUserPlugins, useAbortRequestWithMessage, useAvailablePluginsQuery, useClearConversationsMutation, useCreatePresetMutation, useDeleteConversationMutation, useDeletePresetMutation, useGetConversationByIdMutation, useGetConversationByIdQuery, useGetConversationsQuery, useGetEndpointsQuery, useGetMessagesByConvoId, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useSearchQuery, useUpdateConversationMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserPluginsMutation };
+export { EModelEndpoint, QueryKeys, SSE, abortRequestWithMessage, anthropicSchema, bingAISchema, chatGPTBrowserSchema, clearAllConversations, createPayload, createPreset, deleteConversation, deletePreset, eModelEndpointSchema, getAIEndpoints, getAvailablePlugins, getConversationById, getConversations, getLoginGoogle, getMessagesByConvoId, getPresets, getResponseSender, getSearchEnabled, getStartupConfig, getUser, googleSchema, gptPluginsSchema, login, logout, openAISchema, parseConvo, refreshToken, register, requestPasswordReset, resetPassword, searchConversations, setAcceptLanguageHeader, setTokenHeader, tAgentOptionsSchema, tConversationSchema, tExampleSchema, tMessageSchema, tPluginAuthConfigSchema, tPluginSchema, tPresetSchema, updateConversation, updatePreset, updateTokenCount, updateUserPlugins, useAbortRequestWithMessage, useAvailablePluginsQuery, useClearConversationsMutation, useCreatePresetMutation, useDeleteConversationMutation, useDeletePresetMutation, useGetConversationByIdMutation, useGetConversationByIdQuery, useGetConversationsQuery, useGetEndpointsQuery, useGetMessagesByConvoId, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useSearchQuery, useUpdateConversationMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserPluginsMutation };