librechat-data-provider 0.6.0 → 0.6.2

package/package.json

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

package/src/api-endpoints.ts

@@ -21,8 +21,8 @@ export const abortRequest = (endpoint: string) => `/api/ask/${endpoint}/abort`;
 
 export const conversationsRoot = '/api/convos';
 
-export const conversations = (pageNumber: string) =>
-  `${conversationsRoot}?pageNumber=${pageNumber}`;
+export const conversations = (pageNumber: string, isArchived?: boolean) =>
+  `${conversationsRoot}?pageNumber=${pageNumber}${isArchived ? '&isArchived=true' : ''}`;
 
 export const conversationById = (id: string) => `${conversationsRoot}/${id}`;
 
@@ -34,6 +34,8 @@ export const deleteConversation = () => `${conversationsRoot}/clear`;
 
 export const importConversation = () => `${conversationsRoot}/import`;
 
+export const forkConversation = () => `${conversationsRoot}/fork`;
+
 export const importConversationJobStatus = (jobId: string) =>
   `${conversationsRoot}/import/jobs/${jobId}`;
 

package/src/config.ts

@@ -306,6 +306,7 @@ export enum KnownEndpoints {
   cohere = 'cohere',
   fireworks = 'fireworks',
   groq = 'groq',
+  huggingface = 'huggingface',
   mistral = 'mistral',
   mlx = 'mlx',
   ollama = 'ollama',
@@ -459,13 +460,13 @@ export const supportsBalanceCheck = {
 };
 
 export const visionModels = [
+  'gpt-4-turbo',
   'gpt-4-vision',
   'llava',
   'llava-13b',
   'gemini-pro-vision',
   'claude-3',
   'gemini-1.5',
-  'gpt-4-turbo',
 ];
 export enum VisionModes {
   generative = 'generative',
@@ -653,6 +654,10 @@ export enum SettingsTabValues {
    * Tab for General Settings
    */
   GENERAL = 'general',
+  /**
+   * Tab for Messages Settings
+   */
+  MESSAGES = 'messages',
   /**
    * Tab for Beta Features
    */
@@ -698,6 +703,21 @@ export enum LocalStorageKeys {
   FILES_TO_DELETE = 'filesToDelete',
   /** Prefix key for the last selected assistant ID by index */
   ASST_ID_PREFIX = 'assistant_id__',
+  /** Key for the last selected fork setting */
+  FORK_SETTING = 'forkSetting',
+  /** Key for remembering the last selected option, instead of manually selecting */
+  REMEMBER_FORK_OPTION = 'rememberForkOption',
+  /** Key for remembering the split at target fork option modifier */
+  FORK_SPLIT_AT_TARGET = 'splitAtTarget',
+}
+
+export enum ForkOptions {
+  /** Key for direct path option */
+  DIRECT_PATH = 'directPath',
+  /** Key for including branches */
+  INCLUDE_BRANCHES = 'includeBranches',
+  /** Key for target level fork (default) */
+  TARGET_LEVEL = '',
 }
 
 /**

package/src/data-service.ts

@@ -261,6 +261,10 @@ export const deleteAction = async (
 
 /* conversations */
 
+export function forkConversation(payload: t.TForkConvoRequest): Promise<t.TForkConvoResponse> {
+  return request.post(endpoints.forkConversation(), payload);
+}
+
 export function deleteConversation(payload: t.TDeleteConversationRequest) {
   //todo: this should be a DELETE request
   return request.post(endpoints.deleteConversation(), { arg: payload });
@@ -275,7 +279,8 @@ export const listConversations = (
 ): Promise<q.ConversationListResponse> => {
   // Assuming params has a pageNumber property
   const pageNumber = params?.pageNumber || '1'; // Default to page 1 if not provided
-  return request.get(endpoints.conversations(pageNumber));
+  const isArchived = params?.isArchived || false; // Default to false if not provided
+  return request.get(endpoints.conversations(pageNumber, isArchived));
 };
 
 export const listConversationsByQuery = (
@@ -312,6 +317,12 @@ export function updateConversation(
   return request.post(endpoints.updateConversation(), { arg: payload });
 }
 
+export function archiveConversation(
+  payload: t.TArchiveConversationRequest,
+): Promise<t.TArchiveConversationResponse> {
+  return request.post(endpoints.updateConversation(), { arg: payload });
+}
+
 export function genTitle(payload: m.TGenTitleRequest): Promise<m.TGenTitleResponse> {
   return request.post(endpoints.genTitle(), payload);
 }

package/src/generate.ts

@@ -72,6 +72,8 @@ export type DynamicSettingProps = Partial<SettingDefinition> & {
   setOption: TSetOption;
   conversation: TConversation | TPreset | null;
   defaultValue?: number | boolean | string | string[];
+  className?: string;
+  inputClassName?: string;
 };
 
 const requiredSettingFields = ['key', 'type', 'component'];
@@ -508,6 +510,7 @@ export const generateOpenAISchema = (customOpenAI: OpenAISettings) => {
       frequency_penalty: true,
       resendFiles: true,
       imageDetail: true,
+      maxContextTokens: true,
     })
     .transform((obj) => ({
       ...obj,
@@ -521,6 +524,7 @@ export const generateOpenAISchema = (customOpenAI: OpenAISettings) => {
       resendFiles:
         typeof obj.resendFiles === 'boolean' ? obj.resendFiles : defaults.resendFiles.default,
       imageDetail: obj.imageDetail ?? defaults.imageDetail.default,
+      maxContextTokens: obj.maxContextTokens ?? undefined,
     }))
     .catch(() => ({
       model: defaults.model.default,
@@ -532,6 +536,7 @@ export const generateOpenAISchema = (customOpenAI: OpenAISettings) => {
       frequency_penalty: defaults.frequency_penalty.default,
       resendFiles: defaults.resendFiles.default,
       imageDetail: defaults.imageDetail.default,
+      maxContextTokens: undefined,
     }));
 };
 
@@ -547,6 +552,7 @@ export const generateGoogleSchema = (customGoogle: GoogleSettings) => {
       maxOutputTokens: true,
       topP: true,
       topK: true,
+      maxContextTokens: true,
     })
     .transform((obj) => {
       const isGemini = obj?.model?.toLowerCase()?.includes('gemini');
@@ -571,6 +577,7 @@ export const generateGoogleSchema = (customGoogle: GoogleSettings) => {
         maxOutputTokens,
         topP: obj.topP ?? defaults.topP.default,
         topK: obj.topK ?? defaults.topK.default,
+        maxContextTokens: obj.maxContextTokens ?? undefined,
       };
     })
     .catch(() => ({
@@ -582,5 +589,6 @@ export const generateGoogleSchema = (customGoogle: GoogleSettings) => {
       maxOutputTokens: defaults.maxOutputTokens.default,
       topP: defaults.topP.default,
       topK: defaults.topK.default,
+      maxContextTokens: undefined,
     }));
 };

package/src/keys.ts

@@ -1,6 +1,7 @@
 export enum QueryKeys {
   messages = 'messages',
   allConversations = 'allConversations',
+  archivedConversations = 'archivedConversations',
   searchConversations = 'searchConversations',
   conversation = 'conversation',
   searchEnabled = 'searchEnabled',

package/src/schemas.ts

@@ -105,6 +105,12 @@ export const openAISettings = {
   resendFiles: {
     default: true,
   },
+  maxContextTokens: {
+    default: undefined,
+  },
+  max_tokens: {
+    default: undefined,
+  },
   imageDetail: {
     default: ImageDetail.auto,
   },
@@ -274,6 +280,13 @@ export type TMessage = z.input<typeof tMessageSchema> & {
   files?: Partial<TFile>[];
 };
 
+export const coerceNumber = z.union([z.number(), z.string()]).transform((val) => {
+  if (typeof val === 'string') {
+    return val.trim() === '' ? undefined : parseFloat(val);
+  }
+  return val;
+});
+
 export const tConversationSchema = z.object({
   conversationId: z.string().nullable(),
   title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
@@ -309,6 +322,8 @@ export const tConversationSchema = z.object({
   maxOutputTokens: z.number().optional(),
   agentOptions: tAgentOptionsSchema.nullable().optional(),
   file_ids: z.array(z.string()).optional(),
+  maxContextTokens: coerceNumber.optional(),
+  max_tokens: coerceNumber.optional(),
   /** @deprecated */
   resendImages: z.boolean().optional(),
   /* vision */
@@ -382,6 +397,8 @@ export const openAISchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
+    max_tokens: true,
   })
   .transform((obj) => ({
     ...obj,
@@ -399,6 +416,8 @@ export const openAISchema = tConversationSchema
     iconURL: obj.iconURL ?? undefined,
     greeting: obj.greeting ?? undefined,
     spec: obj.spec ?? undefined,
+    maxContextTokens: obj.maxContextTokens ?? undefined,
+    max_tokens: obj.max_tokens ?? undefined,
   }))
   .catch(() => ({
     model: openAISettings.model.default,
@@ -414,6 +433,8 @@ export const openAISchema = tConversationSchema
     iconURL: undefined,
     greeting: undefined,
     spec: undefined,
+    maxContextTokens: undefined,
+    max_tokens: undefined,
   }));
 
 export const googleSchema = tConversationSchema
@@ -429,6 +450,7 @@ export const googleSchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
   })
   .transform((obj) => {
     const isGemini = obj?.model?.toLowerCase()?.includes('gemini');
@@ -456,6 +478,7 @@ export const googleSchema = tConversationSchema
       iconURL: obj.iconURL ?? undefined,
       greeting: obj.greeting ?? undefined,
       spec: obj.spec ?? undefined,
+      maxContextTokens: obj.maxContextTokens ?? undefined,
     };
   })
   .catch(() => ({
@@ -470,6 +493,7 @@ export const googleSchema = tConversationSchema
     iconURL: undefined,
     greeting: undefined,
     spec: undefined,
+    maxContextTokens: undefined,
   }));
 
 export const bingAISchema = tConversationSchema
@@ -520,6 +544,7 @@ export const anthropicSchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
   })
   .transform((obj) => ({
     ...obj,
@@ -534,6 +559,7 @@ export const anthropicSchema = tConversationSchema
     iconURL: obj.iconURL ?? undefined,
     greeting: obj.greeting ?? undefined,
     spec: obj.spec ?? undefined,
+    maxContextTokens: obj.maxContextTokens ?? undefined,
   }))
   .catch(() => ({
     model: 'claude-1',
@@ -547,6 +573,7 @@ export const anthropicSchema = tConversationSchema
     iconURL: undefined,
     greeting: undefined,
     spec: undefined,
+    maxContextTokens: undefined,
   }));
 
 export const chatGPTBrowserSchema = tConversationSchema
@@ -576,6 +603,7 @@ export const gptPluginsSchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
   })
   .transform((obj) => ({
     ...obj,
@@ -596,6 +624,7 @@ export const gptPluginsSchema = tConversationSchema
     iconURL: obj.iconURL ?? undefined,
     greeting: obj.greeting ?? undefined,
     spec: obj.spec ?? undefined,
+    maxContextTokens: obj.maxContextTokens ?? undefined,
   }))
   .catch(() => ({
     model: 'gpt-3.5-turbo',
@@ -615,6 +644,7 @@ export const gptPluginsSchema = tConversationSchema
     iconURL: undefined,
     greeting: undefined,
     spec: undefined,
+    maxContextTokens: undefined,
   }));
 
 export function removeNullishValues<T extends object>(obj: T): T {
@@ -688,6 +718,8 @@ export const compactOpenAISchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
+    max_tokens: true,
   })
   .transform((obj: Partial<TConversation>) => {
     const newObj: Partial<TConversation> = { ...obj };
@@ -727,6 +759,7 @@ export const compactGoogleSchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
   })
   .transform((obj) => {
     const newObj: Partial<TConversation> = { ...obj };
@@ -760,6 +793,7 @@ export const compactAnthropicSchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
   })
   .transform((obj) => {
     const newObj: Partial<TConversation> = { ...obj };
@@ -807,6 +841,7 @@ export const compactPluginsSchema = tConversationSchema
     iconURL: true,
     greeting: true,
     spec: true,
+    maxContextTokens: true,
   })
   .transform((obj) => {
     const newObj: Partial<TConversation> = { ...obj };

package/src/types/mutations.ts

@@ -1,4 +1,4 @@
-import { TPreset, TDeleteConversationResponse, TDeleteConversationRequest } from '../types';
+import type * as types from '../types';
 import {
   Assistant,
   AssistantCreateParams,
@@ -9,6 +9,18 @@ import {
   Action,
 } from './assistants';
 
+export type MutationOptions<
+  Response,
+  Request,
+  Context = unknown,
+  Error = unknown,
+  Snapshot = void,
+> = {
+  onSuccess?: (data: Response, variables: Request, context?: Context) => void;
+  onMutate?: (variables: Request) => Snapshot | Promise<Snapshot>;
+  onError?: (error: Error, variables: Request, context?: Context, snapshot?: Snapshot) => void;
+};
+
 export type TGenTitleRequest = {
   conversationId: string;
 };
@@ -22,27 +34,11 @@ export type PresetDeleteResponse = {
   deletedCount: number;
 };
 
-export type UpdatePresetOptions = {
-  onSuccess?: (data: TPreset, variables: TPreset, context?: unknown) => void;
-  onMutate?: (variables: TPreset) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: TPreset, context?: unknown) => void;
-};
+export type UpdatePresetOptions = MutationOptions<types.TPreset, types.TPreset>;
 
-export type DeletePresetOptions = {
-  onSuccess?: (
-    data: PresetDeleteResponse,
-    variables: TPreset | undefined,
-    context?: unknown,
-  ) => void;
-  onMutate?: (variables: TPreset | undefined) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: TPreset | undefined, context?: unknown) => void;
-};
+export type DeletePresetOptions = MutationOptions<PresetDeleteResponse, types.TPreset | undefined>;
 
-export type LogoutOptions = {
-  onSuccess?: (data: unknown, variables: undefined, context?: unknown) => void;
-  onMutate?: (variables: undefined) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: undefined, context?: unknown) => void;
-};
+export type LogoutOptions = MutationOptions<unknown, undefined>;
 
 export type AssistantAvatarVariables = {
   assistant_id: string;
@@ -59,53 +55,26 @@ export type UpdateActionVariables = {
   model: string;
 };
 
-export type UploadAssistantAvatarOptions = {
-  onSuccess?: (data: Assistant, variables: AssistantAvatarVariables, context?: unknown) => void;
-  onMutate?: (variables: AssistantAvatarVariables) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: AssistantAvatarVariables, context?: unknown) => void;
-};
+export type UploadAssistantAvatarOptions = MutationOptions<Assistant, AssistantAvatarVariables>;
 
