librechat-data-provider 0.4.0 → 0.4.1

package/src/config.ts

@@ -1,9 +1,36 @@
+/* eslint-disable max-len */
 import { z } from 'zod';
 import { EModelEndpoint, eModelEndpointSchema } from './schemas';
+import { fileConfigSchema } from './file-config';
 import { FileSources } from './types/files';
 
 export const fileSourceSchema = z.nativeEnum(FileSources);
 
+export const assistantEndpointSchema = z.object({
+  /* assistants specific */
+  disableBuilder: z.boolean().optional(),
+  pollIntervalMs: z.number().optional(),
+  timeoutMs: z.number().optional(),
+  supportedIds: z.array(z.string()).min(1).optional(),
+  excludedIds: z.array(z.string()).min(1).optional(),
+  /* general */
+  apiKey: z.string().optional(),
+  baseURL: z.string().optional(),
+  models: z
+    .object({
+      default: z.array(z.string()).min(1),
+      fetch: z.boolean().optional(),
+      userIdQuery: z.boolean().optional(),
+    })
+    .optional(),
+  titleConvo: z.boolean().optional(),
+  titleMethod: z.union([z.literal('completion'), z.literal('functions')]).optional(),
+  titleModel: z.string().optional(),
+  headers: z.record(z.any()).optional(),
+});
+
+export type TAssistantEndpoint = z.infer<typeof assistantEndpointSchema>;
+
 export const endpointSchema = z.object({
   name: z.string().refine((value) => !eModelEndpointSchema.safeParse(value).success, {
     message: `Value cannot be one of the default endpoint (EModelEndpoint) values: ${Object.values(
@@ -25,6 +52,19 @@ export const endpointSchema = z.object({
   forcePrompt: z.boolean().optional(),
   modelDisplayLabel: z.string().optional(),
   headers: z.record(z.any()).optional(),
+  addParams: z.record(z.any()).optional(),
+  dropParams: z.array(z.string()).optional(),
+});
+
+export const rateLimitSchema = z.object({
+  fileUploads: z
+    .object({
+      ipMax: z.number().optional(),
+      ipWindowInMinutes: z.number().optional(),
+      userMax: z.number().optional(),
+      userWindowInMinutes: z.number().optional(),
+    })
+    .optional(),
 });
 
 export const configSchema = z.object({
@@ -37,11 +77,17 @@ export const configSchema = z.object({
       allowedDomains: z.array(z.string()).optional(),
     })
     .optional(),
+  rateLimits: rateLimitSchema.optional(),
+  fileConfig: fileConfigSchema.optional(),
   endpoints: z
     .object({
-      custom: z.array(endpointSchema.partial()),
+      [EModelEndpoint.assistants]: assistantEndpointSchema.optional(),
+      custom: z.array(endpointSchema.partial()).optional(),
     })
     .strict()
+    .refine((data) => Object.keys(data).length > 0, {
+      message: 'At least one `endpoints` field must be provided.',
+    })
     .optional(),
 });
 
@@ -54,7 +100,7 @@ export enum KnownEndpoints {
 
 export const defaultEndpoints: EModelEndpoint[] = [
   EModelEndpoint.openAI,
-  EModelEndpoint.assistant,
+  EModelEndpoint.assistants,
   EModelEndpoint.azureOpenAI,
   EModelEndpoint.bingAI,
   EModelEndpoint.chatGPTBrowser,
@@ -66,7 +112,7 @@ export const defaultEndpoints: EModelEndpoint[] = [
 
 export const alternateName = {
   [EModelEndpoint.openAI]: 'OpenAI',
-  [EModelEndpoint.assistant]: 'Assistants',
+  [EModelEndpoint.assistants]: 'Assistants',
   [EModelEndpoint.azureOpenAI]: 'Azure OpenAI',
   [EModelEndpoint.bingAI]: 'Bing',
   [EModelEndpoint.chatGPTBrowser]: 'ChatGPT',
@@ -77,6 +123,21 @@ export const alternateName = {
 };
 
 export const defaultModels = {
+  [EModelEndpoint.assistants]: [
+    'gpt-3.5-turbo-0125',
+    '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-3.5-turbo',
+    'gpt-4',
+    'gpt-4-0314',
+    'gpt-4-32k-0314',
+    'gpt-4-0613',
+    'gpt-3.5-turbo-0613',
+  ],
   [EModelEndpoint.google]: [
     'gemini-pro',
     'gemini-pro-vision',
@@ -121,6 +182,14 @@ export const defaultModels = {
   ],
 };
 
+export const supportsRetrieval = new Set([
+  'gpt-3.5-turbo-0125',
+  'gpt-4-0125-preview',
+  'gpt-4-turbo-preview',
+  'gpt-4-1106-preview',
+  'gpt-3.5-turbo-1106',
+]);
+
 export const EndpointURLs: { [key in EModelEndpoint]: string } = {
   [EModelEndpoint.openAI]: `/api/ask/${EModelEndpoint.openAI}`,
   [EModelEndpoint.bingAI]: `/api/ask/${EModelEndpoint.bingAI}`,
@@ -130,7 +199,7 @@ 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.assistant]: '/api/assistants/chat',
+  [EModelEndpoint.assistants]: '/api/assistants/chat',
 };
 
 export const modularEndpoints = new Set<EModelEndpoint | string>([
@@ -142,14 +211,6 @@ export const modularEndpoints = new Set<EModelEndpoint | string>([
   EModelEndpoint.custom,
 ]);
 
-export const supportsFiles = {
-  [EModelEndpoint.openAI]: true,
-  [EModelEndpoint.google]: true,
-  [EModelEndpoint.assistant]: true,
-  [EModelEndpoint.azureOpenAI]: true,
-  [EModelEndpoint.custom]: true,
-};
-
 export const supportsBalanceCheck = {
   [EModelEndpoint.openAI]: true,
   [EModelEndpoint.azureOpenAI]: true,
@@ -159,6 +220,19 @@ export const supportsBalanceCheck = {
 
 export const visionModels = ['gpt-4-vision', 'llava-13b', 'gemini-pro-vision'];
 
+export function validateVisionModel(
+  model: string | undefined,
+  additionalModels: string[] | undefined = [],
+) {
+  if (!model) {
+    return false;
+  }
+
+  return visionModels.concat(additionalModels).some((visionModel) => model.includes(visionModel));
+}
+
+export const imageGenTools = new Set(['dalle', 'dall-e', 'stable-diffusion']);
+
 /**
  * Enum for cache keys.
  */
@@ -175,6 +249,11 @@ export enum CacheKeys {
    * Key for the title generation cache.
    */
   GEN_TITLE = 'genTitle',
+  /**
+  /**
+   * Key for the tools cache.
+   */
+  TOOLS = 'tools',
   /**
    * Key for the model config cache.
    */
@@ -195,10 +274,18 @@ export enum CacheKeys {
    * Key for the custom config cache.
    */
   CUSTOM_CONFIG = 'customConfig',
+  /**
+   * Key for accessing Abort Keys
+   */
+  ABORT_KEYS = 'abortKeys',
   /**
    * Key for the override config cache.
    */
   OVERRIDE_CONFIG = 'overrideConfig',
+  /**
+   * Key for accessing File Upload Violations (exceeding limit).
+   */
+  FILE_UPLOAD_LIMIT = 'file_upload_limit',
 }
 
 /**
@@ -260,3 +347,27 @@ export enum SettingsTabValues {
    */
   ACCOUNT = 'account',
 }
+
+/**
+ * Enum for app-wide constants
+ */
+export enum Constants {
+  /**
+   * Key for the app's version.
+   */
+  VERSION = 'v0.6.9',
+  /**
+   * Key for the Custom Config's version (librechat.yaml).
+   */
+  CONFIG_VERSION = '1.0.3',
+  /**
+   * Standard value for the first message's `parentMessageId` value, to indicate no parent exists.
+   */
+  NO_PARENT = '00000000-0000-0000-0000-000000000000',
+}
+
+export const defaultOrderQuery: {
+  order: 'asc';
+} = {
+  order: 'asc',
+};

package/src/createPayload.ts

@@ -12,7 +12,7 @@ export default function createPayload(submission: TSubmission) {
 
   let server = EndpointURLs[endpointType ?? endpoint];
 
-  if (isEdited && endpoint === EModelEndpoint.assistant) {
+  if (isEdited && endpoint === EModelEndpoint.assistants) {
     server += '/modify';
   } else if (isEdited) {
     server = server.replace('/ask/', '/edit/');
@@ -32,9 +32,5 @@ export default function createPayload(submission: TSubmission) {
     conversationId,
   };
 
-  if (endpoint === EModelEndpoint.assistant) {
-    payload.messages = messages;
-  }
-
   return { server, payload };
 }

package/src/data-service.ts

@@ -196,23 +196,57 @@ export const listAssistants = (
   return request.get(endpoints.assistants(), { params });
 };
 
+/* Tools */
+
+export const getAvailableTools = (): Promise<s.TPlugin[]> => {
+  return request.get(`${endpoints.assistants()}/tools`);
+};
+
 /* Files */
 
 export const getFiles = (): Promise<f.TFile[]> => {
   return request.get(endpoints.files());
 };
 
+export const getFileConfig = (): Promise<f.FileConfig> => {
+  return request.get(`${endpoints.files()}/config`);
+};
+
 export const uploadImage = (data: FormData): Promise<f.TFileUpload> => {
   return request.postMultiPart(endpoints.images(), data);
 };
 
+export const uploadFile = (data: FormData): Promise<f.TFileUpload> => {
+  return request.postMultiPart(endpoints.files(), data);
+};
+
 export const uploadAvatar = (data: FormData): Promise<f.AvatarUploadResponse> => {
   return request.postMultiPart(endpoints.avatar(), data);
 };
 
-export const deleteFiles = async (files: f.BatchFile[]): Promise<f.DeleteFilesResponse> =>
+export const uploadAssistantAvatar = (data: m.AssistantAvatarVariables): Promise<a.Assistant> => {
+  return request.postMultiPart(endpoints.assistants(`avatar/${data.assistant_id}`), data.formData);
+};
+
+export const updateAction = (data: m.UpdateActionVariables): Promise<m.UpdateActionResponse> => {
+  const { assistant_id, ...body } = data;
+  return request.post(endpoints.assistants(`actions/${assistant_id}`), body);
+};
+
+export function getActions(): Promise<a.Action[]> {
+  return request.get(endpoints.assistants('actions'));
+}
+
+export function getAssistantDocs(): Promise<a.AssistantDocument[]> {
+  return request.get(endpoints.assistants('documents'));
+}
+
+export const deleteFiles = async (
+  files: f.BatchFile[],
+  assistant_id?: string,
+): Promise<f.DeleteFilesResponse> =>
   request.deleteWithOptions(endpoints.files(), {
-    data: { files },
+    data: { files, assistant_id },
   });
 
 /* conversations */
@@ -237,3 +271,6 @@ export const listConversationsByQuery = (
     return request.get(endpoints.conversations(pageNumber));
   }
 };
+
+export const deleteAction = async (assistant_id: string, action_id: string): Promise<void> =>
+  request.delete(endpoints.assistants(`actions/${assistant_id}/${action_id}`));

package/src/file-config.ts

@@ -0,0 +1,264 @@
+/* eslint-disable max-len */
+import { z } from 'zod';
+import { EModelEndpoint } from './schemas';
+import type { FileConfig, EndpointFileConfig } from './types/files';
+
+export const supportsFiles = {
+  [EModelEndpoint.openAI]: true,
+  [EModelEndpoint.google]: true,
+  [EModelEndpoint.assistants]: true,
+  [EModelEndpoint.azureOpenAI]: true,
+  [EModelEndpoint.custom]: true,
+};
+
+export const excelFileTypes = [
+  'application/vnd.ms-excel',
+  'application/msexcel',
+  'application/x-msexcel',
+  'application/x-ms-excel',
+  'application/x-excel',
+  'application/x-dos_ms_excel',
+  'application/xls',
+  'application/x-xls',
+  'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet',
+];
+
+export const fullMimeTypesList = [
+  'text/x-c',
+  'text/x-c++',
+  'application/csv',
+  'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+  'text/html',
+  'text/x-java',
+  'application/json',
+  'text/markdown',
+  'application/pdf',
+  'text/x-php',
+  'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+  'text/x-python',
+  'text/x-script.python',
+  'text/x-ruby',
+  'text/x-tex',
+  'text/plain',
+  'text/css',
+  'image/jpeg',
+  'text/javascript',
+  'image/gif',
+  'image/png',
+  'application/x-tar',
+  'application/typescript',
+  'application/xml',
+  'application/zip',
+  ...excelFileTypes,
+];
+
+export const codeInterpreterMimeTypesList = [
+  'text/x-c',
+  'text/x-c++',
+  'application/csv',
+  'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+  'text/html',
+  'text/x-java',
+  'application/json',
+  'text/markdown',
+  'application/pdf',
+  'text/x-php',
+  'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+  'text/x-python',
+  'text/x-script.python',
+  'text/x-ruby',
+  'text/x-tex',
+  'text/plain',
+  'text/css',
+  'image/jpeg',
+  'text/javascript',
+  'image/gif',
+  'image/png',
+  'application/x-tar',
+  'application/typescript',
+  'application/xml',
+  'application/zip',
+  ...excelFileTypes,
+];
+
+export const retrievalMimeTypesList = [
+  'text/x-c',
+  'text/x-c++',
+  'application/vnd.openxmlformats-officedocument.wordprocessingml.document',
+  'text/html',
+  'text/x-java',
+  'application/json',
+  'text/markdown',
+  'application/pdf',
+  'text/x-php',
+  'application/vnd.openxmlformats-officedocument.presentationml.presentation',
+  'text/x-python',
+  'text/x-script.python',
+  'text/x-ruby',
+  'text/x-tex',
+  'text/plain',
+];
+
+export const imageExtRegex = /\.(jpg|jpeg|png|gif|webp)$/i;
+
+export const excelMimeTypes =
+  /^application\/(vnd\.ms-excel|msexcel|x-msexcel|x-ms-excel|x-excel|x-dos_ms_excel|xls|x-xls|vnd\.openxmlformats-officedocument\.spreadsheetml\.sheet)$/;
+
+export const textMimeTypes =
+  /^(text\/(x-c|x-c\+\+|x-java|html|markdown|x-php|x-python|x-script\.python|x-ruby|x-tex|plain|css|javascript|csv))$/;
+
+export const applicationMimeTypes =
+  /^(application\/(csv|json|pdf|x-tar|typescript|vnd\.openxmlformats-officedocument\.(wordprocessingml\.document|presentationml\.presentation|spreadsheetml\.sheet)|xml|zip))$/;
+
+export const imageMimeTypes = /^image\/(jpeg|gif|png|webp)$/;
+
+export const supportedMimeTypes = [
+  textMimeTypes,
+  excelMimeTypes,
+  applicationMimeTypes,
+  imageMimeTypes,
+];
+
+export const codeInterpreterMimeTypes = [
+  textMimeTypes,
+  excelMimeTypes,
+  applicationMimeTypes,
+  imageMimeTypes,
+];
+
+export const retrievalMimeTypes = [
+  /^(text\/(x-c|x-c\+\+|html|x-java|markdown|x-php|x-python|x-script\.python|x-ruby|x-tex|plain))$/,
+  /^(application\/(json|pdf|vnd\.openxmlformats-officedocument\.(wordprocessingml\.document|presentationml\.presentation)))$/,
+];
+
+export const megabyte = 1024 * 1024;
+/** Helper function to get megabytes value */
+export const mbToBytes = (mb: number): number => mb * megabyte;
+
+export const fileConfig = {
+  endpoints: {
+    [EModelEndpoint.assistants]: {
+      fileLimit: 10,
+      fileSizeLimit: mbToBytes(512),
+      totalSizeLimit: mbToBytes(512),
+      supportedMimeTypes,
+      disabled: false,
+    },
+    default: {
+      fileLimit: 10,
+      fileSizeLimit: mbToBytes(20),
+      totalSizeLimit: mbToBytes(25),
+      supportedMimeTypes: [imageMimeTypes],
+      disabled: false,
+    },
+  },
+  serverFileSizeLimit: mbToBytes(512),
+  avatarSizeLimit: mbToBytes(2),
+  checkType: function (fileType: string, supportedTypes: RegExp[] = supportedMimeTypes) {
+    return supportedTypes.some((regex) => regex.test(fileType));
+  },
+};
+
+const supportedMimeTypesSchema = z
+  .array(z.any())
+  .optional()
+  .refine(
+    (mimeTypes) => {
+      if (!mimeTypes) {
+        return true;
+      }
+      return mimeTypes.every(
+        (mimeType) => mimeType instanceof RegExp || typeof mimeType === 'string',
+      );
+    },
+    {
+      message: 'Each mimeType must be a string or a RegExp object.',
+    },
+  );
+
+export const endpointFileConfigSchema = z.object({
+  disabled: z.boolean().optional(),
+  fileLimit: z.number().min(0).optional(),
+  fileSizeLimit: z.number().min(0).optional(),
+  totalSizeLimit: z.number().min(0).optional(),
+  supportedMimeTypes: supportedMimeTypesSchema.optional(),
+});
+
+export const fileConfigSchema = z.object({
+  endpoints: z.record(endpointFileConfigSchema).optional(),
+  serverFileSizeLimit: z.number().min(0).optional(),
+  avatarSizeLimit: z.number().min(0).optional(),
+});
+
+/** Helper function to safely convert string patterns to RegExp objects */
+export const convertStringsToRegex = (patterns: string[]): RegExp[] =>
+  patterns.reduce((acc: RegExp[], pattern) => {
+    try {
+      const regex = new RegExp(pattern);
+      acc.push(regex);
+    } catch (error) {
+      console.error(`Invalid regex pattern "${pattern}" skipped.`);
+    }
+    return acc;
+  }, []);
+
+export function mergeFileConfig(dynamic: z.infer<typeof fileConfigSchema> | undefined): FileConfig {
+  const mergedConfig = fileConfig as FileConfig;
+  if (!dynamic) {
+    return mergedConfig;
+  }
+
+  if (dynamic.serverFileSizeLimit !== undefined) {
+    mergedConfig.serverFileSizeLimit = mbToBytes(dynamic.serverFileSizeLimit);
+  }
+
+  if (dynamic.avatarSizeLimit !== undefined) {
+    mergedConfig.avatarSizeLimit = mbToBytes(dynamic.avatarSizeLimit);
+  }
+
+  if (!dynamic.endpoints) {
+    return mergedConfig;
+  }
+
+  for (const key in dynamic.endpoints) {
+    const dynamicEndpoint = (dynamic.endpoints as Record<string, EndpointFileConfig>)[key];
+
+    if (!mergedConfig.endpoints[key]) {
+      mergedConfig.endpoints[key] = {};
+    }
+
+    const mergedEndpoint = mergedConfig.endpoints[key];
+
+    if (dynamicEndpoint.disabled === true) {
+      mergedEndpoint.disabled = true;
+      mergedEndpoint.fileLimit = 0;
+      mergedEndpoint.fileSizeLimit = 0;
+      mergedEndpoint.totalSizeLimit = 0;
+      mergedEndpoint.supportedMimeTypes = [];
+      continue;
+    }
+
+    if (dynamicEndpoint.fileSizeLimit !== undefined) {
+      mergedEndpoint.fileSizeLimit = mbToBytes(dynamicEndpoint.fileSizeLimit);
+    }
+
+    if (dynamicEndpoint.totalSizeLimit !== undefined) {
+      mergedEndpoint.totalSizeLimit = mbToBytes(dynamicEndpoint.totalSizeLimit);
+    }
+
+    const configKeys = ['fileLimit'] as const;
+    configKeys.forEach((field) => {
+      if (dynamicEndpoint[field] !== undefined) {
+        mergedEndpoint[field] = dynamicEndpoint[field];
+      }
+    });
+
+    if (dynamicEndpoint.supportedMimeTypes) {
+      mergedEndpoint.supportedMimeTypes = convertStringsToRegex(
+        dynamicEndpoint.supportedMimeTypes as unknown as string[],
+      );
+    }
+  }
+
+  return mergedConfig;
+}

package/src/index.ts

@@ -1,5 +1,6 @@
 /* config */
 export * from './config';
+export * from './file-config';
 /* schema helpers  */
 export * from './parsers';
 /* types (exports schemas from `./types` as they contain needed in other defs) */
@@ -17,4 +18,5 @@ import * as dataService from './data-service';
 export { dataService };
 /* general helpers */
 export * from './sse';
+export * from './actions';
 export { default as createPayload } from './createPayload';

package/src/keys.ts

@@ -18,13 +18,20 @@ export enum QueryKeys {
   assistant = 'assistant',
   endpointsConfigOverride = 'endpointsConfigOverride',
   files = 'files',
+  fileConfig = 'fileConfig',
+  tools = 'tools',
+  actions = 'actions',
+  assistantDocs = 'assistantDocs',
 }
 
 export enum MutationKeys {
-  imageUpload = 'imageUpload',
+  fileUpload = 'fileUpload',
   fileDelete = 'fileDelete',
   updatePreset = 'updatePreset',
   deletePreset = 'deletePreset',
   logoutUser = 'logoutUser',
   avatarUpload = 'avatarUpload',
+  assistantAvatarUpload = 'assistantAvatarUpload',
+  updateAction = 'updateAction',
+  deleteAction = 'deleteAction',
 }

package/src/parsers.ts

@@ -35,7 +35,7 @@ const endpointSchemas: Record<EModelEndpoint, EndpointSchema> = {
   [EModelEndpoint.anthropic]: anthropicSchema,
   [EModelEndpoint.chatGPTBrowser]: chatGPTBrowserSchema,
   [EModelEndpoint.gptPlugins]: gptPluginsSchema,
-  [EModelEndpoint.assistant]: assistantSchema,
+  [EModelEndpoint.assistants]: assistantSchema,
 };
 
 // const schemaCreators: Record<EModelEndpoint, (customSchema: DefaultSchemaValues) => EndpointSchema> = {
@@ -172,16 +172,16 @@ type CompactEndpointSchema =
   | typeof compactPluginsSchema;
 
 const compactEndpointSchemas: Record<string, CompactEndpointSchema> = {
-  openAI: compactOpenAISchema,
-  azureOpenAI: compactOpenAISchema,
-  custom: compactOpenAISchema,
-  assistant: assistantSchema,
-  google: compactGoogleSchema,
+  [EModelEndpoint.openAI]: compactOpenAISchema,
+  [EModelEndpoint.azureOpenAI]: compactOpenAISchema,
+  [EModelEndpoint.custom]: compactOpenAISchema,
+  [EModelEndpoint.assistants]: assistantSchema,
+  [EModelEndpoint.google]: compactGoogleSchema,
   /* BingAI needs all fields */
-  bingAI: bingAISchema,
-  anthropic: compactAnthropicSchema,
-  chatGPTBrowser: compactChatGPTSchema,
-  gptPlugins: compactPluginsSchema,
+  [EModelEndpoint.bingAI]: bingAISchema,
+  [EModelEndpoint.anthropic]: compactAnthropicSchema,
+  [EModelEndpoint.chatGPTBrowser]: compactChatGPTSchema,
+  [EModelEndpoint.gptPlugins]: compactPluginsSchema,
 };
 
 export const parseCompactConvo = ({

package/src/react-query/index.ts

@@ -1,2 +1 @@
 export * from './react-query-service';
-export * from './assistants';

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

@@ -9,6 +9,7 @@ import {
 import * as t from '../types';
 import * as s from '../schemas';
 import * as m from '../types/mutations';
+import { defaultOrderQuery } from '../config';
 import * as dataService from '../data-service';
 import request from '../request';
 import { QueryKeys } from '../keys';
@@ -136,6 +137,14 @@ export const useRevokeUserKeyMutation = (name: string): UseMutationResult<unknow
   return useMutation(() => dataService.revokeUserKey(name), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.name]);
+      if (name === s.EModelEndpoint.assistants) {
+        queryClient.invalidateQueries([QueryKeys.assistants, defaultOrderQuery]);
+        queryClient.invalidateQueries([QueryKeys.assistantDocs]);
+        queryClient.invalidateQueries([QueryKeys.assistants]);
+        queryClient.invalidateQueries([QueryKeys.assistant]);
+        queryClient.invalidateQueries([QueryKeys.actions]);
+        queryClient.invalidateQueries([QueryKeys.tools]);
+      }
     },
   });
 };
@@ -145,6 +154,12 @@ export const useRevokeAllUserKeysMutation = (): UseMutationResult<unknown> => {
   return useMutation(() => dataService.revokeAllUserKeys(), {
     onSuccess: () => {
       queryClient.invalidateQueries([QueryKeys.name]);
+      queryClient.invalidateQueries([QueryKeys.assistants, defaultOrderQuery]);
+      queryClient.invalidateQueries([QueryKeys.assistantDocs]);
+      queryClient.invalidateQueries([QueryKeys.assistants]);
+      queryClient.invalidateQueries([QueryKeys.assistant]);
+      queryClient.invalidateQueries([QueryKeys.actions]);
+      queryClient.invalidateQueries([QueryKeys.tools]);
     },
   });
 };
@@ -277,7 +292,7 @@ export const useLoginUserMutation = (): UseMutationResult<
       localStorage.removeItem('lastSelectedModel');
       localStorage.removeItem('lastSelectedTools');
       localStorage.removeItem('filesToDelete');
-      localStorage.removeItem('lastAssistant');
+      // localStorage.removeItem('lastAssistant');
     },
   });
 };