librechat-data-provider 0.6.3 → 0.6.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "librechat-data-provider",
-  "version": "0.6.3",
+  "version": "0.6.5",
   "description": "data services for librechat apps",
   "main": "dist/index.js",
   "module": "dist/index.es.js",

package/src/actions.ts

@@ -212,6 +212,10 @@ export function resolveRef(
   return schema as OpenAPIV3.SchemaObject;
 }
 
+function sanitizeOperationId(input: string) {
+  return input.replace(/[^a-zA-Z0-9_-]/g, '');
+}
+
 /** Function to convert OpenAPI spec to function signatures and request builders */
 export function openapiToFunction(openapiSpec: OpenAPIV3.Document): {
   functionSignatures: FunctionSignature[];
@@ -231,7 +235,8 @@ export function openapiToFunction(openapiSpec: OpenAPIV3.Document): {
       };
 
       // Operation ID is used as the function name
-      const operationId = operationObj.operationId || `${method}_${path}`;
+      const defaultOperationId = `${method}_${path}`;
+      const operationId = operationObj.operationId || sanitizeOperationId(defaultOperationId);
       const description = operationObj.summary || operationObj.description || '';
 
       const parametersSchema: ParametersSchema = { type: 'object', properties: {}, required: [] };

package/src/api-endpoints.ts

@@ -1,3 +1,5 @@
+import type { AssistantsEndpoint } from './schemas';
+
 export const user = () => '/api/user';
 
 export const balance = () => '/api/balance';
@@ -7,6 +9,13 @@ export const userPlugins = () => '/api/user/plugins';
 export const messages = (conversationId: string, messageId?: string) =>
   `/api/messages/${conversationId}${messageId ? `/${messageId}` : ''}`;
 
+const shareRoot = '/api/share';
+export const shareMessages = (shareId: string) => `${shareRoot}/${shareId}`;
+export const getSharedLinks = (pageNumber: string, isPublic: boolean) =>
+  `${shareRoot}?pageNumber=${pageNumber}&isPublic=${isPublic}`;
+export const createSharedLink = shareRoot;
+export const updateSharedLink = shareRoot;
+
 const keysEndpoint = '/api/keys';
 
 export const keys = () => keysEndpoint;
@@ -76,15 +85,32 @@ export const plugins = () => '/api/plugins';
 
 export const config = () => '/api/config';
 
-export const assistants = (id?: string, options?: Record<string, string>) => {
-  let url = '/api/assistants';
+export const assistants = ({
+  path,
+  options,
+  version,
+  endpoint,
+}: {
+  path?: string;
+  options?: object;
+  endpoint?: AssistantsEndpoint;
+  version: number | string;
+}) => {
+  let url = `/api/assistants/v${version}`;
+
+  if (path) {
+    url += `/${path}`;
+  }
 
-  if (id) {
-    url += `/${id}`;
+  if (endpoint) {
+    options = {
+      ...(options ?? {}),
+      endpoint,
+    };
   }
 
   if (options && Object.keys(options).length > 0) {
-    const queryParams = new URLSearchParams(options).toString();
+    const queryParams = new URLSearchParams(options as Record<string, string>).toString();
     url += `?${queryParams}`;
   }
 

package/src/config.ts

@@ -10,6 +10,8 @@ import { TModelsConfig } from './types';
 export const defaultSocialLogins = ['google', 'facebook', 'openid', 'github', 'discord'];
 
 export const defaultRetrievalModels = [
+  'gpt-4o',
+  'gpt-4o-2024-05-13',
   'gpt-4-turbo-preview',
   'gpt-3.5-turbo-0125',
   'gpt-4-0125-preview',
@@ -129,11 +131,17 @@ export enum Capabilities {
   tools = 'tools',
 }
 
+export const defaultAssistantsVersion = {
+  [EModelEndpoint.assistants]: 2,
+  [EModelEndpoint.azureAssistants]: 1,
+};
+
 export const assistantEndpointSchema = z.object({
   /* assistants specific */
   disableBuilder: z.boolean().optional(),
   pollIntervalMs: z.number().optional(),
   timeoutMs: z.number().optional(),
+  version: z.union([z.string(), z.number()]).default(2),
   supportedIds: z.array(z.string()).min(1).optional(),
   excludedIds: z.array(z.string()).min(1).optional(),
   retrievalModels: z.array(z.string()).min(1).optional().default(defaultRetrievalModels),
@@ -245,6 +253,7 @@ export const configSchema = z.object({
   cache: z.boolean().default(true),
   secureImageLinks: z.boolean().optional(),
   imageOutputType: z.nativeEnum(EImageOutputType).default(EImageOutputType.PNG),
+  includedTools: z.array(z.string()).optional(),
   filteredTools: z.array(z.string()).optional(),
   interface: z
     .object({
@@ -286,6 +295,7 @@ export const configSchema = z.object({
   endpoints: z
     .object({
       [EModelEndpoint.azureOpenAI]: azureEndpointSchema.optional(),
+      [EModelEndpoint.azureAssistants]: assistantEndpointSchema.optional(),
       [EModelEndpoint.assistants]: assistantEndpointSchema.optional(),
       custom: z.array(endpointSchema.partial()).optional(),
     })
@@ -323,6 +333,7 @@ export enum FetchTokenConfig {
 export const defaultEndpoints: EModelEndpoint[] = [
   EModelEndpoint.openAI,
   EModelEndpoint.assistants,
+  EModelEndpoint.azureAssistants,
   EModelEndpoint.azureOpenAI,
   EModelEndpoint.bingAI,
   EModelEndpoint.chatGPTBrowser,
@@ -335,6 +346,7 @@ export const defaultEndpoints: EModelEndpoint[] = [
 export const alternateName = {
   [EModelEndpoint.openAI]: 'OpenAI',
   [EModelEndpoint.assistants]: 'Assistants',
+  [EModelEndpoint.azureAssistants]: 'Azure Assistants',
   [EModelEndpoint.azureOpenAI]: 'Azure OpenAI',
   [EModelEndpoint.bingAI]: 'Bing',
   [EModelEndpoint.chatGPTBrowser]: 'ChatGPT',
@@ -344,24 +356,27 @@ export const alternateName = {
   [EModelEndpoint.custom]: 'Custom',
 };
 
+const sharedOpenAIModels = [
+  'gpt-3.5-turbo',
+  'gpt-3.5-turbo-0125',
+  'gpt-4-turbo',
+  'gpt-4-turbo-2024-04-09',
+  'gpt-4-0125-preview',
+  'gpt-4-turbo-preview',
+  'gpt-4-1106-preview',
+  'gpt-3.5-turbo-1106',
+  'gpt-3.5-turbo-16k-0613',
+  'gpt-3.5-turbo-16k',
+  'gpt-4',
+  'gpt-4-0314',
+  'gpt-4-32k-0314',
+  'gpt-4-0613',
+  'gpt-3.5-turbo-0613',
+];
+
 export const defaultModels = {
-  [EModelEndpoint.assistants]: [
-    'gpt-3.5-turbo',
-    'gpt-3.5-turbo-0125',
-    'gpt-4-turbo',
-    'gpt-4-turbo-2024-04-09',
-    'gpt-4-0125-preview',
-    'gpt-4-turbo-preview',
-    'gpt-4-1106-preview',
-    'gpt-3.5-turbo-1106',
-    'gpt-3.5-turbo-16k-0613',
-    'gpt-3.5-turbo-16k',
-    'gpt-4',
-    'gpt-4-0314',
-    'gpt-4-32k-0314',
-    'gpt-4-0613',
-    'gpt-3.5-turbo-0613',
-  ],
+  [EModelEndpoint.azureAssistants]: sharedOpenAIModels,
+  [EModelEndpoint.assistants]: ['gpt-4o', ...sharedOpenAIModels],
   [EModelEndpoint.google]: [
     'gemini-pro',
     'gemini-pro-vision',
@@ -389,25 +404,13 @@ export const defaultModels = {
     'claude-instant-1-100k',
   ],
   [EModelEndpoint.openAI]: [
-    'gpt-3.5-turbo-0125',
-    'gpt-4-turbo',
-    'gpt-4-turbo-2024-04-09',
-    'gpt-3.5-turbo-16k-0613',
-    'gpt-3.5-turbo-16k',
-    'gpt-4-turbo-preview',
-    'gpt-4-0125-preview',
-    'gpt-4-1106-preview',
-    'gpt-3.5-turbo',
-    'gpt-3.5-turbo-1106',
+    'gpt-4o',
+    ...sharedOpenAIModels,
     'gpt-4-vision-preview',
-    'gpt-4',
     'gpt-3.5-turbo-instruct-0914',
-    'gpt-3.5-turbo-0613',
     'gpt-3.5-turbo-0301',
     'gpt-3.5-turbo-instruct',
-    'gpt-4-0613',
     'text-davinci-003',
-    'gpt-4-0314',
   ],
 };
 
@@ -438,7 +441,8 @@ export const EndpointURLs: { [key in EModelEndpoint]: string } = {
   [EModelEndpoint.gptPlugins]: `/api/ask/${EModelEndpoint.gptPlugins}`,
   [EModelEndpoint.azureOpenAI]: `/api/ask/${EModelEndpoint.azureOpenAI}`,
   [EModelEndpoint.chatGPTBrowser]: `/api/ask/${EModelEndpoint.chatGPTBrowser}`,
-  [EModelEndpoint.assistants]: '/api/assistants/chat',
+  [EModelEndpoint.azureAssistants]: '/api/assistants/v1/chat',
+  [EModelEndpoint.assistants]: '/api/assistants/v2/chat',
 };
 
 export const modularEndpoints = new Set<EModelEndpoint | string>([
@@ -456,10 +460,12 @@ export const supportsBalanceCheck = {
   [EModelEndpoint.anthropic]: true,
   [EModelEndpoint.gptPlugins]: true,
   [EModelEndpoint.assistants]: true,
+  [EModelEndpoint.azureAssistants]: true,
   [EModelEndpoint.azureOpenAI]: true,
 };
 
 export const visionModels = [
+  'gpt-4o',
   'gpt-4-turbo',
   'gpt-4-vision',
   'llava',
@@ -677,7 +683,7 @@ export enum Constants {
   /** Key for the app's version. */
   VERSION = 'v0.7.2',
   /** Key for the Custom Config's version (librechat.yaml). */
-  CONFIG_VERSION = '1.0.9',
+  CONFIG_VERSION = '1.1.1',
   /** Standard value for the first message's `parentMessageId` value, to indicate no parent exists. */
   NO_PARENT = '00000000-0000-0000-0000-000000000000',
   /** Fixed, encoded domain length for Azure OpenAI Assistants Function name parsing. */

package/src/createPayload.ts

@@ -1,5 +1,5 @@
 import type { TSubmission, TMessage, TEndpointOption } from './types';
-import { tConvoUpdateSchema, EModelEndpoint } from './schemas';
+import { tConvoUpdateSchema, EModelEndpoint, isAssistantsEndpoint } from './schemas';
 import { EndpointURLs } from './config';
 
 export default function createPayload(submission: TSubmission) {
@@ -12,7 +12,7 @@ export default function createPayload(submission: TSubmission) {
 
   let server = EndpointURLs[endpointType ?? endpoint];
 
-  if (isEdited && endpoint === EModelEndpoint.assistants) {
+  if (isEdited && isAssistantsEndpoint(endpoint)) {
     server += '/modify';
   } else if (isEdited) {
     server = server.replace('/ask/', '/edit/');

package/src/data-service.ts

@@ -31,6 +31,34 @@ export function getMessagesByConvoId(conversationId: string): Promise<s.TMessage
   return request.get(endpoints.messages(conversationId));
 }
 
+export function getSharedMessages(shareId: string): Promise<t.TSharedMessagesResponse> {
+  return request.get(endpoints.shareMessages(shareId));
+}
+
+export const listSharedLinks = (
+  params?: q.SharedLinkListParams,
+): Promise<q.SharedLinksResponse> => {
+  const pageNumber = params?.pageNumber || '1'; // Default to page 1 if not provided
+  const isPublic = params?.isPublic || true; // Default to true if not provided
+  return request.get(endpoints.getSharedLinks(pageNumber, isPublic));
+};
+
+export function getSharedLink(shareId: string): Promise<t.TSharedLinkResponse> {
+  return request.get(endpoints.shareMessages(shareId));
+}
+
+export function createSharedLink(payload: t.TSharedLinkRequest): Promise<t.TSharedLinkResponse> {
+  return request.post(endpoints.createSharedLink, payload);
+}
+
+export function updateSharedLink(payload: t.TSharedLinkRequest): Promise<t.TSharedLinkResponse> {
+  return request.patch(endpoints.updateSharedLink, payload);
+}
+
+export function deleteSharedLink(shareId: string): Promise<t.TDeleteSharedLinkResponse> {
+  return request.delete(endpoints.shareMessages(shareId));
+}
+
 export function updateMessage(payload: t.TUpdateMessageRequest): Promise<unknown> {
   const { conversationId, messageId, text } = payload;
   if (!conversationId) {
@@ -138,39 +166,105 @@ export const getEndpointsConfigOverride = (): Promise<unknown | boolean> => {
 
 /* Assistants */
 
-export const createAssistant = (data: a.AssistantCreateParams): Promise<a.Assistant> => {
-  return request.post(endpoints.assistants(), data);
-};
-
-export const getAssistantById = (assistant_id: string): Promise<a.Assistant> => {
-  return request.get(endpoints.assistants(assistant_id));
+export const createAssistant = ({
+  version,
+  ...data
+}: a.AssistantCreateParams): Promise<a.Assistant> => {
+  return request.post(endpoints.assistants({ version }), data);
+};
+
+export const getAssistantById = ({
+  endpoint,
+  assistant_id,
+  version,
+}: {
+  endpoint: s.AssistantsEndpoint;
+  assistant_id: string;
+  version: number | string | number;
+}): Promise<a.Assistant> => {
+  return request.get(
+    endpoints.assistants({
+      path: assistant_id,
+      endpoint,
+      version,
+    }),
+  );
 };
 
-export const updateAssistant = (
-  assistant_id: string,
-  data: a.AssistantUpdateParams,
-): Promise<a.Assistant> => {
-  return request.patch(endpoints.assistants(assistant_id), data);
+export const updateAssistant = ({
+  assistant_id,
+  data,
+  version,
+}: {
+  assistant_id: string;
+  data: a.AssistantUpdateParams;
+  version: number | string;
+}): Promise<a.Assistant> => {
+  return request.patch(
+    endpoints.assistants({
+      path: assistant_id,
+      version,
+    }),
+    data,
+  );
 };
 
-export const deleteAssistant = (assistant_id: string, model: string): Promise<void> => {
-  return request.delete(endpoints.assistants(assistant_id, { model }));
+export const deleteAssistant = ({
+  assistant_id,
+  model,
+  endpoint,
+  version,
+}: m.DeleteAssistantBody & { version: number | string }): Promise<void> => {
+  return request.delete(
+    endpoints.assistants({
+      path: assistant_id,
+      options: { model, endpoint },
+      version,
+    }),
+  );
 };
 
 export const listAssistants = (
-  params?: a.AssistantListParams,
+  params: a.AssistantListParams,
+  version: number | string,
 ): Promise<a.AssistantListResponse> => {
-  return request.get(endpoints.assistants(), { params });
+  return request.get(
+    endpoints.assistants({
+      version,
+      options: params,
+    }),
+  );
 };
 
-export function getAssistantDocs(): Promise<a.AssistantDocument[]> {
-  return request.get(endpoints.assistants('documents'));
+export function getAssistantDocs({
+  endpoint,
+  version,
+}: {
+  endpoint: s.AssistantsEndpoint;
+  version: number | string;
+}): Promise<a.AssistantDocument[]> {
+  return request.get(
+    endpoints.assistants({
+      path: 'documents',
+      version,
+      endpoint,
+    }),
+  );
 }
 
 /* Tools */
 
-export const getAvailableTools = (): Promise<s.TPlugin[]> => {
-  return request.get(`${endpoints.assistants()}/tools`);
+export const getAvailableTools = (
+  version: number | string,
+  endpoint: s.AssistantsEndpoint,
+): Promise<s.TPlugin[]> => {
+  return request.get(
+    endpoints.assistants({
+      path: 'tools',
+      endpoint,
+      version,
+    }),
+  );
 };
 
 /* Files */
@@ -219,7 +313,11 @@ export const uploadAvatar = (data: FormData): Promise<f.AvatarUploadResponse> =>
 
 export const uploadAssistantAvatar = (data: m.AssistantAvatarVariables): Promise<a.Assistant> => {
   return request.postMultiPart(
-    endpoints.assistants(`avatar/${data.assistant_id}`, { model: data.model }),
+    endpoints.assistants({
+      path: `avatar/${data.assistant_id}`,
+      options: { model: data.model, endpoint: data.endpoint },
+      version: data.version,
+    }),
     data.formData,
   );
 };
@@ -236,28 +334,55 @@ export const getFileDownload = async (userId: string, file_id: string): Promise<
 export const deleteFiles = async (
   files: f.BatchFile[],
   assistant_id?: string,
+  tool_resource?: a.EToolResources,
 ): Promise<f.DeleteFilesResponse> =>
   request.deleteWithOptions(endpoints.files(), {
-    data: { files, assistant_id },
+    data: { files, assistant_id, tool_resource },
   });
 
 /* actions */
 
 export const updateAction = (data: m.UpdateActionVariables): Promise<m.UpdateActionResponse> => {
-  const { assistant_id, ...body } = data;
-  return request.post(endpoints.assistants(`actions/${assistant_id}`), body);
+  const { assistant_id, version, ...body } = data;
+  return request.post(
+    endpoints.assistants({
+      path: `actions/${assistant_id}`,
+      version,
+    }),
+    body,
+  );
 };
 
-export function getActions(): Promise<a.Action[]> {
-  return request.get(endpoints.assistants('actions'));
+export function getActions({
+  endpoint,
+  version,
+}: {
+  endpoint: s.AssistantsEndpoint;
+  version: number | string;
+}): Promise<a.Action[]> {
+  return request.get(
+    endpoints.assistants({
+      path: 'actions',
+      version,
+      endpoint,
+    }),
+  );
 }
 
-export const deleteAction = async (
-  assistant_id: string,
-  action_id: string,
-  model: string,
-): Promise<void> =>
-  request.delete(endpoints.assistants(`actions/${assistant_id}/${action_id}/${model}`));
+export const deleteAction = async ({
+  assistant_id,
+  action_id,
+  model,
+  version,
+  endpoint,
+}: m.DeleteActionVariables & { version: number | string }): Promise<void> =>
+  request.delete(
+    endpoints.assistants({
+      path: `actions/${assistant_id}/${action_id}/${model}`,
+      version,
+      endpoint,
+    }),
+  );
 
 /* conversations */
 

package/src/file-config.ts

@@ -7,6 +7,7 @@ export const supportsFiles = {
   [EModelEndpoint.openAI]: true,
   [EModelEndpoint.google]: true,
   [EModelEndpoint.assistants]: true,
+  [EModelEndpoint.azureAssistants]: true,
   [EModelEndpoint.azureOpenAI]: true,
   [EModelEndpoint.anthropic]: true,
   [EModelEndpoint.custom]: true,
@@ -152,24 +153,28 @@ export const megabyte = 1024 * 1024;
 /** Helper function to get megabytes value */
 export const mbToBytes = (mb: number): number => mb * megabyte;
 
+const defaultSizeLimit = mbToBytes(512);
+const assistantsFileConfig = {
+  fileLimit: 10,
+  fileSizeLimit: defaultSizeLimit,
+  totalSizeLimit: defaultSizeLimit,
+  supportedMimeTypes,
+  disabled: false,
+};
+
 export const fileConfig = {
   endpoints: {
-    [EModelEndpoint.assistants]: {
-      fileLimit: 10,
-      fileSizeLimit: mbToBytes(512),
-      totalSizeLimit: mbToBytes(512),
-      supportedMimeTypes,
-      disabled: false,
-    },
+    [EModelEndpoint.assistants]: assistantsFileConfig,
+    [EModelEndpoint.azureAssistants]: assistantsFileConfig,
     default: {
       fileLimit: 10,
-      fileSizeLimit: mbToBytes(512),
-      totalSizeLimit: mbToBytes(512),
+      fileSizeLimit: defaultSizeLimit,
+      totalSizeLimit: defaultSizeLimit,
       supportedMimeTypes,
       disabled: false,
     },
   },
-  serverFileSizeLimit: mbToBytes(512),
+  serverFileSizeLimit: defaultSizeLimit,
   avatarSizeLimit: mbToBytes(2),
   checkType: function (fileType: string, supportedTypes: RegExp[] = supportedMimeTypes) {
     return supportedTypes.some((regex) => regex.test(fileType));

package/src/keys.ts

@@ -1,5 +1,7 @@
 export enum QueryKeys {
   messages = 'messages',
+  sharedMessages = 'sharedMessages',
+  sharedLinks = 'sharedLinks',
   allConversations = 'allConversations',
   archivedConversations = 'archivedConversations',
   searchConversations = 'searchConversations',

package/src/parsers.ts

@@ -38,6 +38,7 @@ const endpointSchemas: Record<EModelEndpoint, EndpointSchema> = {
   [EModelEndpoint.chatGPTBrowser]: chatGPTBrowserSchema,
   [EModelEndpoint.gptPlugins]: gptPluginsSchema,
   [EModelEndpoint.assistants]: assistantSchema,
+  [EModelEndpoint.azureAssistants]: assistantSchema,
 };
 
 // const schemaCreators: Record<EModelEndpoint, (customSchema: DefaultSchemaValues) => EndpointSchema> = {
@@ -49,6 +50,7 @@ export function getEnabledEndpoints() {
   const defaultEndpoints: string[] = [
     EModelEndpoint.openAI,
     EModelEndpoint.assistants,
+    EModelEndpoint.azureAssistants,
     EModelEndpoint.azureOpenAI,
     EModelEndpoint.google,
     EModelEndpoint.bingAI,
@@ -273,6 +275,7 @@ const compactEndpointSchemas: Record<string, CompactEndpointSchema> = {
   [EModelEndpoint.azureOpenAI]: compactOpenAISchema,
   [EModelEndpoint.custom]: compactOpenAISchema,
   [EModelEndpoint.assistants]: compactAssistantSchema,
+  [EModelEndpoint.azureAssistants]: compactAssistantSchema,
   [EModelEndpoint.google]: compactGoogleSchema,
   /* BingAI needs all fields */
   [EModelEndpoint.bingAI]: bingAISchema,

package/src/react-query/react-query-service.ts

@@ -1,13 +1,11 @@
-import {
+import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
+import type {
   UseQueryOptions,
-  useQuery,
-  useMutation,
-  useQueryClient,
   UseMutationResult,
   QueryObserverResult,
 } from '@tanstack/react-query';
-import { defaultOrderQuery } from '../types/assistants';
 import { initialModelsConfig, LocalStorageKeys } from '../config';
+import { defaultOrderQuery } from '../types/assistants';
 import * as dataService from '../data-service';
 import * as m from '../types/mutations';
 import { QueryKeys } from '../keys';
@@ -60,6 +58,22 @@ export const useGetMessagesByConvoId = <TData = s.TMessage[]>(
   );
 };
 
+export const useGetSharedMessages = (
+  shareId: string,
+  config?: UseQueryOptions<t.TSharedMessagesResponse>,
+): QueryObserverResult<t.TSharedMessagesResponse> => {
+  return useQuery<t.TSharedMessagesResponse>(
+    [QueryKeys.sharedMessages, shareId],
+    () => dataService.getSharedMessages(shareId),
+    {
+      refetchOnWindowFocus: false,
+      refetchOnReconnect: false,
+      refetchOnMount: false,
+      ...config,
+    },
+  );
+};
+
 export const useGetUserBalance = (
   config?: UseQueryOptions<string>,
 ): QueryObserverResult<string> => {
@@ -138,8 +152,8 @@ export const useRevokeUserKeyMutation = (name: string): UseMutationResult<unknow
   return useMutation(() => dataService.revokeUserKey(name), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.name, name]);
-      if (name === s.EModelEndpoint.assistants) {
-        queryClient.invalidateQueries([QueryKeys.assistants, defaultOrderQuery]);
+      if (s.isAssistantsEndpoint(name)) {
+        queryClient.invalidateQueries([QueryKeys.assistants, name, defaultOrderQuery]);
         queryClient.invalidateQueries([QueryKeys.assistantDocs]);
         queryClient.invalidateQueries([QueryKeys.assistants]);
         queryClient.invalidateQueries([QueryKeys.assistant]);
@@ -155,7 +169,16 @@ export const useRevokeAllUserKeysMutation = (): UseMutationResult<unknown> => {
   return useMutation(() => dataService.revokeAllUserKeys(), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.name]);
-      queryClient.invalidateQueries([QueryKeys.assistants, defaultOrderQuery]);
+      queryClient.invalidateQueries([
+        QueryKeys.assistants,
+        s.EModelEndpoint.assistants,
+        defaultOrderQuery,
+      ]);
+      queryClient.invalidateQueries([
+        QueryKeys.assistants,
+        s.EModelEndpoint.azureAssistants,
+        defaultOrderQuery,
+      ]);
       queryClient.invalidateQueries([QueryKeys.assistantDocs]);
       queryClient.invalidateQueries([QueryKeys.assistants]);
       queryClient.invalidateQueries([QueryKeys.assistant]);
@@ -398,7 +421,9 @@ export const useUpdateUserPluginsMutation = (): UseMutationResult<
   });
 };
 
-export const useGetStartupConfig = (): QueryObserverResult<t.TStartupConfig> => {
+export const useGetStartupConfig = (
+  config?: UseQueryOptions<t.TStartupConfig>,
+): QueryObserverResult<t.TStartupConfig> => {
   return useQuery<t.TStartupConfig>(
     [QueryKeys.startupConfig],
     () => dataService.getStartupConfig(),
@@ -406,6 +431,7 @@ export const useGetStartupConfig = (): QueryObserverResult<t.TStartupConfig> =>
       refetchOnWindowFocus: false,
       refetchOnReconnect: false,
       refetchOnMount: false,
+      ...config,
     },
   );
 };