librechat-data-provider 0.1.9 → 0.2.1

package/dist/index.es.js

@@ -1,6 +1,6 @@
 import axios from 'axios';
 import { z } from 'zod';
-import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
+import { useQueryClient, useMutation, useQuery } from '@tanstack/react-query';
 
 /******************************************************************************
 Copyright (c) Microsoft Corporation.
@@ -94,49 +94,59 @@ var processQueue = function (error, 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) {
+axios.interceptors.response.use(function (response) { return response; }, function (error) { return __awaiter(void 0, void 0, void 0, function () {
+    var originalRequest, token, err_1;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                originalRequest = error.config;
+                if (!(error.response.status === 401 && !originalRequest._retry)) return [3 /*break*/, 7];
+                if (!isRefreshing) return [3 /*break*/, 6];
+                _a.label = 1;
+            case 1:
+                _a.trys.push([1, 4, , 6]);
+                return [4 /*yield*/, new Promise(function (resolve, reject) {
+                        failedQueue.push({ resolve: resolve, reject: reject });
+                    })];
+            case 2:
+                token = _a.sent();
                 originalRequest.headers['Authorization'] = 'Bearer ' + token;
-                return axios(originalRequest);
-            })
-                .catch(function (err) {
-                return Promise.reject(err);
-            });
+                return [4 /*yield*/, axios(originalRequest)];
+            case 3: return [2 /*return*/, _a.sent()];
+            case 4:
+                err_1 = _a.sent();
+                return [4 /*yield*/, Promise.reject(err_1)];
+            case 5: return [2 /*return*/, _a.sent()];
+            case 6:
+                originalRequest._retry = true;
+                isRefreshing = true;
+                return [2 /*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;
+                        });
+                    })];
+            case 7: return [2 /*return*/, Promise.reject(error)];
         }
-        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;
@@ -245,6 +255,7 @@ var request = {
 };
 
 var user = function () { return '/api/user'; };
