librechat-data-provider 0.1.6 → 0.1.8

package/dist/index.es.js

@@ -73,6 +73,70 @@ typeof SuppressedError === "function" ? SuppressedError : function (error, suppr
     return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
 };
 
+function setAcceptLanguageHeader(value) {
+    axios.defaults.headers.common['Accept-Language'] = value;
+}
+function setTokenHeader(token) {
+    axios.defaults.headers.common['Authorization'] = 'Bearer ' + token;
+}
+
+var isRefreshing = false;
+var failedQueue = [];
+var processQueue = function (error, token) {
+    if (token === void 0) { token = null; }
+    failedQueue.forEach(function (prom) {
+        if (error) {
+            prom.reject(error);
+        }
+        else {
+            prom.resolve(token);
+        }
+    });
+    failedQueue = [];
+};
+axios.interceptors.response.use(function (response) { return response; }, function (error) {
+    var originalRequest = error.config;
+    if (error.response.status === 401 && !originalRequest._retry) {
+        if (isRefreshing) {
+            return new Promise(function (resolve, reject) {
+                failedQueue.push({ resolve: resolve, reject: reject });
+            })
+                .then(function (token) {
+                originalRequest.headers['Authorization'] = 'Bearer ' + token;
+                return axios(originalRequest);
+            })
+                .catch(function (err) {
+                return Promise.reject(err);
+            });
+        }
+        originalRequest._retry = true;
+        isRefreshing = true;
+        return new Promise(function (resolve, reject) {
+            refreshToken()
+                .then(function (_a) {
+                var token = _a.token;
+                if (token) {
+                    originalRequest.headers['Authorization'] = 'Bearer ' + token;
+                    setTokenHeader(token);
+                    window.dispatchEvent(new CustomEvent('tokenUpdated', { detail: token }));
+                    processQueue(null, token);
+                    resolve(axios(originalRequest));
+                }
+                else {
+                    window.location.href = '/login';
+                }
+            })
+                .catch(function (err) {
+                processQueue(err, null);
+                reject(err);
+            })
+                .then(function () {
+                isRefreshing = false;
+            });
+        });
+    }
+    return Promise.reject(error);
+});
 function _get(url, options) {
     return __awaiter(this, void 0, void 0, function () {
         var response;
@@ -180,75 +244,38 @@ var request = {
     patch: _patch,
 };
 
-var user = function () {
-    return '/api/user';
-};
-var userPlugins = function () {
-    return '/api/user/plugins';
-};
+var user = function () { return '/api/user'; };
+var userPlugins = function () { return '/api/user/plugins'; };
 var messages = function (conversationId, messageId) {
     return "/api/messages/".concat(conversationId).concat(messageId ? "/".concat(messageId) : '');
 };
-var abortRequest = function (endpoint) {
-    return "/api/ask/".concat(endpoint, "/abort");
-};
-var conversations = function (pageNumber) {
-    return "/api/convos?pageNumber=".concat(pageNumber);
-};
-var conversationById = function (id) {
-    return "/api/convos/".concat(id);
-};
-var updateConversation$1 = function () {
-    return '/api/convos/update';
-};
-var deleteConversation$1 = function () {
-    return '/api/convos/clear';
-};
+var keysEndpoint = '/api/keys';
+var keys = function () { return keysEndpoint; };
+var userKeyQuery$1 = function (name) { return "".concat(keysEndpoint, "?name=").concat(name); };
+var revokeUserKey$1 = function (name) { return "".concat(keysEndpoint, "/").concat(name); };
+var revokeAllUserKeys$1 = function () { return "".concat(keysEndpoint, "?all=true"); };
+var abortRequest = function (endpoint) { return "/api/ask/".concat(endpoint, "/abort"); };
+var conversations = function (pageNumber) { return "/api/convos?pageNumber=".concat(pageNumber); };
+var conversationById = function (id) { return "/api/convos/".concat(id); };
+var updateConversation$1 = function () { return '/api/convos/update'; };
+var deleteConversation$1 = function () { return '/api/convos/clear'; };
 var search = function (q, pageNumber) {
     return "/api/search?q=".concat(q, "&pageNumber=").concat(pageNumber);
 };
-var searchEnabled = function () {
-    return '/api/search/enable';
-};
-var presets = function () {
-    return '/api/presets';
-};
-var deletePreset$1 = function () {
-    return '/api/presets/delete';
-};
-var aiEndpoints = function () {
-    return '/api/endpoints';
-};
-var tokenizer = function () {
-    return '/api/tokenizer';
-};
-var login$1 = function () {
-    return '/api/auth/login';
-};
-var logout$1 = function () {
-    return '/api/auth/logout';
-};
-var register$1 = function () {
-    return '/api/auth/register';
-};
-var loginGoogle = function () {
-    return '/api/auth/google';
-};
-var refreshToken$1 = function () {
-    return '/api/auth/refresh';
-};
-var requestPasswordReset$1 = function () {
-    return '/api/auth/requestPasswordReset';
-};
-var resetPassword$1 = function () {
-    return '/api/auth/resetPassword';
-};
-var plugins = function () {
-    return '/api/plugins';
-};
-var config = function () {
-    return '/api/config';
-};
+var searchEnabled = function () { return '/api/search/enable'; };
+var presets = function () { return '/api/presets'; };
+var deletePreset$1 = function () { return '/api/presets/delete'; };
+var aiEndpoints = function () { return '/api/endpoints'; };
+var tokenizer = function () { return '/api/tokenizer'; };
+var login$1 = function () { return '/api/auth/login'; };
+var logout$1 = function () { return '/api/auth/logout'; };
+var register$1 = function () { return '/api/auth/register'; };
+var loginGoogle = function () { return '/api/auth/google'; };
+var refreshToken$1 = function () { return '/api/auth/refresh'; };
+var requestPasswordReset$1 = function () { return '/api/auth/requestPasswordReset'; };
+var resetPassword$1 = function () { return '/api/auth/resetPassword'; };
+var plugins = function () { return '/api/plugins'; };
+var config = function () { return '/api/config'; };
 
 function getConversations(pageNumber) {
     return request.get(conversations(pageNumber));
@@ -263,6 +290,12 @@ function deleteConversation(payload) {
 function clearAllConversations() {
     return request.post(deleteConversation$1(), { arg: {} });
 }
+function revokeUserKey(name) {
+    return request.delete(revokeUserKey$1(name));
+}
+function revokeAllUserKeys() {
+    return request.delete(revokeAllUserKeys$1());
+}
 function getMessagesByConvoId(conversationId) {
     return request.get(messages(conversationId));
 }
@@ -279,6 +312,13 @@ function updateMessage(payload) {
     }
     return request.put(messages(conversationId, messageId), { text: text });
 }
+function updateUserKey(payload) {
+    var value = payload.value;
+    if (!value) {
+        throw new Error('value is required');
+    }
+    return request.put(keys(), payload);
+}
 function getPresets() {
     return request.get(presets());
 }
@@ -317,8 +357,9 @@ var logout = function () {
 var register = function (payload) {
     return request.post(register$1(), payload);
 };
-var refreshToken = function () {
-    return request.post(refreshToken$1());
+var refreshToken = function () { return request.post(refreshToken$1()); };
+var userKeyQuery = function (name) {
+    return request.get(userKeyQuery$1(name));
 };
 var getLoginGoogle = function () {
     return request.get(loginGoogle());
@@ -391,6 +432,7 @@ var tMessageSchema = z.object({
     sender: z.string(),
     text: z.string(),
     generation: z.string().nullable().optional(),
+    isEdited: z.boolean().optional(),
     isCreatedByUser: z.boolean(),
     error: z.boolean(),
     createdAt: z
@@ -538,14 +580,14 @@ var anthropicSchema = tConversationSchema
 })
     .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 }));
+    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 : 4000, 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,
+    maxOutputTokens: 4000,
     topP: 0.7,
     topK: 5,
 }); });
