librechat-data-provider 0.7.41 → 0.7.52

package/src/schemas.ts

@@ -8,7 +8,7 @@ export const isUUID = z.string().uuid();
 export enum AuthType {
   OVERRIDE_AUTH = 'override_auth',
   USER_PROVIDED = 'user_provided',
-  SYSTEM_DEFINED = 'SYSTEM_DEFINED',
+  SYSTEM_DEFINED = 'system_defined',
 }
 
 export const authTypeSchema = z.nativeEnum(AuthType);
@@ -23,18 +23,79 @@ export enum EModelEndpoint {
   anthropic = 'anthropic',
   assistants = 'assistants',
   azureAssistants = 'azureAssistants',
+  agents = 'agents',
   custom = 'custom',
+  bedrock = 'bedrock',
 }
 
+export const paramEndpoints = new Set<EModelEndpoint | string>([
+  EModelEndpoint.agents,
+  EModelEndpoint.openAI,
+  EModelEndpoint.bedrock,
+  EModelEndpoint.azureOpenAI,
+  EModelEndpoint.anthropic,
+  EModelEndpoint.custom,
+]);
+
+export enum BedrockProviders {
+  AI21 = 'ai21',
+  Amazon = 'amazon',
+  Anthropic = 'anthropic',
+  Cohere = 'cohere',
+  Meta = 'meta',
+  MistralAI = 'mistral',
+  StabilityAI = 'stability',
+}
+
+export const getModelKey = (endpoint: EModelEndpoint | string, model: string) => {
+  if (endpoint === EModelEndpoint.bedrock) {
+    return model.split('.')[0] as BedrockProviders;
+  }
+  return model;
+};
+
+export const getSettingsKeys = (endpoint: EModelEndpoint | string, model: string) => {
+  const endpointKey = endpoint;
+  const modelKey = getModelKey(endpointKey, model);
+  const combinedKey = `${endpointKey}-${modelKey}`;
+  return [combinedKey, endpointKey];
+};
+
 export type AssistantsEndpoint = EModelEndpoint.assistants | EModelEndpoint.azureAssistants;
 