+var balance = function () { return '/api/balance'; };
 var userPlugins = function () { return '/api/user/plugins'; };
 var messages = function (conversationId, messageId) {
     return "/api/messages/".concat(conversationId).concat(messageId ? "/".concat(messageId) : '');
@@ -338,6 +349,9 @@ function getSearchEnabled() {
 function getUser() {
     return request.get(user());
 }
+function getUserBalance() {
+    return request.get(balance());
+}
 var searchConversations = function (q, pageNumber) { return __awaiter(void 0, void 0, void 0, function () {
     return __generator(this, function (_a) {
         return [2 /*return*/, request.get(search(q, pageNumber))];
@@ -346,9 +360,11 @@ var searchConversations = function (q, pageNumber) { return __awaiter(void 0, vo
 var getAIEndpoints = function () {
     return request.get(aiEndpoints());
 };
-var getModels = function () {
-    return request.get(models());
-};
+var getModels = function () { return __awaiter(void 0, void 0, void 0, function () {
+    return __generator(this, function (_a) {
+        return [2 /*return*/, request.get(models())];
+    });
+}); };
 var updateTokenCount = function (text) {
     return request.post(tokenizer(), { arg: text });
 };
@@ -698,13 +714,14 @@ var getResponseSender = function (endpointOption) {
 
 var QueryKeys;
 (function (QueryKeys) {
-    QueryKeys["messages"] = "messsages";
+    QueryKeys["messages"] = "messages";
     QueryKeys["allConversations"] = "allConversations";
     QueryKeys["conversation"] = "conversation";
     QueryKeys["searchEnabled"] = "searchEnabled";
     QueryKeys["user"] = "user";
     QueryKeys["name"] = "name";
     QueryKeys["models"] = "models";
+    QueryKeys["balance"] = "balance";
     QueryKeys["endpoints"] = "endpoints";
     QueryKeys["presets"] = "presets";
     QueryKeys["searchResults"] = "searchResults";
@@ -713,9 +730,14 @@ var QueryKeys;
     QueryKeys["startupConfig"] = "startupConfig";
 })(QueryKeys || (QueryKeys = {}));
 var useAbortRequestWithMessage = function () {
+    var queryClient = useQueryClient();
     return useMutation(function (_a) {
         var endpoint = _a.endpoint, abortKey = _a.abortKey, message = _a.message;
         return abortRequestWithMessage(endpoint, abortKey, message);
+    }, {
+        onSuccess: function () {
+            queryClient.invalidateQueries([QueryKeys.balance]);
+        },
     });
 };
 var useGetUserQuery = function (config) {
@@ -724,6 +746,9 @@ var useGetUserQuery = function (config) {
 var useGetMessagesByConvoId = function (id, config) {
     return useQuery([QueryKeys.messages, id], function () { return getMessagesByConvoId(id); }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false }, config));
 };
+var useGetUserBalance = function (config) {
+    return useQuery([QueryKeys.balance], function () { return getUserBalance(); }, __assign({ refetchOnWindowFocus: true, refetchOnReconnect: true, refetchOnMount: true }, config));
+};
 var useGetConversationByIdQuery = function (id, config) {
     return useQuery([QueryKeys.conversation, id], function () { return getConversationById(id); }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false }, config));
 };
@@ -809,12 +834,8 @@ var useGetEndpointsQuery = function () {
         refetchOnMount: false,
     });
 };
-var useGetModelsQuery = function () {
-    return useQuery([QueryKeys.models], function () { return getModels(); }, {
-        refetchOnWindowFocus: false,
-        refetchOnReconnect: false,
-        refetchOnMount: false,
-    });
+var useGetModelsQuery = function (config) {
+    return useQuery([QueryKeys.models], function () { return getModels(); }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false }, config));
 };
 var useCreatePresetMutation = function () {
     var queryClient = useQueryClient();
@@ -1192,4 +1213,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, getModels, 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, useGetModelsQuery, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useRevokeAllUserKeysMutation, useRevokeUserKeyMutation, useSearchQuery, useUpdateConversationMutation, useUpdateMessageMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserKeysMutation, useUpdateUserPluginsMutation, useUserKeyQuery, userKeyQuery };
+export { EModelEndpoint, QueryKeys, SSE, abortRequestWithMessage, anthropicSchema, bingAISchema, chatGPTBrowserSchema, clearAllConversations, createPayload, createPreset, deleteConversation, deletePreset, eModelEndpointSchema, getAIEndpoints, getAvailablePlugins, getConversationById, getConversations, getLoginGoogle, getMessagesByConvoId, getModels, getPresets, getResponseSender, getSearchEnabled, getStartupConfig, getUser, getUserBalance, googleSchema, gptPluginsSchema, login, logout, openAISchema, parseConvo, refreshToken, register, request, 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, useGetModelsQuery, useGetPresetsQuery, useGetSearchEnabledQuery, useGetStartupConfig, useGetUserBalance, useGetUserQuery, useLoginUserMutation, useLogoutUserMutation, useRefreshTokenMutation, useRegisterUserMutation, useRequestPasswordResetMutation, useResetPasswordMutation, useRevokeAllUserKeysMutation, useRevokeUserKeyMutation, useSearchQuery, useUpdateConversationMutation, useUpdateMessageMutation, useUpdatePresetMutation, useUpdateTokenCountMutation, useUpdateUserKeysMutation, useUpdateUserPluginsMutation, useUserKeyQuery, userKeyQuery };

package/dist/index.js

@@ -96,49 +96,59 @@ var processQueue = function (error, 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) {
+axios.interceptors.response.use(function (response) { return response; }, function (error) { return __awaiter(void 0, void 0, void 0, function () {
+    var originalRequest, token, err_1;
+    return __generator(this, function (_a) {
+        switch (_a.label) {
+            case 0:
+                originalRequest = error.config;
+                if (!(error.response.status === 401 && !originalRequest._retry)) return [3 /*break*/, 7];
+                if (!isRefreshing) return [3 /*break*/, 6];
+                _a.label = 1;
+            case 1:
+                _a.trys.push([1, 4, , 6]);
+                return [4 /*yield*/, new Promise(function (resolve, reject) {
+                        failedQueue.push({ resolve: resolve, reject: reject });
+                    })];
+            case 2:
+                token = _a.sent();
                 originalRequest.headers['Authorization'] = 'Bearer ' + token;
-                return axios(originalRequest);
-            })
-                .catch(function (err) {
-                return Promise.reject(err);
-            });
+                return [4 /*yield*/, axios(originalRequest)];
+            case 3: return [2 /*return*/, _a.sent()];
+            case 4:
+                err_1 = _a.sent();
+                return [4 /*yield*/, Promise.reject(err_1)];
+            case 5: return [2 /*return*/, _a.sent()];
+            case 6:
+                originalRequest._retry = true;
+                isRefreshing = true;
+                return [2 /*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;
+                        });
+                    })];
+            case 7: return [2 /*return*/, Promise.reject(error)];
         }
-        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;
@@ -247,6 +257,7 @@ var request = {
 };
 
 var user = function () { return '/api/user'; };
+var balance = function () { return '/api/balance'; };
 var userPlugins = function () { return '/api/user/plugins'; };
 var messages = function (conversationId, messageId) {
     return "/api/messages/".concat(conversationId).concat(messageId ? "/".concat(messageId) : '');
@@ -340,6 +351,9 @@ function getSearchEnabled() {
 function getUser() {
     return request.get(user());
 }
+function getUserBalance() {
+    return request.get(balance());
+}
 var searchConversations = function (q, pageNumber) { return __awaiter(void 0, void 0, void 0, function () {
     return __generator(this, function (_a) {
         return [2 /*return*/, request.get(search(q, pageNumber))];
@@ -348,9 +362,11 @@ var searchConversations = function (q, pageNumber) { return __awaiter(void 0, vo
 var getAIEndpoints = function () {
     return request.get(aiEndpoints());
 };
-var getModels = function () {
-    return request.get(models());
-};
+var getModels = function () { return __awaiter(void 0, void 0, void 0, function () {
+    return __generator(this, function (_a) {
+        return [2 /*return*/, request.get(models())];
+    });
+}); };
 var updateTokenCount = function (text) {
     return request.post(tokenizer(), { arg: text });
 };
@@ -700,13 +716,14 @@ var getResponseSender = function (endpointOption) {
 
 exports.QueryKeys = void 0;
 (function (QueryKeys) {
-    QueryKeys["messages"] = "messsages";
+    QueryKeys["messages"] = "messages";
     QueryKeys["allConversations"] = "allConversations";
     QueryKeys["conversation"] = "conversation";
     QueryKeys["searchEnabled"] = "searchEnabled";
     QueryKeys["user"] = "user";
     QueryKeys["name"] = "name";
     QueryKeys["models"] = "models";
+    QueryKeys["balance"] = "balance";
     QueryKeys["endpoints"] = "endpoints";
     QueryKeys["presets"] = "presets";
     QueryKeys["searchResults"] = "searchResults";
@@ -715,9 +732,14 @@ exports.QueryKeys = void 0;
     QueryKeys["startupConfig"] = "startupConfig";
 })(exports.QueryKeys || (exports.QueryKeys = {}));
 var useAbortRequestWithMessage = function () {
+    var queryClient = reactQuery.useQueryClient();
     return reactQuery.useMutation(function (_a) {
         var endpoint = _a.endpoint, abortKey = _a.abortKey, message = _a.message;
         return abortRequestWithMessage(endpoint, abortKey, message);
+    }, {
+        onSuccess: function () {
+            queryClient.invalidateQueries([exports.QueryKeys.balance]);
+        },
     });
 };
 var useGetUserQuery = function (config) {
@@ -726,6 +748,9 @@ var useGetUserQuery = function (config) {
 var useGetMessagesByConvoId = function (id, config) {
     return reactQuery.useQuery([exports.QueryKeys.messages, id], function () { return getMessagesByConvoId(id); }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false }, config));
 };
+var useGetUserBalance = function (config) {
+    return reactQuery.useQuery([exports.QueryKeys.balance], function () { return getUserBalance(); }, __assign({ refetchOnWindowFocus: true, refetchOnReconnect: true, refetchOnMount: true }, config));
+};
 var useGetConversationByIdQuery = function (id, config) {
     return reactQuery.useQuery([exports.QueryKeys.conversation, id], function () { return getConversationById(id); }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false }, config));
 };
@@ -811,12 +836,8 @@ var useGetEndpointsQuery = function () {
         refetchOnMount: false,
     });
 };
-var useGetModelsQuery = function () {
-    return reactQuery.useQuery([exports.QueryKeys.models], function () { return getModels(); }, {
-        refetchOnWindowFocus: false,
-        refetchOnReconnect: false,
-        refetchOnMount: false,
-    });
+var useGetModelsQuery = function (config) {
+    return reactQuery.useQuery([exports.QueryKeys.models], function () { return getModels(); }, __assign({ refetchOnWindowFocus: false, refetchOnReconnect: false, refetchOnMount: false }, config));
 };
 var useCreatePresetMutation = function () {
     var queryClient = reactQuery.useQueryClient();
@@ -1217,6 +1238,7 @@ exports.getResponseSender = getResponseSender;
 exports.getSearchEnabled = getSearchEnabled;
 exports.getStartupConfig = getStartupConfig;
 exports.getUser = getUser;
+exports.getUserBalance = getUserBalance;
 exports.googleSchema = googleSchema;
 exports.gptPluginsSchema = gptPluginsSchema;
 exports.login = login;
@@ -1225,6 +1247,7 @@ exports.openAISchema = openAISchema;
 exports.parseConvo = parseConvo;
 exports.refreshToken = refreshToken;
 exports.register = register;
+exports.request = request;
 exports.requestPasswordReset = requestPasswordReset;
 exports.resetPassword = resetPassword;
 exports.revokeAllUserKeys = revokeAllUserKeys;
@@ -1260,6 +1283,7 @@ exports.useGetModelsQuery = useGetModelsQuery;
 exports.useGetPresetsQuery = useGetPresetsQuery;
 exports.useGetSearchEnabledQuery = useGetSearchEnabledQuery;
 exports.useGetStartupConfig = useGetStartupConfig;
+exports.useGetUserBalance = useGetUserBalance;
 exports.useGetUserQuery = useGetUserQuery;
 exports.useLoginUserMutation = useLoginUserMutation;
 exports.useLogoutUserMutation = useLogoutUserMutation;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "librechat-data-provider",
-  "version": "0.1.9",
+  "version": "0.2.1",
   "description": "data services for librechat apps",
   "main": "dist/index.js",
   "module": "dist/index.es.js",
@@ -28,7 +28,8 @@
   "dependencies": {
     "@tanstack/react-query": "^4.28.0",
     "axios": "^1.3.4",
-    "zod": "^3.22.2"
+    "openai": "^4.11.1",
+    "zod": "^3.22.4"
   },
   "devDependencies": {
     "@babel/preset-env": "^7.21.5",

package/src/api-endpoints.ts

@@ -1,5 +1,7 @@
 export const user = () => '/api/user';
 
+export const balance = () => '/api/balance';
+
 export const userPlugins = () => '/api/user/plugins';
 
 export const messages = (conversationId: string, messageId?: string) =>

package/src/data-service.ts

@@ -90,6 +90,10 @@ export function getUser(): Promise<t.TUser> {
   return request.get(endpoints.user());
 }
 
+export function getUserBalance(): Promise<string> {
+  return request.get(endpoints.balance());
+}
+
 export const searchConversations = async (
   q: string,
   pageNumber: string,
@@ -101,7 +105,7 @@ export const getAIEndpoints = () => {
   return request.get(endpoints.aiEndpoints());
 };
 
-export const getModels = () => {
+export const getModels = async (): Promise<t.TModelsConfig> => {
   return request.get(endpoints.models());
 };
 

package/src/index.ts

@@ -1,5 +1,5 @@
 export * from './data-service';
-export * from './request';
+export { default as request } from './request';
 export * from './types';
 export * from './react-query-service';
 export * from './headers-helpers';

package/src/react-query-service.ts

@@ -11,13 +11,14 @@ import * as s from './schemas';
 import * as dataService from './data-service';
 
 export enum QueryKeys {
-  messages = 'messsages',
+  messages = 'messages',
   allConversations = 'allConversations',
   conversation = 'conversation',
   searchEnabled = 'searchEnabled',
   user = 'user',
   name = 'name', // user key name
   models = 'models',
+  balance = 'balance',
   endpoints = 'endpoints',
   presets = 'presets',
   searchResults = 'searchResults',
@@ -31,8 +32,15 @@ export const useAbortRequestWithMessage = (): UseMutationResult<
   Error,
   { endpoint: string; abortKey: string; message: string }
 > => {
-  return useMutation(({ endpoint, abortKey, message }) =>
-    dataService.abortRequestWithMessage(endpoint, abortKey, message),
+  const queryClient = useQueryClient();
+  return useMutation(
+    ({ endpoint, abortKey, message }) =>
+      dataService.abortRequestWithMessage(endpoint, abortKey, message),
+    {
+      onSuccess: () => {
+        queryClient.invalidateQueries([QueryKeys.balance]);
+      },
+    },
   );
 };
 
@@ -64,6 +72,17 @@ export const useGetMessagesByConvoId = (
   );
 };
 
+export const useGetUserBalance = (
+  config?: UseQueryOptions<string>,
+): QueryObserverResult<string> => {
+  return useQuery<string>([QueryKeys.balance], () => dataService.getUserBalance(), {
+    refetchOnWindowFocus: true,
+    refetchOnReconnect: true,
+    refetchOnMount: true,
+    ...config,
+  });
+};
+
 export const useGetConversationByIdQuery = (
   id: string,
   config?: UseQueryOptions<s.TConversation>,
@@ -219,11 +238,14 @@ export const useGetEndpointsQuery = (): QueryObserverResult<t.TEndpointsConfig>
   });
 };
 
-export const useGetModelsQuery = (): QueryObserverResult<t.TModelsConfig> => {
-  return useQuery([QueryKeys.models], () => dataService.getModels(), {
+export const useGetModelsQuery = (
+  config?: UseQueryOptions<t.TModelsConfig>,
+): QueryObserverResult<t.TModelsConfig> => {
+  return useQuery<t.TModelsConfig>([QueryKeys.models], () => dataService.getModels(), {
     refetchOnWindowFocus: false,
     refetchOnReconnect: false,
     refetchOnMount: false,
+    ...config,
   });
 };
 

package/src/request.ts

@@ -21,20 +21,19 @@ const processQueue = (error: AxiosError | null, token: string | null = null) =>
 
 axios.interceptors.response.use(
   (response) => response,
-  (error) => {
+  async (error) => {
     const originalRequest = error.config;
     if (error.response.status === 401 && !originalRequest._retry) {
       if (isRefreshing) {
-        return new Promise(function (resolve, reject) {
-          failedQueue.push({ resolve, reject });
-        })
-          .then((token) => {
-            originalRequest.headers['Authorization'] = 'Bearer ' + token;
-            return axios(originalRequest);
-          })
-          .catch((err) => {
-            return Promise.reject(err);
+        try {
+          const token = await new Promise(function (resolve, reject) {
+            failedQueue.push({ resolve, reject });
           });
+          originalRequest.headers['Authorization'] = 'Bearer ' + token;
+          return await axios(originalRequest);
+        } catch (err) {
+          return await Promise.reject(err);
+        }
       }
 
       originalRequest._retry = true;

package/src/types.ts

@@ -1,5 +1,10 @@
-import type { TResPlugin, TMessage, TConversation, TEndpointOption } from './schemas';
+import OpenAI from 'openai';
 import type { UseMutationResult } from '@tanstack/react-query';
+import type { TResPlugin, TMessage, TConversation, TEndpointOption } from './schemas';
+
+export type TOpenAIMessage = OpenAI.Chat.ChatCompletionMessageParam;
+export type TOpenAIFunction = OpenAI.Chat.ChatCompletionCreateParams.Function;
+export type TOpenAIFunctionCall = OpenAI.Chat.ChatCompletionCreateParams.FunctionCallOption;
 
 export type TMutation = UseMutationResult<unknown>;
 
@@ -69,6 +74,7 @@ export type TGetConversationsResponse = {
 export type TUpdateMessageRequest = {
   conversationId: string;
   messageId: string;
+  model: string;
   text: string;
 };
 
@@ -174,6 +180,7 @@ export type TStartupConfig = {
   registrationEnabled: boolean;
   socialLoginEnabled: boolean;
   emailEnabled: boolean;
+  checkBalance: boolean;
 };
 
 export type TRefreshTokenResponse = {

package/types/api-endpoints.d.ts

@@ -1,4 +1,5 @@
 export declare const user: () => string;
+export declare const balance: () => string;
 export declare const userPlugins: () => string;
 export declare const messages: (conversationId: string, messageId?: string) => string;
 export declare const keys: () => string;

package/types/data-service.d.ts

@@ -17,9 +17,10 @@ export declare function updatePreset(payload: s.TPreset): Promise<s.TPreset[]>;
 export declare function deletePreset(arg: s.TPreset | object): Promise<s.TPreset[]>;
 export declare function getSearchEnabled(): Promise<boolean>;
 export declare function getUser(): Promise<t.TUser>;
+export declare function getUserBalance(): Promise<string>;
 export declare const searchConversations: (q: string, pageNumber: string) => Promise<t.TSearchResults>;
 export declare const getAIEndpoints: () => Promise<unknown>;
-export declare const getModels: () => Promise<unknown>;
+export declare const getModels: () => Promise<t.TModelsConfig>;
 export declare const updateTokenCount: (text: string) => Promise<any>;
 export declare const login: (payload: t.TLoginUser) => Promise<any>;
 export declare const logout: () => Promise<any>;

package/types/index.d.ts

@@ -1,5 +1,5 @@
 export * from './data-service';
-export * from './request';
+export { default as request } from './request';
 export * from './types';
 export * from './react-query-service';
 export * from './headers-helpers';

package/types/react-query-service.d.ts

@@ -2,13 +2,14 @@ import { UseQueryOptions, UseMutationResult, QueryObserverResult } from '@tansta
 import * as t from './types';
 import * as s from './schemas';
 export declare enum QueryKeys {
-    messages = "messsages",
+    messages = "messages",
     allConversations = "allConversations",
     conversation = "conversation",
     searchEnabled = "searchEnabled",
     user = "user",
     name = "name",
     models = "models",
+    balance = "balance",
     endpoints = "endpoints",
     presets = "presets",
     searchResults = "searchResults",
@@ -23,6 +24,7 @@ export declare const useAbortRequestWithMessage: () => UseMutationResult<void, E
 }>;
 export declare const useGetUserQuery: (config?: UseQueryOptions<t.TUser>) => QueryObserverResult<t.TUser>;
 export declare const useGetMessagesByConvoId: (id: string, config?: UseQueryOptions<s.TMessage[]>) => QueryObserverResult<s.TMessage[]>;
+export declare const useGetUserBalance: (config?: UseQueryOptions<string>) => QueryObserverResult<string>;
 export declare const useGetConversationByIdQuery: (id: string, config?: UseQueryOptions<s.TConversation>) => QueryObserverResult<s.TConversation>;
 export declare const useGetConversationByIdMutation: (id: string) => UseMutationResult<s.TConversation>;
 export declare const useUpdateConversationMutation: (id: string) => UseMutationResult<t.TUpdateConversationResponse, unknown, t.TUpdateConversationRequest, unknown>;
@@ -35,7 +37,7 @@ export declare const useRevokeAllUserKeysMutation: () => UseMutationResult<unkno
 export declare const useGetConversationsQuery: (pageNumber: string, config?: UseQueryOptions<t.TGetConversationsResponse>) => QueryObserverResult<t.TGetConversationsResponse>;
 export declare const useGetSearchEnabledQuery: (config?: UseQueryOptions<boolean>) => QueryObserverResult<boolean>;
 export declare const useGetEndpointsQuery: () => QueryObserverResult<t.TEndpointsConfig>;
-export declare const useGetModelsQuery: () => QueryObserverResult<t.TModelsConfig>;
+export declare const useGetModelsQuery: (config?: UseQueryOptions<t.TModelsConfig>) => QueryObserverResult<t.TModelsConfig>;
 export declare const useCreatePresetMutation: () => UseMutationResult<s.TPreset[], unknown, s.TPreset, unknown>;
 export declare const useUpdatePresetMutation: () => UseMutationResult<s.TPreset[], unknown, s.TPreset, unknown>;
 export declare const useGetPresetsQuery: (config?: UseQueryOptions<s.TPreset[]>) => QueryObserverResult<s.TPreset[], unknown>;

package/types/types.d.ts

@@ -1,5 +1,9 @@
-import type { TResPlugin, TMessage, TConversation, TEndpointOption } from './schemas';
+import OpenAI from 'openai';
 import type { UseMutationResult } from '@tanstack/react-query';
+import type { TResPlugin, TMessage, TConversation, TEndpointOption } from './schemas';
+export type TOpenAIMessage = OpenAI.Chat.ChatCompletionMessageParam;
+export type TOpenAIFunction = OpenAI.Chat.ChatCompletionCreateParams.Function;
+export type TOpenAIFunctionCall = OpenAI.Chat.ChatCompletionCreateParams.FunctionCallOption;
 export type TMutation = UseMutationResult<unknown>;
 export * from './schemas';
 export type TMessages = TMessage[];
@@ -58,6 +62,7 @@ export type TGetConversationsResponse = {
 export type TUpdateMessageRequest = {
     conversationId: string;
     messageId: string;
+    model: string;
     text: string;
 };
 export type TUpdateUserKeyRequest = {
@@ -144,6 +149,7 @@ export type TStartupConfig = {
     registrationEnabled: boolean;
     socialLoginEnabled: boolean;
     emailEnabled: boolean;
+    checkBalance: boolean;
 };
 export type TRefreshTokenResponse = {
     token: string;