@@ -653,6 +695,7 @@ var QueryKeys;
     QueryKeys["conversation"] = "conversation";
     QueryKeys["searchEnabled"] = "searchEnabled";
     QueryKeys["user"] = "user";
+    QueryKeys["name"] = "name";
     QueryKeys["endpoints"] = "endpoints";
     QueryKeys["presets"] = "presets";
     QueryKeys["searchResults"] = "searchResults";
@@ -703,6 +746,14 @@ var useUpdateMessageMutation = function (id) {
         },
     });
 };
+var useUpdateUserKeysMutation = function () {
+    var queryClient = useQueryClient();
+    return useMutation(function (payload) { return updateUserKey(payload); }, {
+        onSuccess: function () {
+            queryClient.invalidateQueries([QueryKeys.name]);
+        },
+    });
+};
 var useDeleteConversationMutation = function (id) {
     var queryClient = useQueryClient();
     return useMutation(function (payload) { return deleteConversation(payload); }, {
@@ -720,6 +771,22 @@ var useClearConversationsMutation = function () {
         },
     });
 };
+var useRevokeUserKeyMutation = function (name) {
+    var queryClient = useQueryClient();
+    return useMutation(function () { return revokeUserKey(name); }, {
+        onSuccess: function () {
+            queryClient.invalidateQueries([QueryKeys.name]);
+        },
+    });
+};
+var useRevokeAllUserKeysMutation = function () {
+    var queryClient = useQueryClient();
+    return useMutation(function () { return revokeAllUserKeys(); }, {
+        onSuccess: function () {
+            queryClient.invalidateQueries([QueryKeys.name]);
+        },
+    });
+};
 var useGetConversationsQuery = function (pageNumber, config) {
     return useQuery([QueryKeys.allConversations, pageNumber], function () { return getConversations(pageNumber); }, __assign({ refetchOnReconnect: false, refetchOnMount: false, retry: 1 }, config));
 };