-export type CreateAssistantMutationOptions = {
-  onSuccess?: (data: Assistant, variables: AssistantCreateParams, context?: unknown) => void;
-  onMutate?: (variables: AssistantCreateParams) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: AssistantCreateParams, context?: unknown) => void;
-};
+export type CreateAssistantMutationOptions = MutationOptions<Assistant, AssistantCreateParams>;
 
-export type UpdateAssistantMutationOptions = {
-  onSuccess?: (
-    data: Assistant,
-    variables: { assistant_id: string; data: AssistantUpdateParams },
-    context?: unknown,
-  ) => void;
-  onMutate?: (variables: {
-    assistant_id: string;
-    data: AssistantUpdateParams;
-  }) => void | Promise<unknown>;
-  onError?: (
-    error: unknown,
-    variables: { assistant_id: string; data: AssistantUpdateParams },
-    context?: unknown,
-  ) => void;
+export type UpdateAssistantVariables = {
+  assistant_id: string;
+  data: AssistantUpdateParams;
 };
 
+export type UpdateAssistantMutationOptions = MutationOptions<Assistant, UpdateAssistantVariables>;
+
 export type DeleteAssistantBody = { assistant_id: string; model: string };
 
-export type DeleteAssistantMutationOptions = {
-  onSuccess?: (data: void, variables: { assistant_id: string }, context?: unknown) => void;
-  onMutate?: (variables: { assistant_id: string }) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: { assistant_id: string }, context?: unknown) => void;
-};
+export type DeleteAssistantMutationOptions = MutationOptions<
+  void,
+  Pick<DeleteAssistantBody, 'assistant_id'>
+>;
 
 export type UpdateActionResponse = [AssistantDocument, Assistant, Action];