-export const isAssistantsEndpoint = (endpoint?: AssistantsEndpoint | null | string): boolean => {
+export const isAssistantsEndpoint = (_endpoint?: AssistantsEndpoint | null | string): boolean => {
+  const endpoint = _endpoint ?? '';
   if (!endpoint) {
     return false;
   }
   return endpoint.toLowerCase().endsWith(EModelEndpoint.assistants);
 };
 
+export type AgentProvider = Exclude<keyof typeof EModelEndpoint, EModelEndpoint.agents> | string;
+
+export const isAgentsEndpoint = (_endpoint?: EModelEndpoint.agents | null | string): boolean => {
+  const endpoint = _endpoint ?? '';
+  if (!endpoint) {
+    return false;
+  }
+  return endpoint === EModelEndpoint.agents;
+};
+
+export const isParamEndpoint = (
+  endpoint: EModelEndpoint | string,
+  endpointType?: EModelEndpoint | string,
+): boolean => {
+  if (paramEndpoints.has(endpoint)) {
+    return true;
+  }
+
+  if (endpointType != null) {
+    return paramEndpoints.has(endpointType);
+  }
+
+  return false;
+};
+
 export enum ImageDetail {
   low = 'low',
   auto = 'auto',
@@ -61,6 +122,7 @@ export const defaultAssistantFormValues = {
   name: '',
   description: '',
   instructions: '',
+  conversation_starters: [],
   model: '',
   functions: [],
   code_interpreter: false,
@@ -68,6 +130,22 @@ export const defaultAssistantFormValues = {
   retrieval: false,
 };
 
+export const defaultAgentFormValues = {
+  agent: {},
+  id: '',
+  name: '',
+  description: '',
+  instructions: '',
+  model: '',
+  model_parameters: {},
+  tools: [],
+  provider: {},
+  projectIds: [],
+  isCollaborative: false,
+  [Tools.execute_code]: false,
+  [Tools.file_search]: false,
+};
+
 export const ImageVisionTool: FunctionTool = {
   type: Tools.function,
   [Tools.function]: {
@@ -82,7 +160,7 @@ export const ImageVisionTool: FunctionTool = {
 };
 
 export const isImageVisionTool = (tool: FunctionTool | FunctionToolCall) =>
-  tool.type === 'function' && tool.function?.name === ImageVisionTool?.function?.name;
+  tool.type === 'function' && tool.function?.name === ImageVisionTool.function?.name;
 
 export const openAISettings = {
   model: {
@@ -90,7 +168,7 @@ export const openAISettings = {
   },
   temperature: {
     min: 0,
-    max: 1,
+    max: 2,
     step: 0.01,
     default: 1,
   },
@@ -123,6 +201,9 @@ export const openAISettings = {
   },
   imageDetail: {
     default: ImageDetail.auto,
+    min: 0,
+    max: 2,
+    step: 1,
   },
 };
 
@@ -151,7 +232,7 @@ export const googleSettings = {
   topK: {
     min: 1,
     max: 40,
-    step: 0.01,
+    step: 1,
     default: 40,
   },
 };
@@ -160,7 +241,7 @@ const ANTHROPIC_MAX_OUTPUT = 8192;
 const LEGACY_ANTHROPIC_MAX_OUTPUT = 4096;
 export const anthropicSettings = {
   model: {
-    default: 'claude-3-5-sonnet-20240620',
+    default: 'claude-3-5-sonnet-20241022',
   },
   temperature: {
     min: 0,
@@ -168,6 +249,9 @@ export const anthropicSettings = {
     step: 0.01,
     default: 1,
   },
+  promptCache: {
+    default: true,
+  },
   maxOutputTokens: {
     min: 1,
     max: ANTHROPIC_MAX_OUTPUT,
@@ -216,10 +300,54 @@ export const anthropicSettings = {
   },
 };
 
+export const agentsSettings = {
+  model: {
+    default: 'gpt-3.5-turbo-test',
+  },
+  temperature: {
+    min: 0,
+    max: 1,
+    step: 0.01,
+    default: 1,
+  },
+  top_p: {
+    min: 0,
+    max: 1,
+    step: 0.01,
+    default: 1,
+  },
+  presence_penalty: {
+    min: 0,
+    max: 2,
+    step: 0.01,
+    default: 0,
+  },
+  frequency_penalty: {
+    min: 0,
+    max: 2,
+    step: 0.01,
+    default: 0,
+  },
+  resendFiles: {
+    default: true,
+  },
+  maxContextTokens: {
+    default: undefined,
+  },
+  max_tokens: {
+    default: undefined,
+  },
+  imageDetail: {
+    default: ImageDetail.auto,
+  },
+};
+
 export const endpointSettings = {
   [EModelEndpoint.openAI]: openAISettings,
   [EModelEndpoint.google]: googleSettings,
   [EModelEndpoint.anthropic]: anthropicSettings,
+  [EModelEndpoint.agents]: agentsSettings,
+  [EModelEndpoint.bedrock]: agentsSettings,
 };
 
 const google = endpointSettings[EModelEndpoint.google];
@@ -241,7 +369,7 @@ export const tPluginSchema = z.object({
   pluginKey: z.string(),
   description: z.string(),
   icon: z.string(),
-  authConfig: z.array(tPluginAuthConfigSchema),
+  authConfig: z.array(tPluginAuthConfigSchema).optional(),
   authenticated: z.boolean().optional(),
   isButton: z.boolean().optional(),
 });
@@ -341,6 +469,13 @@ export const tMessageSchema = z.object({
   iconURL: z.string().optional(),
 });
 
+export type TAttachmentMetadata = { messageId: string; toolCallId: string };
+export type TAttachment =
+  | (TFile & TAttachmentMetadata)
+  | (Pick<TFile, 'filename' | 'filepath' | 'conversationId'> & {
+      expiresAt: number;
+    } & TAttachmentMetadata);
+
 export type TMessage = z.input<typeof tMessageSchema> & {
   children?: TMessage[];
   plugin?: TResPlugin | null;
@@ -349,6 +484,7 @@ export type TMessage = z.input<typeof tMessageSchema> & {
   files?: Partial<TFile>[];
   depth?: number;
   siblingIndex?: number;
+  attachments?: TAttachment[];
 };
 
 export const coerceNumber = z.union([z.number(), z.string()]).transform((val) => {
@@ -358,44 +494,76 @@ export const coerceNumber = z.union([z.number(), z.string()]).transform((val) =>
   return val;
 });
 
+type DocumentTypeValue =
+  | null
+  | boolean
+  | number
+  | string
+  | DocumentTypeValue[]
+  | { [key: string]: DocumentTypeValue };
+
+const DocumentType: z.ZodType<DocumentTypeValue> = z.lazy(() =>
+  z.union([
+    z.null(),
+    z.boolean(),
+    z.number(),
+    z.string(),
+    z.array(z.lazy(() => DocumentType)),
+    z.record(z.lazy(() => DocumentType)),
+  ]),
+);
+
 export const tConversationSchema = z.object({
   conversationId: z.string().nullable(),
-  title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
-  user: z.string().optional(),
   endpoint: eModelEndpointSchema.nullable(),
   endpointType: eModelEndpointSchema.optional(),
-  suggestions: z.array(z.string()).optional(),
+  isArchived: z.boolean().optional(),
+  title: z.string().nullable().or(z.literal('New Chat')).default('New Chat'),
+  user: z.string().optional(),
   messages: z.array(z.string()).optional(),
   tools: z.union([z.array(tPluginSchema), z.array(z.string())]).optional(),
-  createdAt: z.string(),
-  updatedAt: z.string(),
   modelLabel: z.string().nullable().optional(),
-  examples: z.array(tExampleSchema).optional(),
-  tags: z.array(z.string()).optional(),
-  /* Prefer modelLabel over chatGptLabel */
-  chatGptLabel: z.string().nullable().optional(),
   userLabel: z.string().optional(),
   model: z.string().nullable().optional(),
   promptPrefix: z.string().nullable().optional(),
   temperature: z.number().optional(),
   topP: z.number().optional(),
   topK: z.number().optional(),
-  context: z.string().nullable().optional(),
   top_p: z.number().optional(),
   frequency_penalty: z.number().optional(),
   presence_penalty: z.number().optional(),
   parentMessageId: z.string().optional(),
-  maxOutputTokens: z.number().optional(),
-  agentOptions: tAgentOptionsSchema.nullable().optional(),
-  file_ids: z.array(z.string()).optional(),
+  maxOutputTokens: coerceNumber.optional(),
   maxContextTokens: coerceNumber.optional(),
   max_tokens: coerceNumber.optional(),
+  /* Anthropic */
+  promptCache: z.boolean().optional(),
+  system: z.string().optional(),
+  /* artifacts */
+  artifacts: z.string().optional(),
+  /* google */
+  context: z.string().nullable().optional(),
+  examples: z.array(tExampleSchema).optional(),
+  /* DB */
+  tags: z.array(z.string()).optional(),
+  createdAt: z.string(),
+  updatedAt: z.string(),
+  /* Files */
+  file_ids: z.array(z.string()).optional(),
   /* vision */
   resendFiles: z.boolean().optional(),
   imageDetail: eImageDetailSchema.optional(),
   /* assistant */
   assistant_id: z.string().optional(),
+  /* agents */
+  agent_id: z.string().optional(),
+  /* AWS Bedrock */
+  region: z.string().optional(),
+  maxTokens: coerceNumber.optional(),
+  additionalModelRequestFields: DocumentType.optional(),
+  /* assistant + agents */
   instructions: z.string().optional(),
+  additional_instructions: z.string().optional(),
   /** Used to overwrite active conversation settings when saving a Preset */
   presetOverride: z.record(z.unknown()).optional(),
   stop: z.array(z.string()).optional(),
@@ -407,6 +575,8 @@ export const tConversationSchema = z.object({
   Deprecated fields
   */
   /** @deprecated */
+  suggestions: z.array(z.string()).optional(),
+  /** @deprecated */
   systemMessage: z.string().nullable().optional(),
   /** @deprecated */
   jailbreak: z.boolean().optional(),
@@ -422,6 +592,10 @@ export const tConversationSchema = z.object({
   toneStyle: z.string().nullable().optional(),
   /** @deprecated */
   resendImages: z.boolean().optional(),
+  /** @deprecated */
+  agentOptions: tAgentOptionsSchema.nullable().optional(),
+  /** @deprecated Prefer `modelLabel` over `chatGptLabel` */
+  chatGptLabel: z.string().nullable().optional(),
 });
 
 export const tPresetSchema = tConversationSchema
@@ -478,6 +652,7 @@ export const tSharedLinkSchema = z.object({
 export type TSharedLink = z.infer<typeof tSharedLinkSchema>;
 
 export const tConversationTagSchema = z.object({
+  _id: z.string(),
   user: z.string(),
   tag: z.string(),
   description: z.string().optional(),
@@ -488,70 +663,6 @@ export const tConversationTagSchema = z.object({
 });
 export type TConversationTag = z.infer<typeof tConversationTagSchema>;
 
-export const openAISchema = tConversationSchema
-  .pick({
-    model: true,
-    modelLabel: true,
-    chatGptLabel: true,
-    promptPrefix: true,
-    temperature: true,
-    top_p: true,
-    presence_penalty: true,
-    frequency_penalty: true,
-    resendFiles: true,
-    imageDetail: true,
-    stop: true,
-    iconURL: true,
-    greeting: true,
-    spec: true,
-    maxContextTokens: true,
-    max_tokens: true,
-  })
-  .transform((obj) => {
-    const result = {
-      ...obj,
-      model: obj.model ?? openAISettings.model.default,
-      chatGptLabel: obj.chatGptLabel ?? obj.modelLabel ?? null,
-      promptPrefix: obj.promptPrefix ?? null,
-      temperature: obj.temperature ?? openAISettings.temperature.default,
-      top_p: obj.top_p ?? openAISettings.top_p.default,
-      presence_penalty: obj.presence_penalty ?? openAISettings.presence_penalty.default,
-      frequency_penalty: obj.frequency_penalty ?? openAISettings.frequency_penalty.default,
-      resendFiles:
-        typeof obj.resendFiles === 'boolean' ? obj.resendFiles : openAISettings.resendFiles.default,
-      imageDetail: obj.imageDetail ?? openAISettings.imageDetail.default,
-      stop: obj.stop ?? undefined,
-      iconURL: obj.iconURL ?? undefined,
-      greeting: obj.greeting ?? undefined,
-      spec: obj.spec ?? undefined,
-      maxContextTokens: obj.maxContextTokens ?? undefined,
-      max_tokens: obj.max_tokens ?? undefined,
-    };
-
-    if (obj.modelLabel) {
-      result.modelLabel = null;
-    }
-
-    return result;
-  })
-  .catch(() => ({
-    model: openAISettings.model.default,
-    chatGptLabel: null,
-    promptPrefix: null,
-    temperature: openAISettings.temperature.default,
-    top_p: openAISettings.top_p.default,
-    presence_penalty: openAISettings.presence_penalty.default,
-    frequency_penalty: openAISettings.frequency_penalty.default,
-    resendFiles: openAISettings.resendFiles.default,
-    imageDetail: openAISettings.imageDetail.default,
-    stop: undefined,
-    iconURL: undefined,
-    greeting: undefined,
-    spec: undefined,
-    maxContextTokens: undefined,
-    max_tokens: undefined,
-  }));
-
 export const googleSchema = tConversationSchema
   .pick({
     model: true,
@@ -560,6 +671,7 @@ export const googleSchema = tConversationSchema
     examples: true,
     temperature: true,
     maxOutputTokens: true,
+    artifacts: true,
     topP: true,
     topK: true,
     iconURL: true,
@@ -634,57 +746,6 @@ export const bingAISchema = tConversationSchema
     invocationId: 1,
   }));
 
-export const anthropicSchema = tConversationSchema
-  .pick({
-    model: true,
-    modelLabel: true,
-    promptPrefix: true,
-    temperature: true,
-    maxOutputTokens: true,
-    topP: true,
-    topK: true,
-    resendFiles: true,
-    iconURL: true,
-    greeting: true,
-    spec: true,
-    maxContextTokens: true,
-  })
-  .transform((obj) => {
-    const model = obj.model ?? anthropicSettings.model.default;
-    return {
-      ...obj,
-      model,
-      modelLabel: obj.modelLabel ?? null,
-      promptPrefix: obj.promptPrefix ?? null,
-      temperature: obj.temperature ?? anthropicSettings.temperature.default,
-      maxOutputTokens: obj.maxOutputTokens ?? anthropicSettings.maxOutputTokens.reset(model),
-      topP: obj.topP ?? anthropicSettings.topP.default,
-      topK: obj.topK ?? anthropicSettings.topK.default,
-      resendFiles:
-        typeof obj.resendFiles === 'boolean'
-          ? obj.resendFiles
-          : anthropicSettings.resendFiles.default,
-      iconURL: obj.iconURL ?? undefined,
-      greeting: obj.greeting ?? undefined,
-      spec: obj.spec ?? undefined,
-      maxContextTokens: obj.maxContextTokens ?? anthropicSettings.maxContextTokens.default,
-    };
-  })
-  .catch(() => ({
-    model: anthropicSettings.model.default,
-    modelLabel: null,
-    promptPrefix: null,
-    temperature: anthropicSettings.temperature.default,
-    maxOutputTokens: anthropicSettings.maxOutputTokens.default,
-    topP: anthropicSettings.topP.default,
-    topK: anthropicSettings.topK.default,
-    resendFiles: anthropicSettings.resendFiles.default,
-    iconURL: undefined,
-    greeting: undefined,
-    spec: undefined,
-    maxContextTokens: anthropicSettings.maxContextTokens.default,
-  }));
-
 export const chatGPTBrowserSchema = tConversationSchema
   .pick({
     model: true,
@@ -704,6 +765,7 @@ export const gptPluginsSchema = tConversationSchema
     chatGptLabel: true,
     promptPrefix: true,
     temperature: true,
+    artifacts: true,
     top_p: true,
     presence_penalty: true,
     frequency_penalty: true,
@@ -737,7 +799,7 @@ export const gptPluginsSchema = tConversationSchema
       maxContextTokens: obj.maxContextTokens ?? undefined,
     };
 
-    if (obj.modelLabel) {
+    if (obj.modelLabel != null && obj.modelLabel !== '') {
       result.modelLabel = null;
     }
 
@@ -764,16 +826,17 @@ export const gptPluginsSchema = tConversationSchema
     maxContextTokens: undefined,
   }));
 
-export function removeNullishValues<T extends object>(obj: T): T {
+export function removeNullishValues<T extends Record<string, unknown>>(obj: T): Partial<T> {
   const newObj: Partial<T> = { ...obj };
 
   (Object.keys(newObj) as Array<keyof T>).forEach((key) => {
-    if (newObj[key] === undefined || newObj[key] === null || newObj[key] === '') {
+    const value = newObj[key];
+    if (value === undefined || value === null) {
       delete newObj[key];
     }
   });
 
-  return newObj as T;
+  return newObj;
 }
 
 export const assistantSchema = tConversationSchema
@@ -781,6 +844,7 @@ export const assistantSchema = tConversationSchema
     model: true,
     assistant_id: true,
     instructions: true,
+    artifacts: true,
     promptPrefix: true,
     iconURL: true,
     greeting: true,
@@ -812,6 +876,7 @@ export const compactAssistantSchema = tConversationSchema
     assistant_id: true,
     instructions: true,
     promptPrefix: true,
+    artifacts: true,
     iconURL: true,
     greeting: true,
     spec: true,
@@ -820,9 +885,62 @@ export const compactAssistantSchema = tConversationSchema
   .transform(removeNullishValues)
   .catch(() => ({}));
 
-export const compactOpenAISchema = tConversationSchema
+export const agentsSchema = tConversationSchema
+  .pick({
+    model: true,
+    modelLabel: true,
+    temperature: true,
+    top_p: true,
+    presence_penalty: true,
+    frequency_penalty: true,
+    resendFiles: true,
+    imageDetail: true,
+    agent_id: true,
+    instructions: true,
+    promptPrefix: true,
+    iconURL: true,
+    greeting: true,
+    maxContextTokens: true,
+  })
+  .transform((obj) => ({
+    ...obj,
+    model: obj.model ?? agentsSettings.model.default,
+    modelLabel: obj.modelLabel ?? null,
+    temperature: obj.temperature ?? 1,
+    top_p: obj.top_p ?? 1,
+    presence_penalty: obj.presence_penalty ?? 0,
+    frequency_penalty: obj.frequency_penalty ?? 0,
+    resendFiles:
+      typeof obj.resendFiles === 'boolean' ? obj.resendFiles : agentsSettings.resendFiles.default,
+    imageDetail: obj.imageDetail ?? ImageDetail.auto,
+    agent_id: obj.agent_id ?? undefined,
+    instructions: obj.instructions ?? undefined,
+    promptPrefix: obj.promptPrefix ?? null,
+    iconURL: obj.iconURL ?? undefined,
+    greeting: obj.greeting ?? undefined,
+    maxContextTokens: obj.maxContextTokens ?? undefined,
+  }))
+  .catch(() => ({
+    model: agentsSettings.model.default,
+    modelLabel: null,
+    temperature: 1,
+    top_p: 1,
+    presence_penalty: 0,
+    frequency_penalty: 0,
+    resendFiles: agentsSettings.resendFiles.default,
+    imageDetail: ImageDetail.auto,
+    agent_id: undefined,
+    instructions: undefined,
+    promptPrefix: null,
+    iconURL: undefined,
+    greeting: undefined,
+    maxContextTokens: undefined,
+  }));
+
+export const openAISchema = tConversationSchema
   .pick({
     model: true,
+    modelLabel: true,
     chatGptLabel: true,
     promptPrefix: true,
     temperature: true,
@@ -830,6 +948,7 @@ export const compactOpenAISchema = tConversationSchema
     presence_penalty: true,
     frequency_penalty: true,
     resendFiles: true,
+    artifacts: true,
     imageDetail: true,
     stop: true,
     iconURL: true,
@@ -838,29 +957,7 @@ export const compactOpenAISchema = tConversationSchema
     maxContextTokens: true,
     max_tokens: true,
   })
-  .transform((obj: Partial<TConversation>) => {
-    const newObj: Partial<TConversation> = { ...obj };
-    if (newObj.temperature === 1) {
-      delete newObj.temperature;
-    }
-    if (newObj.top_p === 1) {
-      delete newObj.top_p;
-    }
-    if (newObj.presence_penalty === 0) {
-      delete newObj.presence_penalty;
-    }
-    if (newObj.frequency_penalty === 0) {
-      delete newObj.frequency_penalty;
-    }
-    if (newObj.resendFiles === true) {
-      delete newObj.resendFiles;
-    }
-    if (newObj.imageDetail === ImageDetail.auto) {
-      delete newObj.imageDetail;
-    }
-
-    return removeNullishValues(newObj);
-  })
+  .transform((obj: Partial<TConversation>) => removeNullishValues(obj))
   .catch(() => ({}));
 
 export const compactGoogleSchema = tConversationSchema
@@ -871,6 +968,7 @@ export const compactGoogleSchema = tConversationSchema
     examples: true,
     temperature: true,
     maxOutputTokens: true,
+    artifacts: true,
     topP: true,
     topK: true,
     iconURL: true,
@@ -897,7 +995,7 @@ export const compactGoogleSchema = tConversationSchema
   })
   .catch(() => ({}));
 
-export const compactAnthropicSchema = tConversationSchema
+export const anthropicSchema = tConversationSchema
   .pick({
     model: true,
     modelLabel: true,
@@ -907,31 +1005,14 @@ export const compactAnthropicSchema = tConversationSchema
     topP: true,
     topK: true,
     resendFiles: true,
+    promptCache: true,
+    artifacts: true,
     iconURL: true,
     greeting: true,
     spec: true,
     maxContextTokens: true,
   })
-  .transform((obj) => {
-    const newObj: Partial<TConversation> = { ...obj };
-    if (newObj.temperature === anthropicSettings.temperature.default) {
-      delete newObj.temperature;
-    }
-    if (newObj.maxOutputTokens === anthropicSettings.legacy.maxOutputTokens.default) {
-      delete newObj.maxOutputTokens;
-    }
-    if (newObj.topP === anthropicSettings.topP.default) {
-      delete newObj.topP;
-    }
-    if (newObj.topK === anthropicSettings.topK.default) {
-      delete newObj.topK;
-    }
-    if (newObj.resendFiles === anthropicSettings.resendFiles.default) {
-      delete newObj.resendFiles;
-    }
-
-    return removeNullishValues(newObj);
-  })
+  .transform((obj) => removeNullishValues(obj))
   .catch(() => ({}));
 
 export const compactChatGPTSchema = tConversationSchema
@@ -947,6 +1028,7 @@ export const compactChatGPTSchema = tConversationSchema
 export const compactPluginsSchema = tConversationSchema
   .pick({
     model: true,
+    modelLabel: true,
     chatGptLabel: true,
     promptPrefix: true,
     temperature: true,
@@ -962,6 +1044,9 @@ export const compactPluginsSchema = tConversationSchema
   })
   .transform((obj) => {
     const newObj: Partial<TConversation> = { ...obj };
+    if (newObj.modelLabel === null) {
+      delete newObj.modelLabel;
+    }
     if (newObj.chatGptLabel === null) {
       delete newObj.chatGptLabel;
     }
@@ -997,3 +1082,27 @@ export const compactPluginsSchema = tConversationSchema
     return removeNullishValues(newObj);
   })
   .catch(() => ({}));
+
+const tBannerSchema = z.object({
+  bannerId: z.string(),
+  message: z.string(),
+  displayFrom: z.string(),
+  displayTo: z.string(),
+  createdAt: z.string(),
+  updatedAt: z.string(),
+  isPublic: z.boolean(),
+});
+export type TBanner = z.infer<typeof tBannerSchema>;
+
+export const compactAgentsSchema = tConversationSchema
+  .pick({
+    model: true,
+    agent_id: true,
+    instructions: true,
+    additional_instructions: true,
+    iconURL: true,
+    greeting: true,
+    spec: true,
+  })
+  .transform(removeNullishValues)
+  .catch(() => ({}));