@@ -801,6 +868,14 @@ var useLogoutUserMutation = function () {
 var useRefreshTokenMutation = function () {
     return useMutation(function () { return refreshToken(); }, {});
 };
+var useUserKeyQuery = function (name, config) {
+    return useQuery([QueryKeys.name, name], function () {
+        if (!name) {
+            return Promise.resolve({ expiresAt: '' });
+        }
+        return userKeyQuery(name);
+    }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false, retry: false }, config));
+};
 var useRequestPasswordResetMutation = function () {
     return useMutation(function (payload) {
         return requestPasswordReset(payload);
@@ -832,19 +907,13 @@ var useGetStartupConfig = function () {
     });
 };
 
-function setAcceptLanguageHeader(value) {
-    axios.defaults.headers.common['Accept-Language'] = value;
-}
-function setTokenHeader(token) {
-    axios.defaults.headers.common['Authorization'] = 'Bearer ' + token;
-}
-
 /* eslint-disable */
 /**
  * Copyright (C) 2016 Maxime Petazzoni <maxime.petazzoni@bulix.org>.
  * All rights reserved.
  */
 
+
 var SSE = function (url, options) {
   if (!(this instanceof SSE)) {
     return new SSE(url, options);
@@ -943,12 +1012,27 @@ var SSE = function (url, options) {
     this.close();
   };
 
-  this._onStreamProgress = function (e) {
+  this._onStreamProgress = async function (e) {
     if (!this.xhr) {
       return;
     }
 
-    if (this.xhr.status !== 200) {
+    if (this.xhr.status === 401 && !this._retry) {
+      this._retry = true;
+      try {
+        const refreshResponse = await refreshToken();
+        this.headers = {
+          'Content-Type': 'application/json',
+          Authorization: `Bearer ${refreshResponse.token}`,
+        };
+        setTokenHeader(refreshResponse.token);
+        window.dispatchEvent(new CustomEvent('tokenUpdated', { detail: refreshResponse.token }));
+        this.stream();
+      } catch (err) {
+        this._onStreamFailure(e);
+        return;
+      }
+    } else if (this.xhr.status !== 200) {
       this._onStreamFailure(e);
       return;
     }
@@ -1084,4 +1168,4 @@ function createPayload(submission) {
     return { server: server, payload: payload };
 }
 
-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, updateMessage, 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, useUpdateMessageMutation, 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, revokeAllUserKeys, revokeUserKey, searchConversations, setAcceptLanguageHeader, setTokenHeader, tAgentOptionsSchema, tConversationSchema, tExampleSchema, tMessageSchema, tPluginAuthConfigSchema, tPluginSchema, tPresetSchema, updateConversation, updateMessage, updatePreset, updateTokenCount, updateUserKey, updateUserPlugins, useAbortRequestWithMessage, useAvailablePluginsQuery, useClearConversationsMutation, useCreatePresetMutation, useDeleteConversationMutation, useDeletePresetMutation, useGetConversationByIdMutation, useGetConversationByIdQuery, useGetConversationsQuery, useGetEndpointsQuery, useGetMessagesByConvoId, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useRevokeAllUserKeysMutation, useRevokeUserKeyMutation, useSearchQuery, useUpdateConversationMutation, useUpdateMessageMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserKeysMutation, useUpdateUserPluginsMutation, useUserKeyQuery, userKeyQuery };