librechat-data-provider 0.1.8 → 0.2.0

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.
@@ -245,6 +245,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) : '');
@@ -266,6 +267,7 @@ 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 models = function () { return '/api/models'; };
 var tokenizer = function () { return '/api/tokenizer'; };
 var login$1 = function () { return '/api/auth/login'; };
 var logout$1 = function () { return '/api/auth/logout'; };
@@ -337,6 +339,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))];
@@ -345,6 +350,9 @@ 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 updateTokenCount = function (text) {
     return request.post(tokenizer(), { arg: text });
 };
@@ -428,7 +436,7 @@ var tMessageSchema = z.object({
     overrideParentMessageId: z.string().nullable().optional(),
     bg: z.string().nullable().optional(),
     model: z.string().nullable().optional(),
-    title: z.string().nullable().optional(),
+    title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
     sender: z.string(),
     text: z.string(),
     generation: z.string().nullable().optional(),
@@ -451,7 +459,7 @@ var tMessageSchema = z.object({
 });
 var tConversationSchema = z.object({
     conversationId: z.string().nullable(),
-    title: z.string(),
+    title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
     user: z.string().optional(),
     endpoint: eModelEndpointSchema.nullable(),
     suggestions: z.array(z.string()).optional(),
@@ -660,14 +668,18 @@ function getFirstDefinedValue(possibleValues) {
     return returnValue;
 }
 var parseConvo = function (endpoint, conversation, possibleValues) {
-    var _a;
+    var _a, _b;
     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;
+    var _c = possibleValues !== null && possibleValues !== void 0 ? possibleValues : {}, models = _c.models, secondaryModels = _c.secondaryModels;
+    if (models && convo) {
+        convo.model = (_a = getFirstDefinedValue(models)) !== null && _a !== void 0 ? _a : convo.model;
+    }
+    if (secondaryModels && convo.agentOptions) {
+        convo.agentOptions.model = (_b = getFirstDefinedValue(secondaryModels)) !== null && _b !== void 0 ? _b : convo.agentOptions.model;
     }
     return convo;
 };
@@ -696,6 +708,8 @@ var QueryKeys;
     QueryKeys["searchEnabled"] = "searchEnabled";
     QueryKeys["user"] = "user";
     QueryKeys["name"] = "name";
+    QueryKeys["models"] = "models";
+    QueryKeys["balance"] = "balance";
     QueryKeys["endpoints"] = "endpoints";
     QueryKeys["presets"] = "presets";
     QueryKeys["searchResults"] = "searchResults";
@@ -704,9 +718,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) {
@@ -715,6 +734,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));
 };
@@ -800,6 +822,13 @@ var useGetEndpointsQuery = function () {
         refetchOnMount: false,
     });
 };
+var useGetModelsQuery = function () {
+    return useQuery([QueryKeys.models], function () { return getModels(); }, {
+        refetchOnWindowFocus: false,
+        refetchOnReconnect: false,
+        refetchOnMount: false,
+    });
+};
 var useCreatePresetMutation = function () {
     var queryClient = useQueryClient();
     return useMutation(function (payload) { return createPreset(payload); }, {
@@ -847,6 +876,9 @@ var useLoginUserMutation = function () {
         onSuccess: function () {
             queryClient.invalidateQueries([QueryKeys.user]);
         },
+        onMutate: function () {
+            queryClient.invalidateQueries([QueryKeys.models]);
+        },
     });
 };
 var useRegisterUserMutation = function () {
@@ -866,7 +898,12 @@ var useLogoutUserMutation = function () {
     });
 };
 var useRefreshTokenMutation = function () {
-    return useMutation(function () { return refreshToken(); }, {});
+    var queryClient = useQueryClient();
+    return useMutation(function () { return refreshToken(); }, {
+        onMutate: function () {
+            queryClient.invalidateQueries([QueryKeys.models]);
+        },
+    });
 };
 var useUserKeyQuery = function (name, config) {
     return useQuery([QueryKeys.name, name], function () {
@@ -1168,4 +1205,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, 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 };
+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, 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

@@ -247,6 +247,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) : '');
@@ -268,6 +269,7 @@ 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 models = function () { return '/api/models'; };
 var tokenizer = function () { return '/api/tokenizer'; };
 var login$1 = function () { return '/api/auth/login'; };
 var logout$1 = function () { return '/api/auth/logout'; };
@@ -339,6 +341,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))];
@@ -347,6 +352,9 @@ 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 updateTokenCount = function (text) {
     return request.post(tokenizer(), { arg: text });
 };