-export type UpdateActionOptions = {
-  onSuccess?: (
-    data: UpdateActionResponse,
-    variables: UpdateActionVariables,
-    context?: unknown,
-  ) => void;
-  onMutate?: (variables: UpdateActionVariables) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: UpdateActionVariables, context?: unknown) => void;
-};
+export type UpdateActionOptions = MutationOptions<UpdateActionResponse, UpdateActionVariables>;
 
 export type DeleteActionVariables = {
   assistant_id: string;
@@ -113,18 +82,11 @@ export type DeleteActionVariables = {
   model: string;
 };
 
-export type DeleteActionOptions = {
-  onSuccess?: (data: void, variables: DeleteActionVariables, context?: unknown) => void;
-  onMutate?: (variables: DeleteActionVariables) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: DeleteActionVariables, context?: unknown) => void;
-};
+export type DeleteActionOptions = MutationOptions<void, DeleteActionVariables>;
 
-export type DeleteConversationOptions = {
-  onSuccess?: (
-    data: TDeleteConversationResponse,
-    variables: TDeleteConversationRequest,
-    context?: unknown,
-  ) => void;
-  onMutate?: (variables: TDeleteConversationRequest) => void | Promise<unknown>;
-  onError?: (error: unknown, variables: TDeleteConversationRequest, context?: unknown) => void;
-};
+export type DeleteConversationOptions = MutationOptions<
+  types.TDeleteConversationResponse,
+  types.TDeleteConversationRequest
+>;
+
+export type ForkConvoOptions = MutationOptions<types.TForkConvoResponse, types.TForkConvoRequest>;

package/src/types/queries.ts

@@ -16,6 +16,7 @@ export type ConversationListParams = {
   order?: 'asc' | 'desc';
   pageNumber: string; // Add this line
   conversationId?: string;
+  isArchived: boolean;
 };
 
 // Type for the response from the conversation list API

package/src/types.ts

@@ -17,6 +17,7 @@ export type TEndpointOption = {
   endpointType?: EModelEndpoint;
   modelDisplayLabel?: string;
   resendFiles?: boolean;
+  maxContextTokens?: number;
   imageDetail?: ImageDetail;
   model?: string | null;
   promptPrefix?: string;
@@ -125,6 +126,26 @@ export type TDeleteConversationResponse = {
   };
 };
 
+export type TArchiveConversationRequest = {
+  conversationId: string;
+  isArchived: boolean;
+};
+
+export type TArchiveConversationResponse = TConversation;
+
+export type TForkConvoRequest = {
+  messageId: string;
+  conversationId: string;
+  option?: string;
+  splitAtTarget?: boolean;
+  latestMessageId?: string;
+};
+
+export type TForkConvoResponse = {
+  conversation: TConversation;
+  messages: TMessage[];
+};
+
 export type TSearchResults = {
   conversations: TConversation[];
   messages: TMessage[];