@@ -430,7 +438,7 @@ var tMessageSchema = zod.z.object({
     overrideParentMessageId: zod.z.string().nullable().optional(),
     bg: zod.z.string().nullable().optional(),
     model: zod.z.string().nullable().optional(),
-    title: zod.z.string().nullable().optional(),
+    title: zod.z.string().nullable().or(zod.z.literal('New Chat')).default('New Chat'),
     sender: zod.z.string(),
     text: zod.z.string(),
     generation: zod.z.string().nullable().optional(),
@@ -453,7 +461,7 @@ var tMessageSchema = zod.z.object({
 });
 var tConversationSchema = zod.z.object({
     conversationId: zod.z.string().nullable(),
-    title: zod.z.string(),
+    title: zod.z.string().nullable().or(zod.z.literal('New Chat')).default('New Chat'),
     user: zod.z.string().optional(),
     endpoint: eModelEndpointSchema.nullable(),
     suggestions: zod.z.array(zod.z.string()).optional(),
@@ -662,14 +670,18 @@ function getFirstDefinedValue(possibleValues) {
     return returnValue;
 }
 var parseConvo = function (endpoint, conversation, possibleValues) {
-    var _a;
+    var _a, _b;
     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;
+    var _c = possibleValues !== null && possibleValues !== void 0 ? possibleValues : {}, models = _c.models, secondaryModels = _c.secondaryModels;
+    if (models && convo) {
+        convo.model = (_a = getFirstDefinedValue(models)) !== null && _a !== void 0 ? _a : convo.model;
+    }
+    if (secondaryModels && convo.agentOptions) {
+        convo.agentOptions.model = (_b = getFirstDefinedValue(secondaryModels)) !== null && _b !== void 0 ? _b : convo.agentOptions.model;
     }
     return convo;
 };
@@ -698,6 +710,8 @@ exports.QueryKeys = void 0;
     QueryKeys["searchEnabled"] = "searchEnabled";
     QueryKeys["user"] = "user";
     QueryKeys["name"] = "name";
+    QueryKeys["models"] = "models";
+    QueryKeys["balance"] = "balance";
     QueryKeys["endpoints"] = "endpoints";
     QueryKeys["presets"] = "presets";
     QueryKeys["searchResults"] = "searchResults";
@@ -706,9 +720,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) {
@@ -717,6 +736,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));
 };
@@ -802,6 +824,13 @@ var useGetEndpointsQuery = function () {
         refetchOnMount: false,
     });
 };
+var useGetModelsQuery = function () {
+    return reactQuery.useQuery([exports.QueryKeys.models], function () { return getModels(); }, {
+        refetchOnWindowFocus: false,
+        refetchOnReconnect: false,
+        refetchOnMount: false,
+    });
+};
 var useCreatePresetMutation = function () {
     var queryClient = reactQuery.useQueryClient();
     return reactQuery.useMutation(function (payload) { return createPreset(payload); }, {
@@ -849,6 +878,9 @@ var useLoginUserMutation = function () {
         onSuccess: function () {
             queryClient.invalidateQueries([exports.QueryKeys.user]);
         },
+        onMutate: function () {
+            queryClient.invalidateQueries([exports.QueryKeys.models]);
+        },
     });
 };
 var useRegisterUserMutation = function () {
@@ -868,7 +900,12 @@ var useLogoutUserMutation = function () {
     });
 };
 var useRefreshTokenMutation = function () {
-    return reactQuery.useMutation(function () { return refreshToken(); }, {});
+    var queryClient = reactQuery.useQueryClient();
+    return reactQuery.useMutation(function () { return refreshToken(); }, {
+        onMutate: function () {
+            queryClient.invalidateQueries([exports.QueryKeys.models]);
+        },
+    });
 };
 var useUserKeyQuery = function (name, config) {
     return reactQuery.useQuery([exports.QueryKeys.name, name], function () {
@@ -1187,11 +1224,13 @@ exports.getConversationById = getConversationById;
 exports.getConversations = getConversations;
 exports.getLoginGoogle = getLoginGoogle;
 exports.getMessagesByConvoId = getMessagesByConvoId;
+exports.getModels = getModels;
 exports.getPresets = getPresets;
 exports.getResponseSender = getResponseSender;
 exports.getSearchEnabled = getSearchEnabled;
 exports.getStartupConfig = getStartupConfig;
 exports.getUser = getUser;
+exports.getUserBalance = getUserBalance;
 exports.googleSchema = googleSchema;
 exports.gptPluginsSchema = gptPluginsSchema;
 exports.login = login;
@@ -1231,9 +1270,11 @@ exports.useGetConversationByIdQuery = useGetConversationByIdQuery;
 exports.useGetConversationsQuery = useGetConversationsQuery;
 exports.useGetEndpointsQuery = useGetEndpointsQuery;
 exports.useGetMessagesByConvoId = useGetMessagesByConvoId;
+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.8",
+  "version": "0.2.0",
   "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) =>
@@ -36,6 +38,8 @@ export const deletePreset = () => '/api/presets/delete';
 
 export const aiEndpoints = () => '/api/endpoints';
 
+export const models = () => '/api/models';
+
 export const tokenizer = () => '/api/tokenizer';
 
 export const login = () => '/api/auth/login';

package/src/data-service.ts

@@ -1,5 +1,7 @@
 import * as t from './types';
 import * as s from './schemas';
+/* TODO: fix dependency cycle */
+// eslint-disable-next-line import/no-cycle
 import request from './request';
 import * as endpoints from './api-endpoints';
 
@@ -88,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,
@@ -99,6 +105,10 @@ export const getAIEndpoints = () => {
   return request.get(endpoints.aiEndpoints());
 };
 
+export const getModels = () => {
+  return request.get(endpoints.models());
+};
+
 export const updateTokenCount = (text: string) => {
   return request.post(endpoints.tokenizer(), { arg: text });
 };

package/src/react-query-service.ts

@@ -17,6 +17,8 @@ export enum QueryKeys {
   searchEnabled = 'searchEnabled',
   user = 'user',
   name = 'name', // user key name
+  models = 'models',
+  balance = 'balance',
   endpoints = 'endpoints',
   presets = 'presets',
   searchResults = 'searchResults',
@@ -30,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]);
+      },
+    },
   );
 };
 
@@ -63,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>,
@@ -218,6 +238,14 @@ export const useGetEndpointsQuery = (): QueryObserverResult<t.TEndpointsConfig>
   });
 };
 
+export const useGetModelsQuery = (): QueryObserverResult<t.TModelsConfig> => {
+  return useQuery([QueryKeys.models], () => dataService.getModels(), {
+    refetchOnWindowFocus: false,
+    refetchOnReconnect: false,
+    refetchOnMount: false,
+  });
+};
+
 export const useCreatePresetMutation = (): UseMutationResult<
   s.TPreset[],
   unknown,
@@ -313,6 +341,9 @@ export const useLoginUserMutation = (): UseMutationResult<
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.user]);
     },
+    onMutate: () => {
+      queryClient.invalidateQueries([QueryKeys.models]);
+    },
   });
 };
 
@@ -345,7 +376,12 @@ export const useRefreshTokenMutation = (): UseMutationResult<
   unknown,
   unknown
 > => {
-  return useMutation(() => dataService.refreshToken(), {});
+  const queryClient = useQueryClient();
+  return useMutation(() => dataService.refreshToken(), {
+    onMutate: () => {
+      queryClient.invalidateQueries([QueryKeys.models]);
+    },
+  });
 };
 
 export const useUserKeyQuery = (

package/src/request.ts

@@ -1,5 +1,6 @@
 /* eslint-disable @typescript-eslint/no-explicit-any */
 import axios, { AxiosRequestConfig, AxiosError } from 'axios';
+/* TODO: fix dependency cycle */
 // eslint-disable-next-line import/no-cycle
 import { refreshToken } from './data-service';
 import { setTokenHeader } from './headers-helpers';

package/src/schemas.ts

@@ -73,7 +73,7 @@ export const tMessageSchema = z.object({
   overrideParentMessageId: z.string().nullable().optional(),
   bg: z.string().nullable().optional(),
   model: z.string().nullable().optional(),
-  title: z.string().nullable().optional(),
+  title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
   sender: z.string(),
   text: z.string(),
   generation: z.string().nullable().optional(),
@@ -103,7 +103,7 @@ export type TMessage = z.input<typeof tMessageSchema> & {
 
 export const tConversationSchema = z.object({
   conversationId: z.string().nullable(),
-  title: z.string(),
+  title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
   user: z.string().optional(),
   endpoint: eModelEndpointSchema.nullable(),
   suggestions: z.array(z.string()).optional(),
@@ -369,7 +369,8 @@ function getFirstDefinedValue(possibleValues: string[]) {
 }
 
 type TPossibleValues = {
-  model: string[];
+  models: string[];
+  secondaryModels?: string[];
 };
 
 export const parseConvo = (
@@ -383,10 +384,15 @@ export const parseConvo = (
     throw new Error(`Unknown endpoint: ${endpoint}`);
   }
 
-  const convo = schema.parse(conversation);
+  const convo = schema.parse(conversation) as TConversation;
+  const { models, secondaryModels } = possibleValues ?? {};
 
-  if (possibleValues && convo) {
-    convo.model = getFirstDefinedValue(possibleValues.model) ?? convo.model;
+  if (models && convo) {
+    convo.model = getFirstDefinedValue(models) ?? convo.model;
+  }
+
+  if (secondaryModels && convo.agentOptions) {
+    convo.agentOptions.model = getFirstDefinedValue(secondaryModels) ?? convo.agentOptions.model;
   }
 
   return convo;

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;
 };
 
@@ -111,21 +117,16 @@ export type TSearchResults = {
 };
 
 export type TConfig = {
-  availableModels: [];
+  availableModels?: [];
   userProvide?: boolean | null;
   availableTools?: [];
   plugins?: [];
+  azure?: boolean;
 };
 
-export type TEndpointsConfig = {
-  azureOpenAI: TConfig | null;
-  bingAI: TConfig | null;
-  chatGPTBrowser: TConfig | null;
-  anthropic: TConfig | null;
-  google: TConfig | null;
-  openAI: TConfig | null;
-  gptPlugins: TConfig | null;
-};
+export type TModelsConfig = Record<string, string[]>;
+
+export type TEndpointsConfig = Record<string, TConfig | null>;
 
 export type TUpdateTokenCountResponse = {
   count: number;
@@ -179,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;
@@ -15,6 +16,7 @@ export declare const searchEnabled: () => string;
 export declare const presets: () => string;
 export declare const deletePreset: () => string;
 export declare const aiEndpoints: () => string;
+export declare const models: () => string;
 export declare const tokenizer: () => string;
 export declare const login: () => string;
 export declare const logout: () => string;

package/types/data-service.d.ts

@@ -17,8 +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 updateTokenCount: (text: string) => Promise<any>;
 export declare const login: (payload: t.TLoginUser) => Promise<any>;
 export declare const logout: () => Promise<any>;

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

@@ -8,6 +8,8 @@ export declare enum QueryKeys {
     searchEnabled = "searchEnabled",
     user = "user",
     name = "name",
+    models = "models",
+    balance = "balance",
     endpoints = "endpoints",
     presets = "presets",
     searchResults = "searchResults",
@@ -22,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>;
@@ -34,6 +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 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/schemas.d.ts

@@ -137,7 +137,7 @@ export declare const tMessageSchema: z.ZodObject<{
     overrideParentMessageId: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     bg: z.ZodOptional<z.ZodNullable<z.ZodString>>;
     model: z.ZodOptional<z.ZodNullable<z.ZodString>>;
-    title: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     sender: z.ZodString;
     text: z.ZodString;
     generation: z.ZodOptional<z.ZodNullable<z.ZodString>>;
@@ -156,6 +156,7 @@ export declare const tMessageSchema: z.ZodObject<{
     messageId: string;
     conversationId: string | null;
     parentMessageId: string | null;
+    title: string | null;
     sender: string;
     text: string;
     isCreatedByUser: boolean;
@@ -166,7 +167,6 @@ export declare const tMessageSchema: z.ZodObject<{
     overrideParentMessageId?: string | null | undefined;
     bg?: string | null | undefined;
     model?: string | null | undefined;
-    title?: string | null | undefined;
     generation?: string | null | undefined;
     isEdited?: boolean | undefined;
     current?: boolean | undefined;
@@ -205,7 +205,7 @@ export type TMessage = z.input<typeof tMessageSchema> & {
 };
 export declare const tConversationSchema: z.ZodObject<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -326,7 +326,7 @@ export declare const tConversationSchema: z.ZodObject<{
     }>>>;
 }, "strip", z.ZodTypeAny, {
     conversationId: string | null;
-    title: string;
+    title: string | null;
     createdAt: string;
     updatedAt: string;
     endpoint: EModelEndpoint | null;
@@ -383,10 +383,10 @@ export declare const tConversationSchema: z.ZodObject<{
     } | null | undefined;
 }, {
     conversationId: string | null;
-    title: string;
     createdAt: string;
     updatedAt: string;
     endpoint: EModelEndpoint | null;
+    title?: string | null | undefined;
     user?: string | undefined;
     suggestions?: string[] | undefined;
     messages?: string[] | undefined;
@@ -676,7 +676,7 @@ export declare const tPresetSchema: z.ZodObject<{
 export type TPreset = z.infer<typeof tPresetSchema>;
 export declare const openAISchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -830,7 +830,7 @@ export declare const openAISchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
 }>>;
 export declare const googleSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -1016,7 +1016,7 @@ export declare const googleSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
 }>>;
 export declare const bingAISchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -1175,7 +1175,7 @@ export declare const bingAISchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
 }>>;
 export declare const anthropicSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -1329,7 +1329,7 @@ export declare const anthropicSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
 }>>;
 export declare const chatGPTBrowserSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -1459,7 +1459,7 @@ export declare const chatGPTBrowserSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<P
 }>>;
 export declare const gptPluginsSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<{
     conversationId: z.ZodNullable<z.ZodString>;
-    title: z.ZodString;
+    title: z.ZodDefault<z.ZodUnion<[z.ZodNullable<z.ZodString>, z.ZodLiteral<"New Chat">]>>;
     user: z.ZodOptional<z.ZodString>;
     endpoint: z.ZodNullable<z.ZodNativeEnum<typeof EModelEndpoint>>;
     suggestions: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
@@ -1688,10 +1688,66 @@ export declare const gptPluginsSchema: z.ZodCatch<z.ZodEffects<z.ZodObject<Pick<
     } | null | undefined;
 }>>;
 type TPossibleValues = {
-    model: string[];
+    models: string[];
+    secondaryModels?: string[];
 };
 export declare const parseConvo: (endpoint: EModelEndpoint, conversation: Partial<TConversation | TPreset>, possibleValues?: TPossibleValues) => {
-    model: string;
+    conversationId: string | null;
+    title: string | null;
+    createdAt: string;
+    updatedAt: string;
+    endpoint: EModelEndpoint | null;
+    user?: string | undefined;
+    suggestions?: string[] | undefined;
+    messages?: string[] | undefined;
+    tools?: {
+        description: string;
+        name: string;
+        pluginKey: string;
+        icon: string;
+        authConfig: {
+            authField: string;
+            label: string;
+            description: string;
+        }[];
+        authenticated?: boolean | undefined;
+        isButton?: boolean | undefined;
+    }[] | undefined;
+    systemMessage?: string | null | undefined;
+    modelLabel?: string | null | undefined;
+    examples?: {
+        input: {
+            content: string;
+        };
+        output: {
+            content: string;
+        };
+    }[] | undefined;
+    chatGptLabel?: string | null | undefined;
+    userLabel?: string | undefined;
+    model?: string | null | undefined;
+    promptPrefix?: string | null | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    topK?: number | undefined;
+    context?: string | null | undefined;
+    top_p?: number | undefined;
+    frequency_penalty?: number | undefined;
+    presence_penalty?: number | undefined;
+    jailbreak?: boolean | undefined;
+    jailbreakConversationId?: string | null | undefined;
+    conversationSignature?: string | null | undefined;
+    parentMessageId?: string | undefined;
+    clientId?: string | null | undefined;
+    invocationId?: number | null | undefined;
+    toneStyle?: string | null | undefined;
+    maxOutputTokens?: number | undefined;
+    agentOptions?: {
+        agent: string;
+        skipCompletion: boolean;
+        model: string;
+        temperature: number;
+    } | null | undefined;
 };
 export type TEndpointOption = {
     endpoint: EModelEndpoint;

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 = {
@@ -93,20 +98,14 @@ export type TSearchResults = {
     filter: object;
 };
 export type TConfig = {
-    availableModels: [];
+    availableModels?: [];
     userProvide?: boolean | null;
     availableTools?: [];
     plugins?: [];
+    azure?: boolean;
 };
-export type TEndpointsConfig = {
-    azureOpenAI: TConfig | null;
-    bingAI: TConfig | null;
-    chatGPTBrowser: TConfig | null;
-    anthropic: TConfig | null;
-    google: TConfig | null;
-    openAI: TConfig | null;
-    gptPlugins: TConfig | null;
-};
+export type TModelsConfig = Record<string, string[]>;
+export type TEndpointsConfig = Record<string, TConfig | null>;
 export type TUpdateTokenCountResponse = {
     count: number;
 };
@@ -150,6 +149,7 @@ export type TStartupConfig = {
     registrationEnabled: boolean;
     socialLoginEnabled: boolean;
     emailEnabled: boolean;
+    checkBalance: boolean;
 };
 export type TRefreshTokenResponse = {
     token: string;