📂 refactor: Cleanup File Filtering Logic, Improve Validation (#10414)

* feat: add filterFilesByEndpointConfig to filter disabled file processing by provider

* chore: explicit define of endpointFileConfig for better debugging

* refactor: move `normalizeEndpointName` to data-provider as used app-wide

* chore: remove overrideEndpoint from useFileHandling

* refactor: improve endpoint file config selection

* refactor: update filterFilesByEndpointConfig to accept structured parameters and improve endpoint file config handling

* refactor: replace defaultFileConfig with getEndpointFileConfig for improved file configuration handling across components

* test: add comprehensive unit tests for getEndpointFileConfig to validate endpoint configuration handling

* refactor: streamline agent endpoint assignment and improve file filtering logic

* feat: add error handling for disabled file uploads in endpoint configuration

* refactor: update encodeAndFormat functions to accept structured parameters for provider and endpoint

* refactor: streamline requestFiles handling in initializeAgent function

* fix: getEndpointFileConfig partial config merging scenarios

* refactor: enhance mergeWithDefault function to support document-supported providers with comprehensive MIME types

* refactor: user-configured default file config in getEndpointFileConfig

* fix: prevent file handling when endpoint is disabled and file is dragged to chat

* refactor: move `getEndpointField` to `data-provider` and update usage across components and hooks

* fix: prioritize endpointType based on agent.endpoint in file filtering logic

* fix: prioritize agent.endpoint in file filtering logic and remove unnecessary endpointType defaulting
This commit is contained in:
Danny Avila 2025-11-10 19:05:30 -05:00 committed by GitHub
parent 06c060b983
commit 2524d33362
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
62 changed files with 2352 additions and 290 deletions

View file

@ -305,11 +305,9 @@ class AnthropicClient extends BaseClient {
} }
async addImageURLs(message, attachments) { async addImageURLs(message, attachments) {
const { files, image_urls } = await encodeAndFormat( const { files, image_urls } = await encodeAndFormat(this.options.req, attachments, {
this.options.req, endpoint: EModelEndpoint.anthropic,
attachments, });
EModelEndpoint.anthropic,
);
message.image_urls = image_urls.length ? image_urls : undefined; message.image_urls = image_urls.length ? image_urls : undefined;
return files; return files;
} }

View file

@ -1213,6 +1213,7 @@ class BaseClient {
attachments, attachments,
{ {
provider: this.options.agent?.provider, provider: this.options.agent?.provider,
endpoint: this.options.agent?.endpoint,
useResponsesApi: this.options.agent?.model_parameters?.useResponsesApi, useResponsesApi: this.options.agent?.model_parameters?.useResponsesApi,
}, },
getStrategyFunctions, getStrategyFunctions,
@ -1228,7 +1229,10 @@ class BaseClient {
const videoResult = await encodeAndFormatVideos( const videoResult = await encodeAndFormatVideos(
this.options.req, this.options.req,
attachments, attachments,
this.options.agent.provider, {
provider: this.options.agent?.provider,
endpoint: this.options.agent?.endpoint,
},
getStrategyFunctions, getStrategyFunctions,
); );
message.videos = message.videos =
@ -1240,7 +1244,10 @@ class BaseClient {
const audioResult = await encodeAndFormatAudios( const audioResult = await encodeAndFormatAudios(
this.options.req, this.options.req,
attachments, attachments,
this.options.agent.provider, {
provider: this.options.agent?.provider,
endpoint: this.options.agent?.endpoint,
},
getStrategyFunctions, getStrategyFunctions,
); );
message.audios = message.audios =

View file

@ -305,7 +305,9 @@ class GoogleClient extends BaseClient {
const { files, image_urls } = await encodeAndFormat( const { files, image_urls } = await encodeAndFormat(
this.options.req, this.options.req,
attachments, attachments,
EModelEndpoint.google, {
endpoint: EModelEndpoint.google,
},
mode, mode,
); );
message.image_urls = image_urls.length ? image_urls : undefined; message.image_urls = image_urls.length ? image_urls : undefined;

View file

@ -354,11 +354,9 @@ class OpenAIClient extends BaseClient {
* @returns {Promise<MongoFile[]>} * @returns {Promise<MongoFile[]>}
*/ */
async addImageURLs(message, attachments) { async addImageURLs(message, attachments) {
const { files, image_urls } = await encodeAndFormat( const { files, image_urls } = await encodeAndFormat(this.options.req, attachments, {
this.options.req, endpoint: this.options.endpoint,
attachments, });
this.options.endpoint,
);
message.image_urls = image_urls.length ? image_urls : undefined; message.image_urls = image_urls.length ? image_urls : undefined;
return files; return files;
} }

View file

@ -210,7 +210,10 @@ class AgentClient extends BaseClient {
const { files, image_urls } = await encodeAndFormat( const { files, image_urls } = await encodeAndFormat(
this.options.req, this.options.req,
attachments, attachments,
this.options.agent.provider, {
provider: this.options.agent.provider,
endpoint: this.options.endpoint,
},
VisionModes.agents, VisionModes.agents,
); );
message.image_urls = image_urls.length ? image_urls : undefined; message.image_urls = image_urls.length ? image_urls : undefined;

View file

@ -3,7 +3,11 @@ const path = require('path');
const crypto = require('crypto'); const crypto = require('crypto');
const multer = require('multer'); const multer = require('multer');
const { sanitizeFilename } = require('@librechat/api'); const { sanitizeFilename } = require('@librechat/api');
const { fileConfig: defaultFileConfig, mergeFileConfig } = require('librechat-data-provider'); const {
mergeFileConfig,
getEndpointFileConfig,
fileConfig: defaultFileConfig,
} = require('librechat-data-provider');
const { getAppConfig } = require('~/server/services/Config'); const { getAppConfig } = require('~/server/services/Config');
const storage = multer.diskStorage({ const storage = multer.diskStorage({
@ -53,12 +57,14 @@ const createFileFilter = (customFileConfig) => {
} }
const endpoint = req.body.endpoint; const endpoint = req.body.endpoint;
const supportedTypes = const endpointType = req.body.endpointType;
customFileConfig?.endpoints?.[endpoint]?.supportedMimeTypes ?? const endpointFileConfig = getEndpointFileConfig({
customFileConfig?.endpoints?.default.supportedMimeTypes ?? fileConfig: customFileConfig,
defaultFileConfig?.endpoints?.[endpoint]?.supportedMimeTypes; endpoint,
endpointType,
});
if (!defaultFileConfig.checkType(file.mimetype, supportedTypes)) { if (!defaultFileConfig.checkType(file.mimetype, endpointFileConfig.supportedMimeTypes)) {
return cb(new Error('Unsupported file type: ' + file.mimetype), false); return cb(new Error('Unsupported file type: ' + file.mimetype), false);
} }

View file

@ -109,7 +109,7 @@ async function getEndpointsConfig(req) {
* @returns {Promise<boolean>} * @returns {Promise<boolean>}
*/ */
const checkCapability = async (req, capability) => { const checkCapability = async (req, capability) => {
const isAgents = isAgentsEndpoint(req.body?.original_endpoint || req.body?.endpoint); const isAgents = isAgentsEndpoint(req.body?.endpointType || req.body?.endpoint);
const endpointsConfig = await getEndpointsConfig(req); const endpointsConfig = await getEndpointsConfig(req);
const capabilities = const capabilities =
isAgents || endpointsConfig?.[EModelEndpoint.agents]?.capabilities != null isAgents || endpointsConfig?.[EModelEndpoint.agents]?.capabilities != null

View file

@ -1,5 +1,9 @@
const { isUserProvided, normalizeEndpointName } = require('@librechat/api'); const { isUserProvided } = require('@librechat/api');
const { EModelEndpoint, extractEnvVariable } = require('librechat-data-provider'); const {
EModelEndpoint,
extractEnvVariable,
normalizeEndpointName,
} = require('librechat-data-provider');
const { fetchModels } = require('~/server/services/ModelService'); const { fetchModels } = require('~/server/services/ModelService');
const { getAppConfig } = require('./app'); const { getAppConfig } = require('./app');

View file

@ -3,12 +3,14 @@ const {
primeResources, primeResources,
getModelMaxTokens, getModelMaxTokens,
extractLibreChatParams, extractLibreChatParams,
filterFilesByEndpointConfig,
optionalChainWithEmptyCheck, optionalChainWithEmptyCheck,
} = require('@librechat/api'); } = require('@librechat/api');
const { const {
ErrorTypes, ErrorTypes,
EModelEndpoint, EModelEndpoint,
EToolResources, EToolResources,
paramEndpoints,
isAgentsEndpoint, isAgentsEndpoint,
replaceSpecialVars, replaceSpecialVars,
providerEndpointMap, providerEndpointMap,
@ -71,6 +73,9 @@ const initializeAgent = async ({
const { resendFiles, maxContextTokens, modelOptions } = extractLibreChatParams(_modelOptions); const { resendFiles, maxContextTokens, modelOptions } = extractLibreChatParams(_modelOptions);
const provider = agent.provider;
agent.endpoint = provider;
if (isInitialAgent && conversationId != null && resendFiles) { if (isInitialAgent && conversationId != null && resendFiles) {
const fileIds = (await getConvoFiles(conversationId)) ?? []; const fileIds = (await getConvoFiles(conversationId)) ?? [];
/** @type {Set<EToolResources>} */ /** @type {Set<EToolResources>} */
@ -88,6 +93,19 @@ const initializeAgent = async ({
currentFiles = await processFiles(requestFiles); currentFiles = await processFiles(requestFiles);
} }
if (currentFiles && currentFiles.length) {
let endpointType;
if (!paramEndpoints.has(agent.endpoint)) {
endpointType = EModelEndpoint.custom;
}
currentFiles = filterFilesByEndpointConfig(req, {
files: currentFiles,
endpoint: agent.endpoint,
endpointType,
});
}
const { attachments, tool_resources } = await primeResources({ const { attachments, tool_resources } = await primeResources({
req, req,
getFiles, getFiles,
@ -98,7 +116,6 @@ const initializeAgent = async ({
requestFileSet: new Set(requestFiles?.map((file) => file.file_id)), requestFileSet: new Set(requestFiles?.map((file) => file.file_id)),
}); });
const provider = agent.provider;
const { const {
tools: structuredTools, tools: structuredTools,
toolContextMap, toolContextMap,
@ -113,7 +130,6 @@ const initializeAgent = async ({
tool_resources, tool_resources,
})) ?? {}; })) ?? {};
agent.endpoint = provider;
const { getOptions, overrideProvider } = getProviderConfig({ provider, appConfig }); const { getOptions, overrideProvider } = getProviderConfig({ provider, appConfig });
if (overrideProvider !== agent.provider) { if (overrideProvider !== agent.provider) {
agent.provider = overrideProvider; agent.provider = overrideProvider;

View file

@ -84,11 +84,15 @@ const blobStorageSources = new Set([FileSources.azure_blob, FileSources.s3]);
* Encodes and formats the given files. * Encodes and formats the given files.
* @param {ServerRequest} req - The request object. * @param {ServerRequest} req - The request object.
* @param {Array<MongoFile>} files - The array of files to encode and format. * @param {Array<MongoFile>} files - The array of files to encode and format.
* @param {EModelEndpoint} [endpoint] - Optional: The endpoint for the image. * @param {object} params - Object containing provider/endpoint information
* @param {Providers | EModelEndpoint | string} [params.provider] - The provider for the image
* @param {string} [params.endpoint] - Optional: The endpoint for the image
* @param {string} [mode] - Optional: The endpoint mode for the image. * @param {string} [mode] - Optional: The endpoint mode for the image.
* @returns {Promise<{ files: MongoFile[]; image_urls: MessageContentImageUrl[] }>} - A promise that resolves to the result object containing the encoded images and file details. * @returns {Promise<{ files: MongoFile[]; image_urls: MessageContentImageUrl[] }>} - A promise that resolves to the result object containing the encoded images and file details.
*/ */
async function encodeAndFormat(req, files, endpoint, mode) { async function encodeAndFormat(req, files, params, mode) {
const { provider, endpoint } = params;
const effectiveEndpoint = endpoint ?? provider;
const promises = []; const promises = [];
/** @type {Record<FileSources, Pick<ReturnType<typeof getStrategyFunctions>, 'prepareImagePayload' | 'getDownloadStream'>>} */ /** @type {Record<FileSources, Pick<ReturnType<typeof getStrategyFunctions>, 'prepareImagePayload' | 'getDownloadStream'>>} */
const encodingMethods = {}; const encodingMethods = {};
@ -134,7 +138,7 @@ async function encodeAndFormat(req, files, endpoint, mode) {
} catch (error) { } catch (error) {
logger.error('Error processing image from blob storage:', error); logger.error('Error processing image from blob storage:', error);
} }
} else if (source !== FileSources.local && base64Only.has(endpoint)) { } else if (source !== FileSources.local && base64Only.has(effectiveEndpoint)) {
const [_file, imageURL] = await preparePayload(req, file); const [_file, imageURL] = await preparePayload(req, file);
promises.push([_file, await fetchImageToBase64(imageURL)]); promises.push([_file, await fetchImageToBase64(imageURL)]);
continue; continue;
@ -184,15 +188,19 @@ async function encodeAndFormat(req, files, endpoint, mode) {
continue; continue;
} }
if (endpoint && endpoint === EModelEndpoint.google && mode === VisionModes.generative) { if (
effectiveEndpoint &&
effectiveEndpoint === EModelEndpoint.google &&
mode === VisionModes.generative
) {
delete imagePart.image_url; delete imagePart.image_url;
imagePart.inlineData = { imagePart.inlineData = {
mimeType: file.type, mimeType: file.type,
data: imageContent, data: imageContent,
}; };
} else if (endpoint && endpoint === EModelEndpoint.google) { } else if (effectiveEndpoint && effectiveEndpoint === EModelEndpoint.google) {
imagePart.image_url = imagePart.image_url.url; imagePart.image_url = imagePart.image_url.url;
} else if (endpoint && endpoint === EModelEndpoint.anthropic) { } else if (effectiveEndpoint && effectiveEndpoint === EModelEndpoint.anthropic) {
imagePart.type = 'image'; imagePart.type = 'image';
imagePart.source = { imagePart.source = {
type: 'base64', type: 'base64',

View file

@ -15,6 +15,7 @@ const {
checkOpenAIStorage, checkOpenAIStorage,
removeNullishValues, removeNullishValues,
isAssistantsEndpoint, isAssistantsEndpoint,
getEndpointFileConfig,
} = require('librechat-data-provider'); } = require('librechat-data-provider');
const { EnvVar } = require('@librechat/agents'); const { EnvVar } = require('@librechat/agents');
const { logger } = require('@librechat/data-schemas'); const { logger } = require('@librechat/data-schemas');
@ -994,7 +995,7 @@ async function saveBase64Image(
*/ */
function filterFile({ req, image, isAvatar }) { function filterFile({ req, image, isAvatar }) {
const { file } = req; const { file } = req;
const { endpoint, file_id, width, height } = req.body; const { endpoint, endpointType, file_id, width, height } = req.body;
if (!file_id && !isAvatar) { if (!file_id && !isAvatar) {
throw new Error('No file_id provided'); throw new Error('No file_id provided');
@ -1016,9 +1017,13 @@ function filterFile({ req, image, isAvatar }) {
const appConfig = req.config; const appConfig = req.config;
const fileConfig = mergeFileConfig(appConfig.fileConfig); const fileConfig = mergeFileConfig(appConfig.fileConfig);
const { fileSizeLimit: sizeLimit, supportedMimeTypes } = const endpointFileConfig = getEndpointFileConfig({
fileConfig.endpoints[endpoint] ?? fileConfig.endpoints.default; endpoint,
const fileSizeLimit = isAvatar === true ? fileConfig.avatarSizeLimit : sizeLimit; fileConfig,
endpointType,
});
const fileSizeLimit =
isAvatar === true ? fileConfig.avatarSizeLimit : endpointFileConfig.fileSizeLimit;
if (file.size > fileSizeLimit) { if (file.size > fileSizeLimit) {
throw new Error( throw new Error(
@ -1028,7 +1033,10 @@ function filterFile({ req, image, isAvatar }) {
); );
} }
const isSupportedMimeType = fileConfig.checkType(file.mimetype, supportedMimeTypes); const isSupportedMimeType = fileConfig.checkType(
file.mimetype,
endpointFileConfig.supportedMimeTypes,
);
if (!isSupportedMimeType) { if (!isSupportedMimeType) {
throw new Error('Unsupported file type'); throw new Error('Unsupported file type');

View file

@ -1,7 +1,7 @@
import React, { createContext, useContext, useMemo } from 'react'; import React, { createContext, useContext, useMemo } from 'react';
import { getEndpointField } from 'librechat-data-provider';
import type { EModelEndpoint } from 'librechat-data-provider'; import type { EModelEndpoint } from 'librechat-data-provider';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { getEndpointField } from '~/utils/endpoints';
import { useChatContext } from './ChatContext'; import { useChatContext } from './ChatContext';
interface DragDropContextValue { interface DragDropContextValue {

View file

@ -5,12 +5,12 @@ import {
EModelEndpoint, EModelEndpoint,
mergeFileConfig, mergeFileConfig,
isAgentsEndpoint, isAgentsEndpoint,
getEndpointField,
isAssistantsEndpoint, isAssistantsEndpoint,
fileConfig as defaultFileConfig, getEndpointFileConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { EndpointFileConfig, TConversation } from 'librechat-data-provider'; import type { TConversation } from 'librechat-data-provider';
import { useGetFileConfig, useGetEndpointsQuery } from '~/data-provider'; import { useGetFileConfig, useGetEndpointsQuery } from '~/data-provider';
import { getEndpointField } from '~/utils/endpoints';
import AttachFileMenu from './AttachFileMenu'; import AttachFileMenu from './AttachFileMenu';
import AttachFile from './AttachFile'; import AttachFile from './AttachFile';
@ -26,7 +26,7 @@ function AttachFileChat({
const isAgents = useMemo(() => isAgentsEndpoint(endpoint), [endpoint]); const isAgents = useMemo(() => isAgentsEndpoint(endpoint), [endpoint]);
const isAssistants = useMemo(() => isAssistantsEndpoint(endpoint), [endpoint]); const isAssistants = useMemo(() => isAssistantsEndpoint(endpoint), [endpoint]);
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
@ -39,9 +39,23 @@ function AttachFileChat({
); );
}, [endpoint, endpointsConfig]); }, [endpoint, endpointsConfig]);
const endpointFileConfig = fileConfig.endpoints[endpoint ?? ''] as EndpointFileConfig | undefined; const endpointFileConfig = useMemo(
const endpointSupportsFiles: boolean = supportsFiles[endpointType ?? endpoint ?? ''] ?? false; () =>
const isUploadDisabled = (disableInputs || endpointFileConfig?.disabled) ?? false; getEndpointFileConfig({
endpoint,
fileConfig,
endpointType,
}),
[endpoint, fileConfig, endpointType],
);
const endpointSupportsFiles: boolean = useMemo(
() => supportsFiles[endpointType ?? endpoint ?? ''] ?? false,
[endpointType, endpoint],
);
const isUploadDisabled = useMemo(
() => (disableInputs || endpointFileConfig?.disabled) ?? false,
[disableInputs, endpointFileConfig?.disabled],
);
if (isAssistants && endpointSupportsFiles && !isUploadDisabled) { if (isAssistants && endpointSupportsFiles && !isUploadDisabled) {
return <AttachFile disabled={disableInputs} />; return <AttachFile disabled={disableInputs} />;

View file

@ -61,13 +61,8 @@ const AttachFileMenu = ({
ephemeralAgentByConvoId(conversationId), ephemeralAgentByConvoId(conversationId),
); );
const [toolResource, setToolResource] = useState<EToolResources | undefined>(); const [toolResource, setToolResource] = useState<EToolResources | undefined>();
const { handleFileChange } = useFileHandling({ const { handleFileChange } = useFileHandling();
overrideEndpoint: EModelEndpoint.agents,
overrideEndpointFileConfig: endpointFileConfig,
});
const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({ const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({
overrideEndpoint: EModelEndpoint.agents,
overrideEndpointFileConfig: endpointFileConfig,
toolResource, toolResource,
}); });

View file

@ -1,10 +1,8 @@
import { useRecoilState } from 'recoil'; import { useState } from 'react';
import { Settings2 } from 'lucide-react'; import { Settings2 } from 'lucide-react';
import { useState, useEffect, useMemo } from 'react'; import { TooltipAnchor } from '@librechat/client';
import { Root, Anchor } from '@radix-ui/react-popover'; import { Root, Anchor } from '@radix-ui/react-popover';
import { PluginStoreDialog, TooltipAnchor } from '@librechat/client'; import { isParamEndpoint, getEndpointField, tConvoUpdateSchema } from 'librechat-data-provider';
import { useUserKeyQuery } from 'librechat-data-provider/react-query';
import { EModelEndpoint, isParamEndpoint, tConvoUpdateSchema } from 'librechat-data-provider';
import type { TPreset, TInterfaceConfig } from 'librechat-data-provider'; import type { TPreset, TInterfaceConfig } from 'librechat-data-provider';
import { EndpointSettings, SaveAsPresetDialog, AlternativeSettings } from '~/components/Endpoints'; import { EndpointSettings, SaveAsPresetDialog, AlternativeSettings } from '~/components/Endpoints';
import { useSetIndexOptions, useLocalize } from '~/hooks'; import { useSetIndexOptions, useLocalize } from '~/hooks';
@ -12,8 +10,6 @@ import { useGetEndpointsQuery } from '~/data-provider';
import OptionsPopover from './OptionsPopover'; import OptionsPopover from './OptionsPopover';
import PopoverButtons from './PopoverButtons'; import PopoverButtons from './PopoverButtons';
import { useChatContext } from '~/Providers'; import { useChatContext } from '~/Providers';
import { getEndpointField } from '~/utils';
import store from '~/store';
export default function HeaderOptions({ export default function HeaderOptions({
interfaceConfig, interfaceConfig,
@ -23,36 +19,11 @@ export default function HeaderOptions({
const { data: endpointsConfig } = useGetEndpointsQuery(); const { data: endpointsConfig } = useGetEndpointsQuery();
const [saveAsDialogShow, setSaveAsDialogShow] = useState<boolean>(false); const [saveAsDialogShow, setSaveAsDialogShow] = useState<boolean>(false);
const [showPluginStoreDialog, setShowPluginStoreDialog] = useRecoilState(
store.showPluginStoreDialog,
);
const localize = useLocalize(); const localize = useLocalize();
const { showPopover, conversation, setShowPopover } = useChatContext(); const { showPopover, conversation, setShowPopover } = useChatContext();
const { setOption } = useSetIndexOptions(); const { setOption } = useSetIndexOptions();
const { endpoint, conversationId } = conversation ?? {}; const { endpoint } = conversation ?? {};
const { data: keyExpiry = { expiresAt: undefined } } = useUserKeyQuery(endpoint ?? '');
const userProvidesKey = useMemo(
() => !!(endpointsConfig?.[endpoint ?? '']?.userProvide ?? false),
[endpointsConfig, endpoint],
);
const keyProvided = useMemo(
() => (userProvidesKey ? !!(keyExpiry.expiresAt ?? '') : true),
[keyExpiry.expiresAt, userProvidesKey],
);
const noSettings = useMemo<{ [key: string]: boolean }>(
() => ({
[EModelEndpoint.chatGPTBrowser]: true,
}),
[conversationId],
);
useEffect(() => {
if (endpoint && noSettings[endpoint]) {
setShowPopover(false);
}
}, [endpoint, noSettings]);
const saveAsPreset = () => { const saveAsPreset = () => {
setSaveAsDialogShow(true); setSaveAsDialogShow(true);
@ -76,22 +47,20 @@ export default function HeaderOptions({
<div className="my-auto lg:max-w-2xl xl:max-w-3xl"> <div className="my-auto lg:max-w-2xl xl:max-w-3xl">
<span className="flex w-full flex-col items-center justify-center gap-0 md:order-none md:m-auto md:gap-2"> <span className="flex w-full flex-col items-center justify-center gap-0 md:order-none md:m-auto md:gap-2">
<div className="z-[61] flex w-full items-center justify-center gap-2"> <div className="z-[61] flex w-full items-center justify-center gap-2">
{!noSettings[endpoint] && {interfaceConfig?.parameters === true && paramEndpoint === false && (
interfaceConfig?.parameters === true && <TooltipAnchor
paramEndpoint === false && ( id="parameters-button"
<TooltipAnchor aria-label={localize('com_ui_model_parameters')}
id="parameters-button" description={localize('com_ui_model_parameters')}
aria-label={localize('com_ui_model_parameters')} tabIndex={0}
description={localize('com_ui_model_parameters')} role="button"
tabIndex={0} onClick={triggerAdvancedMode}
role="button" data-testid="parameters-button"
onClick={triggerAdvancedMode} className="inline-flex size-10 items-center justify-center rounded-lg border border-border-light bg-transparent text-text-primary transition-all ease-in-out hover:bg-surface-tertiary disabled:pointer-events-none disabled:opacity-50 radix-state-open:bg-surface-tertiary"
data-testid="parameters-button" >
className="inline-flex size-10 items-center justify-center rounded-lg border border-border-light bg-transparent text-text-primary transition-all ease-in-out hover:bg-surface-tertiary disabled:pointer-events-none disabled:opacity-50 radix-state-open:bg-surface-tertiary" <Settings2 size={16} aria-label="Settings/Parameters Icon" />
> </TooltipAnchor>
<Settings2 size={16} aria-label="Settings/Parameters Icon" /> )}
</TooltipAnchor>
)}
</div> </div>
{interfaceConfig?.parameters === true && paramEndpoint === false && ( {interfaceConfig?.parameters === true && paramEndpoint === false && (
<OptionsPopover <OptionsPopover
@ -122,12 +91,6 @@ export default function HeaderOptions({
} }
/> />
)} )}
{interfaceConfig?.parameters === true && (
<PluginStoreDialog
isOpen={showPluginStoreDialog}
setIsOpen={setShowPluginStoreDialog}
/>
)}
</span> </span>
</div> </div>
</Anchor> </Anchor>

View file

@ -1,7 +1,6 @@
import React from 'react'; import React from 'react';
import { EModelEndpoint } from 'librechat-data-provider'; import { EModelEndpoint, getEndpointField } from 'librechat-data-provider';
import { SetKeyDialog } from '~/components/Input/SetKeyDialog'; import { SetKeyDialog } from '~/components/Input/SetKeyDialog';
import { getEndpointField } from '~/utils';
interface DialogManagerProps { interface DialogManagerProps {
keyDialogOpen: boolean; keyDialogOpen: boolean;

View file

@ -1,7 +1,8 @@
import React, { memo } from 'react'; import React, { memo } from 'react';
import { getEndpointField } from 'librechat-data-provider';
import type { TModelSpec, TEndpointsConfig } from 'librechat-data-provider'; import type { TModelSpec, TEndpointsConfig } from 'librechat-data-provider';
import type { IconMapProps } from '~/common'; import type { IconMapProps } from '~/common';
import { getModelSpecIconURL, getIconKey, getEndpointField } from '~/utils'; import { getModelSpecIconURL, getIconKey } from '~/utils';
import { URLIcon } from '~/components/Endpoints/URLIcon'; import { URLIcon } from '~/components/Endpoints/URLIcon';
import { icons } from '~/hooks/Endpoint/Icons'; import { icons } from '~/hooks/Endpoint/Icons';

View file

@ -1,20 +1,21 @@
import { useRecoilValue } from 'recoil'; import { useRecoilValue } from 'recoil';
import { Close } from '@radix-ui/react-popover'; import { Close } from '@radix-ui/react-popover';
import { Flipper, Flipped } from 'react-flip-toolkit'; import { Flipper, Flipped } from 'react-flip-toolkit';
import { getEndpointField } from 'librechat-data-provider';
import { import {
Dialog, Dialog,
DialogTrigger,
Label, Label,
DialogTemplate,
PinIcon, PinIcon,
EditIcon, EditIcon,
TrashIcon, TrashIcon,
DialogTrigger,
DialogTemplate,
} from '@librechat/client'; } from '@librechat/client';
import type { TPreset } from 'librechat-data-provider'; import type { TPreset } from 'librechat-data-provider';
import type { FC } from 'react'; import type { FC } from 'react';
import { getPresetTitle, getEndpointField, getIconKey } from '~/utils';
import FileUpload from '~/components/Chat/Input/Files/FileUpload'; import FileUpload from '~/components/Chat/Input/Files/FileUpload';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { getPresetTitle, getIconKey } from '~/utils';
import { MenuSeparator, MenuItem } from '../UI'; import { MenuSeparator, MenuItem } from '../UI';
import { icons } from '~/hooks/Endpoint/Icons'; import { icons } from '~/hooks/Endpoint/Icons';
import { useLocalize } from '~/hooks'; import { useLocalize } from '~/hooks';

View file

@ -1,9 +1,10 @@
import React, { useMemo, memo } from 'react'; import React, { useMemo, memo } from 'react';
import { getEndpointField } from 'librechat-data-provider';
import type { Assistant, Agent } from 'librechat-data-provider'; import type { Assistant, Agent } from 'librechat-data-provider';
import type { TMessageIcon } from '~/common'; import type { TMessageIcon } from '~/common';
import { getEndpointField, getIconEndpoint, logger } from '~/utils';
import ConvoIconURL from '~/components/Endpoints/ConvoIconURL'; import ConvoIconURL from '~/components/Endpoints/ConvoIconURL';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { getIconEndpoint, logger } from '~/utils';
import Icon from '~/components/Endpoints/Icon'; import Icon from '~/components/Endpoints/Icon';
const MessageIcon = memo( const MessageIcon = memo(

View file

@ -1,6 +1,7 @@
import React, { useMemo } from 'react'; import React, { useMemo } from 'react';
import { getEndpointField } from 'librechat-data-provider';
import type * as t from 'librechat-data-provider'; import type * as t from 'librechat-data-provider';
import { getEndpointField, getIconKey, getEntity, getIconEndpoint } from '~/utils'; import { getIconKey, getEntity, getIconEndpoint } from '~/utils';
import ConvoIconURL from '~/components/Endpoints/ConvoIconURL'; import ConvoIconURL from '~/components/Endpoints/ConvoIconURL';
import { icons } from '~/hooks/Endpoint/Icons'; import { icons } from '~/hooks/Endpoint/Icons';

View file

@ -1,13 +1,13 @@
import { isAssistantsEndpoint } from 'librechat-data-provider'; import { getEndpointField, isAssistantsEndpoint } from 'librechat-data-provider';
import type { import type {
TConversation,
TEndpointsConfig,
TPreset, TPreset,
TConversation,
TAssistantsMap, TAssistantsMap,
TEndpointsConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import ConvoIconURL from '~/components/Endpoints/ConvoIconURL'; import ConvoIconURL from '~/components/Endpoints/ConvoIconURL';
import MinimalIcon from '~/components/Endpoints/MinimalIcon'; import MinimalIcon from '~/components/Endpoints/MinimalIcon';
import { getEndpointField, getIconEndpoint } from '~/utils'; import { getIconEndpoint } from '~/utils';
export default function EndpointIcon({ export default function EndpointIcon({
conversation, conversation,

View file

@ -1,10 +1,11 @@
import { useRecoilValue } from 'recoil'; import { useRecoilValue } from 'recoil';
import { SettingsViews, TConversation } from 'librechat-data-provider';
import { useGetModelsQuery } from 'librechat-data-provider/react-query'; import { useGetModelsQuery } from 'librechat-data-provider/react-query';
import { getEndpointField, SettingsViews } from 'librechat-data-provider';
import type { TConversation } from 'librechat-data-provider';
import type { TSettingsProps } from '~/common'; import type { TSettingsProps } from '~/common';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { cn, getEndpointField } from '~/utils';
import { getSettings } from './Settings'; import { getSettings } from './Settings';
import { cn } from '~/utils';
import store from '~/store'; import store from '~/store';
export default function Settings({ export default function Settings({

View file

@ -1,12 +1,11 @@
import React, { useState, useMemo, useCallback } from 'react'; import React, { useState, useMemo, useCallback } from 'react';
import { useToastContext } from '@librechat/client'; import { useToastContext } from '@librechat/client';
import { EModelEndpoint } from 'librechat-data-provider';
import { Controller, useWatch, useFormContext } from 'react-hook-form'; import { Controller, useWatch, useFormContext } from 'react-hook-form';
import { EModelEndpoint, getEndpointField } from 'librechat-data-provider';
import type { AgentForm, AgentPanelProps, IconComponentTypes } from '~/common'; import type { AgentForm, AgentPanelProps, IconComponentTypes } from '~/common';
import { import {
removeFocusOutlines, removeFocusOutlines,
processAgentOption, processAgentOption,
getEndpointField,
defaultTextProps, defaultTextProps,
validateEmail, validateEmail,
getIconKey, getIconKey,

View file

@ -6,9 +6,8 @@ import {
EModelEndpoint, EModelEndpoint,
mergeFileConfig, mergeFileConfig,
AgentCapabilities, AgentCapabilities,
fileConfig as defaultFileConfig, getEndpointFileConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { EndpointFileConfig } from 'librechat-data-provider';
import type { ExtendedFile, AgentForm } from '~/common'; import type { ExtendedFile, AgentForm } from '~/common';
import { useFileHandling, useLocalize, useLazyEffect } from '~/hooks'; import { useFileHandling, useLocalize, useLazyEffect } from '~/hooks';
import FileRow from '~/components/Chat/Input/Files/FileRow'; import FileRow from '~/components/Chat/Input/Files/FileRow';
@ -30,12 +29,11 @@ export default function Files({
const { watch } = useFormContext<AgentForm>(); const { watch } = useFormContext<AgentForm>();
const fileInputRef = useRef<HTMLInputElement>(null); const fileInputRef = useRef<HTMLInputElement>(null);
const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map()); const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map());
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const { abortUpload, handleFileChange } = useFileHandling({ const { abortUpload, handleFileChange } = useFileHandling({
fileSetter: setFiles, fileSetter: setFiles,
overrideEndpoint: EModelEndpoint.agents,
additionalMetadata: { agent_id, tool_resource }, additionalMetadata: { agent_id, tool_resource },
}); });
@ -51,9 +49,11 @@ export default function Files({
const codeChecked = watch(AgentCapabilities.execute_code); const codeChecked = watch(AgentCapabilities.execute_code);
const endpointFileConfig = fileConfig.endpoints[EModelEndpoint.agents] as const endpointFileConfig = getEndpointFileConfig({
| EndpointFileConfig fileConfig,
| undefined; endpoint: EModelEndpoint.agents,
endpointType: EModelEndpoint.agents,
});
const isUploadDisabled = endpointFileConfig?.disabled ?? false; const isUploadDisabled = endpointFileConfig?.disabled ?? false;
if (isUploadDisabled) { if (isUploadDisabled) {

View file

@ -5,7 +5,7 @@ import {
EModelEndpoint, EModelEndpoint,
EToolResources, EToolResources,
mergeFileConfig, mergeFileConfig,
fileConfig as defaultFileConfig, getEndpointFileConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import { import {
HoverCard, HoverCard,
@ -41,17 +41,15 @@ export default function FileContext({
const { data: startupConfig } = useGetStartupConfig(); const { data: startupConfig } = useGetStartupConfig();
const sharePointEnabled = startupConfig?.sharePointFilePickerEnabled; const sharePointEnabled = startupConfig?.sharePointFilePickerEnabled;
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const { handleFileChange } = useFileHandling({ const { handleFileChange } = useFileHandling({
overrideEndpoint: EModelEndpoint.agents,
additionalMetadata: { agent_id, tool_resource: EToolResources.context }, additionalMetadata: { agent_id, tool_resource: EToolResources.context },
fileSetter: setFiles, fileSetter: setFiles,
}); });
const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({ const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({
overrideEndpoint: EModelEndpoint.agents,
additionalMetadata: { agent_id, tool_resource: EToolResources.file_search }, additionalMetadata: { agent_id, tool_resource: EToolResources.file_search },
fileSetter: setFiles, fileSetter: setFiles,
}); });
@ -65,8 +63,12 @@ export default function FileContext({
750, 750,
); );
const endpointFileConfig = fileConfig.endpoints[EModelEndpoint.agents]; const endpointFileConfig = getEndpointFileConfig({
const isUploadDisabled = endpointFileConfig.disabled ?? false; fileConfig,
endpoint: EModelEndpoint.agents,
endpointType: EModelEndpoint.agents,
});
const isUploadDisabled = endpointFileConfig?.disabled ?? false;
const handleSharePointFilesSelected = async (sharePointFiles: any[]) => { const handleSharePointFilesSelected = async (sharePointFiles: any[]) => {
try { try {
await handleSharePointFiles(sharePointFiles); await handleSharePointFiles(sharePointFiles);

View file

@ -8,7 +8,7 @@ import {
EToolResources, EToolResources,
mergeFileConfig, mergeFileConfig,
AgentCapabilities, AgentCapabilities,
fileConfig as defaultFileConfig, getEndpointFileConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { ExtendedFile, AgentForm } from '~/common'; import type { ExtendedFile, AgentForm } from '~/common';
import useSharePointFileHandling from '~/hooks/Files/useSharePointFileHandling'; import useSharePointFileHandling from '~/hooks/Files/useSharePointFileHandling';
@ -38,18 +38,16 @@ export default function FileSearch({
// Get startup configuration for SharePoint feature flag // Get startup configuration for SharePoint feature flag
const { data: startupConfig } = useGetStartupConfig(); const { data: startupConfig } = useGetStartupConfig();
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const { handleFileChange } = useFileHandling({ const { handleFileChange } = useFileHandling({
overrideEndpoint: EModelEndpoint.agents,
additionalMetadata: { agent_id, tool_resource: EToolResources.file_search }, additionalMetadata: { agent_id, tool_resource: EToolResources.file_search },
fileSetter: setFiles, fileSetter: setFiles,
}); });
const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({ const { handleSharePointFiles, isProcessing, downloadProgress } = useSharePointFileHandling({
overrideEndpoint: EModelEndpoint.agents,
additionalMetadata: { agent_id, tool_resource: EToolResources.file_search }, additionalMetadata: { agent_id, tool_resource: EToolResources.file_search },
fileSetter: setFiles, fileSetter: setFiles,
}); });
@ -66,8 +64,12 @@ export default function FileSearch({
const fileSearchChecked = watch(AgentCapabilities.file_search); const fileSearchChecked = watch(AgentCapabilities.file_search);
const endpointFileConfig = fileConfig.endpoints[EModelEndpoint.agents]; const endpointFileConfig = getEndpointFileConfig({
const isUploadDisabled = endpointFileConfig.disabled ?? false; fileConfig,
endpoint: EModelEndpoint.agents,
endpointType: EModelEndpoint.agents,
});
const isUploadDisabled = endpointFileConfig?.disabled ?? false;
const sharePointEnabled = startupConfig?.sharePointFilePickerEnabled; const sharePointEnabled = startupConfig?.sharePointFilePickerEnabled;
const disabledUploadButton = isEphemeralAgent(agent_id) || fileSearchChecked === false; const disabledUploadButton = isEphemeralAgent(agent_id) || fileSearchChecked === false;

View file

@ -7,6 +7,7 @@ import { componentMapping } from '~/components/SidePanel/Parameters/components';
import { import {
alternateName, alternateName,
getSettingsKeys, getSettingsKeys,
getEndpointField,
LocalStorageKeys, LocalStorageKeys,
SettingDefinition, SettingDefinition,
agentParamSettings, agentParamSettings,
@ -14,9 +15,9 @@ import {
import type * as t from 'librechat-data-provider'; import type * as t from 'librechat-data-provider';
import type { AgentForm, AgentModelPanelProps, StringOption } from '~/common'; import type { AgentForm, AgentModelPanelProps, StringOption } from '~/common';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { getEndpointField, cn } from '~/utils';
import { useLocalize } from '~/hooks'; import { useLocalize } from '~/hooks';
import { Panel } from '~/common'; import { Panel } from '~/common';
import { cn } from '~/utils';
export default function ModelPanel({ export default function ModelPanel({
providers, providers,

View file

@ -1,10 +1,6 @@
import { useState, useRef, useEffect } from 'react'; import { useState, useRef, useEffect } from 'react';
import { import { EToolResources, mergeFileConfig, getEndpointFileConfig } from 'librechat-data-provider';
EToolResources, import type { AssistantsEndpoint } from 'librechat-data-provider';
mergeFileConfig,
fileConfig as defaultFileConfig,
} from 'librechat-data-provider';
import type { AssistantsEndpoint, EndpointFileConfig } from 'librechat-data-provider';
import type { ExtendedFile } from '~/common'; import type { ExtendedFile } from '~/common';
import FileRow from '~/components/Chat/Input/Files/FileRow'; import FileRow from '~/components/Chat/Input/Files/FileRow';
import { useGetFileConfig } from '~/data-provider'; import { useGetFileConfig } from '~/data-provider';
@ -28,11 +24,10 @@ export default function CodeFiles({
const { setFilesLoading } = useChatContext(); const { setFilesLoading } = useChatContext();
const fileInputRef = useRef<HTMLInputElement>(null); const fileInputRef = useRef<HTMLInputElement>(null);
const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map()); const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map());
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const { handleFileChange } = useFileHandling({ const { handleFileChange } = useFileHandling({
overrideEndpoint: endpoint,
additionalMetadata: { assistant_id, tool_resource }, additionalMetadata: { assistant_id, tool_resource },
fileSetter: setFiles, fileSetter: setFiles,
}); });
@ -43,7 +38,11 @@ export default function CodeFiles({
} }
}, [_files]); }, [_files]);
const endpointFileConfig = fileConfig.endpoints[endpoint] as EndpointFileConfig | undefined; const endpointFileConfig = getEndpointFileConfig({
fileConfig,
endpoint,
endpointType: endpoint,
});
const isUploadDisabled = endpointFileConfig?.disabled ?? false; const isUploadDisabled = endpointFileConfig?.disabled ?? false;
if (isUploadDisabled) { if (isUploadDisabled) {

View file

@ -2,9 +2,9 @@ import { useState, useRef, useEffect } from 'react';
import { import {
mergeFileConfig, mergeFileConfig,
retrievalMimeTypes, retrievalMimeTypes,
fileConfig as defaultFileConfig, getEndpointFileConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { AssistantsEndpoint, EndpointFileConfig } from 'librechat-data-provider'; import type { AssistantsEndpoint } from 'librechat-data-provider';
import type { ExtendedFile } from '~/common'; import type { ExtendedFile } from '~/common';
import FileRow from '~/components/Chat/Input/Files/FileRow'; import FileRow from '~/components/Chat/Input/Files/FileRow';
import { useGetFileConfig } from '~/data-provider'; import { useGetFileConfig } from '~/data-provider';
@ -38,11 +38,10 @@ export default function Knowledge({
const { setFilesLoading } = useChatContext(); const { setFilesLoading } = useChatContext();
const fileInputRef = useRef<HTMLInputElement>(null); const fileInputRef = useRef<HTMLInputElement>(null);
const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map()); const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map());
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const { handleFileChange } = useFileHandling({ const { handleFileChange } = useFileHandling({
overrideEndpoint: endpoint,
additionalMetadata: { assistant_id }, additionalMetadata: { assistant_id },
fileSetter: setFiles, fileSetter: setFiles,
}); });
@ -53,7 +52,11 @@ export default function Knowledge({
} }
}, [_files]); }, [_files]);
const endpointFileConfig = fileConfig.endpoints[endpoint] as EndpointFileConfig | undefined; const endpointFileConfig = getEndpointFileConfig({
fileConfig,
endpoint,
endpointType: endpoint,
});
const isUploadDisabled = endpointFileConfig?.disabled ?? false; const isUploadDisabled = endpointFileConfig?.disabled ?? false;
if (isUploadDisabled) { if (isUploadDisabled) {

View file

@ -30,6 +30,7 @@ import {
mergeFileConfig, mergeFileConfig,
megabyte, megabyte,
isAssistantsEndpoint, isAssistantsEndpoint,
getEndpointFileConfig,
type TFile, type TFile,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import { useFileMapContext, useChatContext } from '~/Providers'; import { useFileMapContext, useChatContext } from '~/Providers';
@ -86,7 +87,7 @@ export default function DataTable<TData, TValue>({ columns, data }: DataTablePro
const fileMap = useFileMapContext(); const fileMap = useFileMapContext();
const { showToast } = useToastContext(); const { showToast } = useToastContext();
const { setFiles, conversation } = useChatContext(); const { setFiles, conversation } = useChatContext();
const { data: fileConfig = defaultFileConfig } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const { addFile } = useUpdateFiles(setFiles); const { addFile } = useUpdateFiles(setFiles);
@ -103,6 +104,7 @@ export default function DataTable<TData, TValue>({ columns, data }: DataTablePro
const fileData = fileMap[file.file_id]; const fileData = fileMap[file.file_id];
const endpoint = conversation.endpoint; const endpoint = conversation.endpoint;
const endpointType = conversation.endpointType;
if (!fileData.source) { if (!fileData.source) {
return; return;
@ -126,20 +128,31 @@ export default function DataTable<TData, TValue>({ columns, data }: DataTablePro
}); });
} }
const { fileSizeLimit, supportedMimeTypes } = const endpointFileConfig = getEndpointFileConfig({
fileConfig.endpoints[endpoint] ?? fileConfig.endpoints.default; fileConfig,
endpoint,
endpointType,
});
if (fileData.bytes > fileSizeLimit) { if (endpointFileConfig.disabled === true) {
showToast({
message: localize('com_ui_attach_error_disabled'),
status: 'error',
});
return;
}
if (fileData.bytes > (endpointFileConfig.fileSizeLimit ?? Number.MAX_SAFE_INTEGER)) {
showToast({ showToast({
message: `${localize('com_ui_attach_error_size')} ${ message: `${localize('com_ui_attach_error_size')} ${
fileSizeLimit / megabyte (endpointFileConfig.fileSizeLimit ?? 0) / megabyte
} MB (${endpoint})`, } MB (${endpoint})`,
status: 'error', status: 'error',
}); });
return; return;
} }
if (!defaultFileConfig.checkType(file.type, supportedMimeTypes)) { if (!defaultFileConfig.checkType(file.type, endpointFileConfig.supportedMimeTypes ?? [])) {
showToast({ showToast({
message: `${localize('com_ui_attach_error_type')} ${file.type} (${endpoint})`, message: `${localize('com_ui_attach_error_type')} ${file.type} (${endpoint})`,
status: 'error', status: 'error',
@ -162,7 +175,7 @@ export default function DataTable<TData, TValue>({ columns, data }: DataTablePro
metadata: fileData.metadata, metadata: fileData.metadata,
}); });
}, },
[addFile, fileMap, conversation, localize, showToast, fileConfig.endpoints], [addFile, fileMap, conversation, localize, showToast, fileConfig],
); );
const filenameFilter = table.getColumn('filename')?.getFilterValue() as string; const filenameFilter = table.getColumn('filename')?.getFilterValue() as string;

View file

@ -5,6 +5,7 @@ import {
excludedKeys, excludedKeys,
paramSettings, paramSettings,
getSettingsKeys, getSettingsKeys,
getEndpointField,
SettingDefinition, SettingDefinition,
tConvoUpdateSchema, tConvoUpdateSchema,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
@ -12,9 +13,9 @@ import type { TPreset } from 'librechat-data-provider';
import { SaveAsPresetDialog } from '~/components/Endpoints'; import { SaveAsPresetDialog } from '~/components/Endpoints';
import { useSetIndexOptions, useLocalize } from '~/hooks'; import { useSetIndexOptions, useLocalize } from '~/hooks';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { getEndpointField, logger } from '~/utils';
import { componentMapping } from './components'; import { componentMapping } from './components';
import { useChatContext } from '~/Providers'; import { useChatContext } from '~/Providers';
import { logger } from '~/utils';
export default function Parameters() { export default function Parameters() {
const localize = useLocalize(); const localize = useLocalize();

View file

@ -1,4 +1,5 @@
import { useState, useCallback, useMemo, memo } from 'react'; import { useState, useCallback, useMemo, memo } from 'react';
import { getEndpointField } from 'librechat-data-provider';
import { useUserKeyQuery } from 'librechat-data-provider/react-query'; import { useUserKeyQuery } from 'librechat-data-provider/react-query';
import { ResizableHandleAlt, ResizablePanel, useMediaQuery } from '@librechat/client'; import { ResizableHandleAlt, ResizablePanel, useMediaQuery } from '@librechat/client';
import type { TEndpointsConfig, TInterfaceConfig } from 'librechat-data-provider'; import type { TEndpointsConfig, TInterfaceConfig } from 'librechat-data-provider';
@ -8,7 +9,7 @@ import { useLocalStorage, useLocalize } from '~/hooks';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import NavToggle from '~/components/Nav/NavToggle'; import NavToggle from '~/components/Nav/NavToggle';
import { useSidePanelContext } from '~/Providers'; import { useSidePanelContext } from '~/Providers';
import { cn, getEndpointField } from '~/utils'; import { cn } from '~/utils';
import Nav from './Nav'; import Nav from './Nav';
const defaultMinSize = 20; const defaultMinSize = 20;

View file

@ -6,6 +6,7 @@ import {
QueryKeys, QueryKeys,
ContentTypes, ContentTypes,
EModelEndpoint, EModelEndpoint,
getEndpointField,
isAgentsEndpoint, isAgentsEndpoint,
parseCompactConvo, parseCompactConvo,
replaceSpecialVars, replaceSpecialVars,
@ -25,10 +26,10 @@ import type { TAskFunction, ExtendedFile } from '~/common';
import useSetFilesToDelete from '~/hooks/Files/useSetFilesToDelete'; import useSetFilesToDelete from '~/hooks/Files/useSetFilesToDelete';
import useGetSender from '~/hooks/Conversations/useGetSender'; import useGetSender from '~/hooks/Conversations/useGetSender';
import store, { useGetEphemeralAgent } from '~/store'; import store, { useGetEphemeralAgent } from '~/store';
import { getEndpointField, logger } from '~/utils';
import useUserKey from '~/hooks/Input/useUserKey'; import useUserKey from '~/hooks/Input/useUserKey';
import { useNavigate } from 'react-router-dom'; import { useNavigate } from 'react-router-dom';
import { useAuthContext } from '~/hooks'; import { useAuthContext } from '~/hooks';
import { logger } from '~/utils';
const logChatRequest = (request: Record<string, unknown>) => { const logChatRequest = (request: Record<string, unknown>) => {
logger.log('=====================================\nAsk function called with:'); logger.log('=====================================\nAsk function called with:');

View file

@ -1,18 +1,18 @@
import { useRecoilValue } from 'recoil'; import { useRecoilValue } from 'recoil';
import { useCallback, useRef, useEffect } from 'react'; import { useCallback, useRef, useEffect } from 'react';
import { useGetModelsQuery } from 'librechat-data-provider/react-query'; import { useGetModelsQuery } from 'librechat-data-provider/react-query';
import { LocalStorageKeys, isAssistantsEndpoint } from 'librechat-data-provider'; import { getEndpointField, LocalStorageKeys, isAssistantsEndpoint } from 'librechat-data-provider';
import type { import type {
TPreset,
TModelsConfig,
TConversation,
TEndpointsConfig, TEndpointsConfig,
EModelEndpoint, EModelEndpoint,
TModelsConfig,
TConversation,
TPreset,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { SetterOrUpdater } from 'recoil';
import type { AssistantListItem } from '~/common'; import type { AssistantListItem } from '~/common';
import { getEndpointField, buildDefaultConvo, getDefaultEndpoint, logger } from '~/utils'; import type { SetterOrUpdater } from 'recoil';
import useAssistantListMap from '~/hooks/Assistants/useAssistantListMap'; import useAssistantListMap from '~/hooks/Assistants/useAssistantListMap';
import { buildDefaultConvo, getDefaultEndpoint, logger } from '~/utils';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { mainTextareaId } from '~/common'; import { mainTextareaId } from '~/common';
import store from '~/store'; import store from '~/store';

View file

@ -2,20 +2,14 @@ import { useCallback } from 'react';
import { useSetRecoilState } from 'recoil'; import { useSetRecoilState } from 'recoil';
import { useNavigate } from 'react-router-dom'; import { useNavigate } from 'react-router-dom';
import { useQueryClient } from '@tanstack/react-query'; import { useQueryClient } from '@tanstack/react-query';
import { QueryKeys, Constants, dataService } from 'librechat-data-provider'; import { QueryKeys, Constants, dataService, getEndpointField } from 'librechat-data-provider';
import type { import type {
TEndpointsConfig, TEndpointsConfig,
TStartupConfig, TStartupConfig,
TModelsConfig, TModelsConfig,
TConversation, TConversation,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import { import { getDefaultEndpoint, clearMessagesCache, buildDefaultConvo, logger } from '~/utils';
getDefaultEndpoint,
clearMessagesCache,
buildDefaultConvo,
getEndpointField,
logger,
} from '~/utils';
import { useApplyModelSpecEffects } from '~/hooks/Agents'; import { useApplyModelSpecEffects } from '~/hooks/Agents';
import store from '~/store'; import store from '~/store';

View file

@ -5,6 +5,7 @@ import {
alternateName, alternateName,
EModelEndpoint, EModelEndpoint,
PermissionTypes, PermissionTypes,
getEndpointField,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { import type {
TEndpointsConfig, TEndpointsConfig,
@ -14,8 +15,8 @@ import type {
Agent, Agent,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { Endpoint } from '~/common'; import type { Endpoint } from '~/common';
import { mapEndpoints, getIconKey, getEndpointField } from '~/utils';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { mapEndpoints, getIconKey } from '~/utils';
import { useHasAccess } from '~/hooks'; import { useHasAccess } from '~/hooks';
import { icons } from './Icons'; import { icons } from './Icons';

View file

@ -1,5 +1,6 @@
import { useState, useMemo, useCallback, useRef } from 'react'; import { useState, useMemo, useCallback, useRef } from 'react';
import { useDrop } from 'react-dnd'; import { useDrop } from 'react-dnd';
import { useToastContext } from '@librechat/client';
import { NativeTypes } from 'react-dnd-html5-backend'; import { NativeTypes } from 'react-dnd-html5-backend';
import { useQueryClient } from '@tanstack/react-query'; import { useQueryClient } from '@tanstack/react-query';
import { useRecoilValue, useSetRecoilState } from 'recoil'; import { useRecoilValue, useSetRecoilState } from 'recoil';
@ -7,10 +8,12 @@ import {
Tools, Tools,
QueryKeys, QueryKeys,
Constants, Constants,
EModelEndpoint,
EToolResources, EToolResources,
EModelEndpoint,
mergeFileConfig,
AgentCapabilities, AgentCapabilities,
isAssistantsEndpoint, isAssistantsEndpoint,
getEndpointFileConfig,
defaultAgentCapabilities, defaultAgentCapabilities,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { DropTargetMonitor } from 'react-dnd'; import type { DropTargetMonitor } from 'react-dnd';
@ -18,9 +21,12 @@ import type * as t from 'librechat-data-provider';
import store, { ephemeralAgentByConvoId } from '~/store'; import store, { ephemeralAgentByConvoId } from '~/store';
import useFileHandling from './useFileHandling'; import useFileHandling from './useFileHandling';
import { isEphemeralAgent } from '~/common'; import { isEphemeralAgent } from '~/common';
import useLocalize from '../useLocalize';
export default function useDragHelpers() { export default function useDragHelpers() {
const queryClient = useQueryClient(); const queryClient = useQueryClient();
const { showToast } = useToastContext();
const localize = useLocalize();
const [showModal, setShowModal] = useState(false); const [showModal, setShowModal] = useState(false);
const [draggedFiles, setDraggedFiles] = useState<File[]>([]); const [draggedFiles, setDraggedFiles] = useState<File[]>([]);
const conversation = useRecoilValue(store.conversationByIndex(0)) || undefined; const conversation = useRecoilValue(store.conversationByIndex(0)) || undefined;
@ -33,9 +39,7 @@ export default function useDragHelpers() {
[conversation?.endpoint], [conversation?.endpoint],
); );
const { handleFiles } = useFileHandling({ const { handleFiles } = useFileHandling();
overrideEndpoint: isAssistants ? undefined : EModelEndpoint.agents,
});
const handleOptionSelect = useCallback( const handleOptionSelect = useCallback(
(toolResource: EToolResources | undefined) => { (toolResource: EToolResources | undefined) => {
@ -62,6 +66,26 @@ export default function useDragHelpers() {
const handleDrop = useCallback( const handleDrop = useCallback(
(item: { files: File[] }) => { (item: { files: File[] }) => {
/** Early block: leverage endpoint file config to prevent drag/drop on disabled endpoints */
const currentEndpoint = conversationRef.current?.endpoint ?? 'default';
const currentEndpointType = conversationRef.current?.endpointType ?? undefined;
const cfg = queryClient.getQueryData<t.FileConfig>([QueryKeys.fileConfig]);
if (cfg) {
const mergedCfg = mergeFileConfig(cfg);
const endpointCfg = getEndpointFileConfig({
fileConfig: mergedCfg,
endpoint: currentEndpoint,
endpointType: currentEndpointType,
});
if (endpointCfg?.disabled === true) {
showToast({
message: localize('com_ui_attach_error_disabled'),
status: 'error',
});
return;
}
}
if (isAssistants) { if (isAssistants) {
handleFilesRef.current(item.files); handleFilesRef.current(item.files);
return; return;
@ -110,7 +134,7 @@ export default function useDragHelpers() {
setDraggedFiles(item.files); setDraggedFiles(item.files);
setShowModal(true); setShowModal(true);
}, },
[isAssistants, queryClient], [isAssistants, queryClient, showToast, localize],
); );
const [{ canDrop, isOver }, drop] = useDrop( const [{ canDrop, isOver }, drop] = useDrop(

View file

@ -1,4 +1,4 @@
import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react'; import React, { useCallback, useEffect, useRef, useMemo, useState } from 'react';
import { v4 } from 'uuid'; import { v4 } from 'uuid';
import { useSetRecoilState } from 'recoil'; import { useSetRecoilState } from 'recoil';
import { useToastContext } from '@librechat/client'; import { useToastContext } from '@librechat/client';
@ -6,16 +6,14 @@ import { useQueryClient } from '@tanstack/react-query';
import { import {
QueryKeys, QueryKeys,
Constants, Constants,
EModelEndpoint,
EToolResources, EToolResources,
mergeFileConfig, mergeFileConfig,
isAgentsEndpoint,
isAssistantsEndpoint, isAssistantsEndpoint,
getEndpointFileConfig,
defaultAssistantsVersion, defaultAssistantsVersion,
fileConfig as defaultFileConfig,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import debounce from 'lodash/debounce'; import debounce from 'lodash/debounce';
import type { EndpointFileConfig, TEndpointsConfig, TError } from 'librechat-data-provider'; import type { TEndpointsConfig, TError } from 'librechat-data-provider';
import type { ExtendedFile, FileSetter } from '~/common'; import type { ExtendedFile, FileSetter } from '~/common';
import { useGetFileConfig, useUploadFileMutation } from '~/data-provider'; import { useGetFileConfig, useUploadFileMutation } from '~/data-provider';
import useLocalize, { TranslationKeys } from '~/hooks/useLocalize'; import useLocalize, { TranslationKeys } from '~/hooks/useLocalize';
@ -29,9 +27,7 @@ import useUpdateFiles from './useUpdateFiles';
type UseFileHandling = { type UseFileHandling = {
fileSetter?: FileSetter; fileSetter?: FileSetter;
overrideEndpoint?: EModelEndpoint;
fileFilter?: (file: File) => boolean; fileFilter?: (file: File) => boolean;
overrideEndpointFileConfig?: EndpointFileConfig;
additionalMetadata?: Record<string, string | undefined>; additionalMetadata?: Record<string, string | undefined>;
}; };
@ -54,17 +50,13 @@ const useFileHandling = (params?: UseFileHandling) => {
const agent_id = params?.additionalMetadata?.agent_id ?? ''; const agent_id = params?.additionalMetadata?.agent_id ?? '';
const assistant_id = params?.additionalMetadata?.assistant_id ?? ''; const assistant_id = params?.additionalMetadata?.assistant_id ?? '';
const endpointType = useMemo(() => conversation?.endpointType, [conversation?.endpointType]);
const endpoint = useMemo(() => conversation?.endpoint ?? 'default', [conversation?.endpoint]);
const { data: fileConfig = null } = useGetFileConfig({ const { data: fileConfig = null } = useGetFileConfig({
select: (data) => mergeFileConfig(data), select: (data) => mergeFileConfig(data),
}); });
const endpoint = useMemo(
() =>
params?.overrideEndpoint ?? conversation?.endpointType ?? conversation?.endpoint ?? 'default',
[params?.overrideEndpoint, conversation?.endpointType, conversation?.endpoint],
);
const displayToast = useCallback(() => { const displayToast = useCallback(() => {
if (errors.length > 1) { if (errors.length > 1) {
// TODO: this should not be a dynamic localize input!! // TODO: this should not be a dynamic localize input!!
@ -169,10 +161,7 @@ const useFileHandling = (params?: UseFileHandling) => {
const formData = new FormData(); const formData = new FormData();
formData.append('endpoint', endpoint); formData.append('endpoint', endpoint);
formData.append( formData.append('endpointType', endpointType ?? '');
'original_endpoint',
conversation?.endpointType || conversation?.endpoint || '',
);
formData.append('file', extendedFile.file as File, encodeURIComponent(filename)); formData.append('file', extendedFile.file as File, encodeURIComponent(filename));
formData.append('file_id', extendedFile.file_id); formData.append('file_id', extendedFile.file_id);
@ -194,7 +183,7 @@ const useFileHandling = (params?: UseFileHandling) => {
} }
} }
if (isAgentsEndpoint(endpoint)) { if (!isAssistantsEndpoint(endpointType ?? endpoint)) {
if (!agent_id) { if (!agent_id) {
formData.append('message_file', 'true'); formData.append('message_file', 'true');
} }
@ -205,9 +194,7 @@ const useFileHandling = (params?: UseFileHandling) => {
if (conversation?.agent_id != null && formData.get('agent_id') == null) { if (conversation?.agent_id != null && formData.get('agent_id') == null) {
formData.append('agent_id', conversation.agent_id); formData.append('agent_id', conversation.agent_id);
} }
}
if (!isAssistantsEndpoint(endpoint)) {
uploadFile.mutate(formData); uploadFile.mutate(formData);
return; return;
} }
@ -264,18 +251,19 @@ const useFileHandling = (params?: UseFileHandling) => {
/* Validate files */ /* Validate files */
let filesAreValid: boolean; let filesAreValid: boolean;
try { try {
const endpointFileConfig = getEndpointFileConfig({
endpoint,
fileConfig,
endpointType,
});
filesAreValid = validateFiles({ filesAreValid = validateFiles({
files, files,
fileList, fileList,
setError, setError,
endpointFileConfig: fileConfig,
params?.overrideEndpointFileConfig ?? endpointFileConfig,
fileConfig?.endpoints?.[endpoint] ??
fileConfig?.endpoints?.default ??
defaultFileConfig.endpoints[endpoint] ??
defaultFileConfig.endpoints.default,
toolResource: _toolResource, toolResource: _toolResource,
fileConfig: fileConfig,
}); });
} catch (error) { } catch (error) {
console.error('file validation error', error); console.error('file validation error', error);

View file

@ -5,11 +5,9 @@ import type { SharePointFile } from '~/data-provider/Files/sharepoint';
interface UseSharePointFileHandlingProps { interface UseSharePointFileHandlingProps {
fileSetter?: any; fileSetter?: any;
toolResource?: string;
fileFilter?: (file: File) => boolean; fileFilter?: (file: File) => boolean;
additionalMetadata?: Record<string, string | undefined>; additionalMetadata?: Record<string, string | undefined>;
overrideEndpoint?: any;
overrideEndpointFileConfig?: any;
toolResource?: string;
} }
interface UseSharePointFileHandlingReturn { interface UseSharePointFileHandlingReturn {

View file

@ -1,6 +1,6 @@
import { getEndpointField } from 'librechat-data-provider';
import { useChatContext } from '~/Providers/ChatContext'; import { useChatContext } from '~/Providers/ChatContext';
import { useGetEndpointsQuery } from '~/data-provider'; import { useGetEndpointsQuery } from '~/data-provider';
import { getEndpointField } from '~/utils';
import useUserKey from './useUserKey'; import useUserKey from './useUserKey';
export default function useRequiresKey() { export default function useRequiresKey() {

View file

@ -1,15 +1,16 @@
import { useCallback } from 'react'; import { useCallback } from 'react';
import { useNavigate, useSearchParams } from 'react-router-dom'; import { useNavigate, useSearchParams } from 'react-router-dom';
import { useGetModelsQuery } from 'librechat-data-provider/react-query'; import { useGetModelsQuery } from 'librechat-data-provider/react-query';
import { useRecoilState, useRecoilValue, useSetRecoilState, useRecoilCallback } from 'recoil';
import { import {
Constants, Constants,
FileSources, FileSources,
EModelEndpoint, EModelEndpoint,
isParamEndpoint, isParamEndpoint,
getEndpointField,
LocalStorageKeys, LocalStorageKeys,
isAssistantsEndpoint, isAssistantsEndpoint,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import { useRecoilState, useRecoilValue, useSetRecoilState, useRecoilCallback } from 'recoil';
import type { import type {
TPreset, TPreset,
TSubmission, TSubmission,
@ -19,19 +20,18 @@ import type {
} from 'librechat-data-provider'; } from 'librechat-data-provider';
import type { AssistantListItem } from '~/common'; import type { AssistantListItem } from '~/common';
import { import {
getEndpointField, updateLastSelectedModel,
buildDefaultConvo, getDefaultModelSpec,
getDefaultEndpoint, getDefaultEndpoint,
getModelSpecPreset, getModelSpecPreset,
getDefaultModelSpec, buildDefaultConvo,
updateLastSelectedModel, logger,
} from '~/utils'; } from '~/utils';
import { useDeleteFilesMutation, useGetEndpointsQuery, useGetStartupConfig } from '~/data-provider'; import { useDeleteFilesMutation, useGetEndpointsQuery, useGetStartupConfig } from '~/data-provider';
import useAssistantListMap from './Assistants/useAssistantListMap'; import useAssistantListMap from './Assistants/useAssistantListMap';
import { useResetChatBadges } from './useChatBadges'; import { useResetChatBadges } from './useChatBadges';
import { useApplyModelSpecEffects } from './Agents'; import { useApplyModelSpecEffects } from './Agents';
import { usePauseGlobalAudio } from './Audio'; import { usePauseGlobalAudio } from './Audio';
import { logger } from '~/utils';
import store from '~/store'; import store from '~/store';
const useNewConvo = (index = 0) => { const useNewConvo = (index = 0) => {

View file

@ -716,6 +716,7 @@
"com_ui_attach_error_openai": "Cannot attach Assistant files to other endpoints", "com_ui_attach_error_openai": "Cannot attach Assistant files to other endpoints",
"com_ui_attach_error_size": "File size limit exceeded for endpoint:", "com_ui_attach_error_size": "File size limit exceeded for endpoint:",
"com_ui_attach_error_type": "Unsupported file type for endpoint:", "com_ui_attach_error_type": "Unsupported file type for endpoint:",
"com_ui_attach_error_disabled": "File uploads are disabled for this endpoint",
"com_ui_attach_remove": "Remove file", "com_ui_attach_remove": "Remove file",
"com_ui_attach_warn_endpoint": "Non-Assistant files may be ignored without a compatible tool", "com_ui_attach_warn_endpoint": "Non-Assistant files may be ignored without a compatible tool",
"com_ui_attachment": "Attachment", "com_ui_attachment": "Attachment",

View file

@ -1,11 +1,6 @@
import { EModelEndpoint } from 'librechat-data-provider'; import { EModelEndpoint, getEndpointField } from 'librechat-data-provider';
import type { TEndpointsConfig, TConfig } from 'librechat-data-provider'; import type { TEndpointsConfig, TConfig } from 'librechat-data-provider';
import { import { getAvailableEndpoints, getEndpointsFilter, mapEndpoints } from './endpoints';
getEndpointField,
getAvailableEndpoints,
getEndpointsFilter,
mapEndpoints,
} from './endpoints';
const mockEndpointsConfig: TEndpointsConfig = { const mockEndpointsConfig: TEndpointsConfig = {
[EModelEndpoint.openAI]: { type: undefined, iconURL: 'openAI_icon.png', order: 0 }, [EModelEndpoint.openAI]: { type: undefined, iconURL: 'openAI_icon.png', order: 0 },

View file

@ -4,6 +4,7 @@ import {
defaultEndpoints, defaultEndpoints,
modularEndpoints, modularEndpoints,
LocalStorageKeys, LocalStorageKeys,
getEndpointField,
isAgentsEndpoint, isAgentsEndpoint,
isAssistantsEndpoint, isAssistantsEndpoint,
} from 'librechat-data-provider'; } from 'librechat-data-provider';
@ -58,24 +59,6 @@ export const getAvailableEndpoints = (
return availableEndpoints; return availableEndpoints;
}; };
/** Get the specified field from the endpoint config */
export function getEndpointField<K extends keyof t.TConfig>(
endpointsConfig: t.TEndpointsConfig | undefined | null,
endpoint: EModelEndpoint | string | null | undefined,
property: K,
): t.TConfig[K] | undefined {
if (!endpointsConfig || endpoint === null || endpoint === undefined) {
return undefined;
}
const config = endpointsConfig[endpoint];
if (!config) {
return undefined;
}
return config[property];
}
export function mapEndpoints(endpointsConfig: t.TEndpointsConfig) { export function mapEndpoints(endpointsConfig: t.TEndpointsConfig) {
const filter = getEndpointsFilter(endpointsConfig); const filter = getEndpointsFilter(endpointsConfig);
return getAvailableEndpoints(filter, endpointsConfig).sort( return getAvailableEndpoints(filter, endpointsConfig).sort(

View file

@ -235,7 +235,13 @@ export const validateFiles = ({
toolResource?: string; toolResource?: string;
fileConfig: FileConfig | null; fileConfig: FileConfig | null;
}) => { }) => {
const { fileLimit, fileSizeLimit, totalSizeLimit, supportedMimeTypes } = endpointFileConfig; const { fileLimit, fileSizeLimit, totalSizeLimit, supportedMimeTypes, disabled } =
endpointFileConfig;
/** Block all uploads if the endpoint is explicitly disabled */
if (disabled === true) {
setError('com_ui_attach_error_disabled');
return false;
}
const existingFiles = Array.from(files.values()); const existingFiles = Array.from(files.values());
const incomingTotalSize = fileList.reduce((total, file) => total + file.size, 0); const incomingTotalSize = fileList.reduce((total, file) => total + file.size, 0);
if (incomingTotalSize === 0) { if (incomingTotalSize === 0) {

View file

@ -40,7 +40,6 @@ jest.mock('@librechat/data-schemas', () => ({
jest.mock('~/utils', () => ({ jest.mock('~/utils', () => ({
isEnabled: jest.fn((value) => value === 'true'), isEnabled: jest.fn((value) => value === 'true'),
normalizeEndpointName: jest.fn((name) => name),
})); }));
describe('getTransactionsConfig', () => { describe('getTransactionsConfig', () => {

View file

@ -1,8 +1,12 @@
import { logger } from '@librechat/data-schemas'; import { logger } from '@librechat/data-schemas';
import { EModelEndpoint, removeNullishValues } from 'librechat-data-provider'; import {
EModelEndpoint,
removeNullishValues,
normalizeEndpointName,
} from 'librechat-data-provider';
import type { TCustomConfig, TEndpoint, TTransactionsConfig } from 'librechat-data-provider'; import type { TCustomConfig, TEndpoint, TTransactionsConfig } from 'librechat-data-provider';
import type { AppConfig } from '@librechat/data-schemas'; import type { AppConfig } from '@librechat/data-schemas';
import { isEnabled, normalizeEndpointName } from '~/utils'; import { isEnabled } from '~/utils';
/** /**
* Retrieves the balance configuration object * Retrieves the balance configuration object

View file

@ -1,7 +1,7 @@
import { EModelEndpoint, extractEnvVariable } from 'librechat-data-provider'; import { EModelEndpoint, extractEnvVariable, normalizeEndpointName } from 'librechat-data-provider';
import type { TCustomEndpoints, TEndpoint, TConfig } from 'librechat-data-provider'; import type { TCustomEndpoints, TEndpoint, TConfig } from 'librechat-data-provider';
import type { TCustomEndpointsConfig } from '~/types/endpoints'; import type { TCustomEndpointsConfig } from '~/types/endpoints';
import { isUserProvided, normalizeEndpointName } from '~/utils'; import { isUserProvided } from '~/utils';
/** /**
* Load config endpoints from the cached configuration object * Load config endpoints from the cached configuration object

View file

@ -9,16 +9,19 @@ import { validateAudio } from '~/files/validation';
* Encodes and formats audio files for different providers * Encodes and formats audio files for different providers
* @param req - The request object * @param req - The request object
* @param files - Array of audio files * @param files - Array of audio files
* @param provider - The provider to format for (currently only google is supported) * @param params - Object containing provider and optional endpoint
* @param params.provider - The provider to format for (currently only google is supported)
* @param params.endpoint - Optional endpoint name for file config lookup
* @param getStrategyFunctions - Function to get strategy functions * @param getStrategyFunctions - Function to get strategy functions
* @returns Promise that resolves to audio and file metadata * @returns Promise that resolves to audio and file metadata
*/ */
export async function encodeAndFormatAudios( export async function encodeAndFormatAudios(
req: ServerRequest, req: ServerRequest,
files: IMongoFile[], files: IMongoFile[],
provider: Providers, params: { provider: Providers; endpoint?: string },
getStrategyFunctions: (source: string) => StrategyFunctions, getStrategyFunctions: (source: string) => StrategyFunctions,
): Promise<AudioResult> { ): Promise<AudioResult> {
const { provider, endpoint } = params;
if (!files?.length) { if (!files?.length) {
return { audios: [], files: [] }; return { audios: [], files: [] };
} }
@ -54,7 +57,10 @@ export async function encodeAndFormatAudios(
const audioBuffer = Buffer.from(content, 'base64'); const audioBuffer = Buffer.from(content, 'base64');
/** Extract configured file size limit from fileConfig for this endpoint */ /** Extract configured file size limit from fileConfig for this endpoint */
const configuredFileSizeLimit = getConfiguredFileSizeLimit(req, provider); const configuredFileSizeLimit = getConfiguredFileSizeLimit(req, {
provider,
endpoint,
});
const validation = await validateAudio( const validation = await validateAudio(
audioBuffer, audioBuffer,

View file

@ -31,14 +31,16 @@ describe('encodeAndFormatDocuments - fileConfig integration', () => {
beforeEach(() => { beforeEach(() => {
jest.clearAllMocks(); jest.clearAllMocks();
/** Default mock implementation for getConfiguredFileSizeLimit */ /** Default mock implementation for getConfiguredFileSizeLimit */
mockedGetConfiguredFileSizeLimit.mockImplementation((req, provider) => { mockedGetConfiguredFileSizeLimit.mockImplementation((req, params) => {
if (!req.config?.fileConfig) { if (!req.config?.fileConfig) {
return undefined; return undefined;
} }
const { provider, endpoint } = params;
const lookupKey = endpoint ?? provider;
const fileConfig = req.config.fileConfig; const fileConfig = req.config.fileConfig;
const endpoints = fileConfig.endpoints; const endpoints = fileConfig.endpoints;
if (endpoints?.[provider]) { if (endpoints?.[lookupKey]) {
const limit = endpoints[provider].fileSizeLimit; const limit = endpoints[lookupKey].fileSizeLimit;
return limit !== undefined ? mbToBytes(limit) : undefined; return limit !== undefined ? mbToBytes(limit) : undefined;
} }
if (endpoints?.default) { if (endpoints?.default) {

View file

@ -14,16 +14,20 @@ import { validatePdf } from '~/files/validation';
* Processes and encodes document files for various providers * Processes and encodes document files for various providers
* @param req - Express request object * @param req - Express request object
* @param files - Array of file objects to process * @param files - Array of file objects to process
* @param provider - The provider name * @param params - Object containing provider, endpoint, and other options
* @param params.provider - The provider name
* @param params.endpoint - Optional endpoint name for file config lookup
* @param params.useResponsesApi - Whether to use responses API format
* @param getStrategyFunctions - Function to get strategy functions * @param getStrategyFunctions - Function to get strategy functions
* @returns Promise that resolves to documents and file metadata * @returns Promise that resolves to documents and file metadata
*/ */
export async function encodeAndFormatDocuments( export async function encodeAndFormatDocuments(
req: ServerRequest, req: ServerRequest,
files: IMongoFile[], files: IMongoFile[],
{ provider, useResponsesApi }: { provider: Providers; useResponsesApi?: boolean }, params: { provider: Providers; endpoint?: string; useResponsesApi?: boolean },
getStrategyFunctions: (source: string) => StrategyFunctions, getStrategyFunctions: (source: string) => StrategyFunctions,
): Promise<DocumentResult> { ): Promise<DocumentResult> {
const { provider, endpoint, useResponsesApi } = params;
if (!files?.length) { if (!files?.length) {
return { documents: [], files: [] }; return { documents: [], files: [] };
} }
@ -68,7 +72,10 @@ export async function encodeAndFormatDocuments(
const pdfBuffer = Buffer.from(content, 'base64'); const pdfBuffer = Buffer.from(content, 'base64');
/** Extract configured file size limit from fileConfig for this endpoint */ /** Extract configured file size limit from fileConfig for this endpoint */
const configuredFileSizeLimit = getConfiguredFileSizeLimit(req, provider); const configuredFileSizeLimit = getConfiguredFileSizeLimit(req, {
provider,
endpoint,
});
const validation = await validatePdf( const validation = await validatePdf(
pdfBuffer, pdfBuffer,

View file

@ -1,24 +1,33 @@
import getStream from 'get-stream'; import getStream from 'get-stream';
import { Providers } from '@librechat/agents'; import { Providers } from '@librechat/agents';
import { FileSources, mergeFileConfig } from 'librechat-data-provider'; import { FileSources, mergeFileConfig, getEndpointFileConfig } from 'librechat-data-provider';
import type { IMongoFile } from '@librechat/data-schemas'; import type { IMongoFile } from '@librechat/data-schemas';
import type { ServerRequest, StrategyFunctions, ProcessedFile } from '~/types'; import type { ServerRequest, StrategyFunctions, ProcessedFile } from '~/types';
/** /**
* Extracts the configured file size limit for a specific provider from fileConfig * Extracts the configured file size limit for a specific provider from fileConfig
* @param req - The server request object containing config * @param req - The server request object containing config
* @param provider - The provider to get the limit for * @param params - Object containing provider and optional endpoint
* @param params.provider - The provider to get the limit for
* @param params.endpoint - Optional endpoint name for lookup
* @returns The configured file size limit in bytes, or undefined if not configured * @returns The configured file size limit in bytes, or undefined if not configured
*/ */
export const getConfiguredFileSizeLimit = ( export const getConfiguredFileSizeLimit = (
req: ServerRequest, req: ServerRequest,
provider: Providers, params: {
provider: Providers;
endpoint?: string;
},
): number | undefined => { ): number | undefined => {
if (!req.config?.fileConfig) { if (!req.config?.fileConfig) {
return undefined; return undefined;
} }
const { provider, endpoint } = params;
const fileConfig = mergeFileConfig(req.config.fileConfig); const fileConfig = mergeFileConfig(req.config.fileConfig);
const endpointConfig = fileConfig.endpoints[provider] ?? fileConfig.endpoints.default; const endpointConfig = getEndpointFileConfig({
fileConfig,
endpoint: endpoint ?? provider,
});
return endpointConfig?.fileSizeLimit; return endpointConfig?.fileSizeLimit;
}; };

View file

@ -9,16 +9,19 @@ import { validateVideo } from '~/files/validation';
* Encodes and formats video files for different providers * Encodes and formats video files for different providers
* @param req - The request object * @param req - The request object
* @param files - Array of video files * @param files - Array of video files
* @param provider - The provider to format for * @param params - Object containing provider and optional endpoint
* @param params.provider - The provider to format for
* @param params.endpoint - Optional endpoint name for file config lookup
* @param getStrategyFunctions - Function to get strategy functions * @param getStrategyFunctions - Function to get strategy functions
* @returns Promise that resolves to videos and file metadata * @returns Promise that resolves to videos and file metadata
*/ */
export async function encodeAndFormatVideos( export async function encodeAndFormatVideos(
req: ServerRequest, req: ServerRequest,
files: IMongoFile[], files: IMongoFile[],
provider: Providers, params: { provider: Providers; endpoint?: string },
getStrategyFunctions: (source: string) => StrategyFunctions, getStrategyFunctions: (source: string) => StrategyFunctions,
): Promise<VideoResult> { ): Promise<VideoResult> {
const { provider, endpoint } = params;
if (!files?.length) { if (!files?.length) {
return { videos: [], files: [] }; return { videos: [], files: [] };
} }
@ -54,7 +57,10 @@ export async function encodeAndFormatVideos(
const videoBuffer = Buffer.from(content, 'base64'); const videoBuffer = Buffer.from(content, 'base64');
/** Extract configured file size limit from fileConfig for this endpoint */ /** Extract configured file size limit from fileConfig for this endpoint */
const configuredFileSizeLimit = getConfiguredFileSizeLimit(req, provider); const configuredFileSizeLimit = getConfiguredFileSizeLimit(req, {
provider,
endpoint,
});
const validation = await validateVideo( const validation = await validateVideo(
videoBuffer, videoBuffer,

View file

@ -0,0 +1,692 @@
import { Types } from 'mongoose';
import { Providers } from '@librechat/agents';
import { EModelEndpoint } from 'librechat-data-provider';
import type { IMongoFile } from '@librechat/data-schemas';
import type { ServerRequest } from '~/types';
import { filterFilesByEndpointConfig } from './filter';
describe('filterFilesByEndpointConfig', () => {
/** Helper to create a mock file */
const createMockFile = (filename: string): IMongoFile =>
({
_id: new Types.ObjectId(),
user: new Types.ObjectId(),
file_id: new Types.ObjectId().toString(),
filename,
type: 'application/pdf',
bytes: 1024,
object: 'file',
usage: 0,
source: 'test',
filepath: `/test/${filename}`,
createdAt: new Date(),
updatedAt: new Date(),
}) as unknown as IMongoFile;
describe('when files are disabled for endpoint', () => {
it('should return empty array when endpoint has disabled: true', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test1.pdf'), createMockFile('test2.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual([]);
});
it('should return empty array when default endpoint has disabled: true and provider not found', () => {
const req = {
config: {
fileConfig: {
endpoints: {
default: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual([]);
});
it('should return empty array for disabled Anthropic endpoint', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.ANTHROPIC]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('doc.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.ANTHROPIC,
});
expect(result).toEqual([]);
});
it('should return empty array for disabled Google endpoint', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.GOOGLE]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('video.mp4')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.GOOGLE,
});
expect(result).toEqual([]);
});
});
describe('when files are enabled for endpoint', () => {
it('should return all files when endpoint has disabled: false', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test1.pdf'), createMockFile('test2.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual(files);
});
it('should return all files when endpoint config exists but disabled is not set', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
fileSizeLimit: 10,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual(files);
});
it('should return all files when no fileConfig exists', () => {
const req = {} as ServerRequest;
const files = [createMockFile('test1.pdf'), createMockFile('test2.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual(files);
});
it('should return all files when endpoint not in config and no default', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.ANTHROPIC]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/** OpenAI not configured, should use base defaults which allow files */
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual(files);
});
});
describe('custom endpoint configuration', () => {
it('should use direct endpoint lookup when endpointType is custom', () => {
const req = {
config: {
fileConfig: {
endpoints: {
ollama: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual([]);
});
it('should use normalized endpoint lookup for custom endpoints', () => {
const req = {
config: {
fileConfig: {
endpoints: {
ollama: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/** Test with non-normalized endpoint name (e.g., "Ollama" vs "ollama") */
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'Ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual([]);
});
it('should fallback to "custom" config when specific custom endpoint not found', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[EModelEndpoint.custom]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'unknownCustomEndpoint',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual([]);
});
it('should return files when custom endpoint has disabled: false', () => {
const req = {
config: {
fileConfig: {
endpoints: {
ollama: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test1.pdf'), createMockFile('test2.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual(files);
});
it('should use agents config as fallback for custom endpoints', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[EModelEndpoint.agents]: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/**
* Lookup order for custom endpoint: explicitConfig -> custom -> agents -> default
* Should find and use agents config
*/
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual(files);
});
it('should fallback to default when agents is not configured for custom endpoint', () => {
const req = {
config: {
fileConfig: {
endpoints: {
/** Only default configured, no agents or custom */
default: {
disabled: false,
fileLimit: 10,
fileSizeLimit: 20,
totalSizeLimit: 50,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/**
* Lookup order: explicitConfig -> custom -> agents -> default
* Since none of first three exist, should fall back to default
*/
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual(files);
});
it('should use default when agents is not configured for custom endpoint', () => {
const req = {
config: {
fileConfig: {
endpoints: {
/** NO agents config - should skip to default */
default: {
disabled: false,
fileLimit: 15,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/**
* Lookup order: explicitConfig -> custom -> agents -> default
* Since agents is not configured, should fall back to default
*/
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual(files);
});
it('should block files when agents is disabled for unconfigured custom endpoint', () => {
const req = {
config: {
fileConfig: {
endpoints: {
agents: {
disabled: true,
},
default: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/**
* Lookup order: explicitConfig -> custom -> agents -> default
* Should use agents config which is disabled: true
*/
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual([]);
});
it('should prioritize specific custom endpoint over generic custom config', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[EModelEndpoint.custom]: {
disabled: false,
},
ollama: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/** Should use ollama config (disabled: true), not custom config (disabled: false) */
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual([]);
});
it('should handle case-insensitive custom endpoint names', () => {
const req = {
config: {
fileConfig: {
endpoints: {
ollama: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/** Test various case combinations */
const result1 = filterFilesByEndpointConfig(req, {
files,
endpoint: 'OLLAMA',
endpointType: EModelEndpoint.custom,
});
const result2 = filterFilesByEndpointConfig(req, {
files,
endpoint: 'OlLaMa',
endpointType: EModelEndpoint.custom,
});
expect(result1).toEqual([]);
expect(result2).toEqual([]);
});
it('should work without endpointType for standard endpoints but require it for custom', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: true,
},
ollama: {
disabled: true,
},
default: {
disabled: false,
fileLimit: 10,
fileSizeLimit: 20,
totalSizeLimit: 50,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
/** Standard endpoint works without endpointType */
const openaiResult = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(openaiResult).toEqual([]);
/** Custom endpoint with endpointType uses specific config */
const customWithTypeResult = filterFilesByEndpointConfig(req, {
files,
endpoint: 'ollama',
endpointType: EModelEndpoint.custom,
});
expect(customWithTypeResult).toEqual([]);
/** Custom endpoint without endpointType tries direct lookup, falls back to default */
const customWithoutTypeResult = filterFilesByEndpointConfig(req, {
files,
endpoint: 'unknownCustom',
});
expect(customWithoutTypeResult).toEqual(files);
});
});
describe('edge cases', () => {
it('should return empty array when files input is undefined', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const result = filterFilesByEndpointConfig(req, {
files: undefined,
endpoint: Providers.OPENAI,
});
expect(result).toEqual([]);
});
it('should return empty array when files input is empty array', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const result = filterFilesByEndpointConfig(req, {
files: [],
endpoint: Providers.OPENAI,
});
expect(result).toEqual([]);
});
it('should handle custom provider strings', () => {
const req = {
config: {
fileConfig: {
endpoints: {
customProvider: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('test.pdf')];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: 'customProvider',
endpointType: EModelEndpoint.custom,
});
expect(result).toEqual([]);
});
});
describe('bypass scenarios from bug report', () => {
it('should block files when switching from enabled to disabled endpoint', () => {
/**
* Scenario: User attaches files under Anthropic (enabled),
* then switches to OpenAI (disabled)
*/
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: true,
},
[Providers.ANTHROPIC]: {
disabled: false,
},
},
},
},
} as unknown as ServerRequest;
const files = [createMockFile('document.pdf')];
/** Files were attached under Anthropic */
const anthropicResult = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.ANTHROPIC,
});
expect(anthropicResult).toEqual(files);
/** User switches to OpenAI - files should be filtered out */
const openaiResult = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(openaiResult).toEqual([]);
});
it('should prevent drag-and-drop bypass by filtering at agent initialization', () => {
/**
* Scenario: User drags file into disabled endpoint
* Server processes it but filter should remove it
*/
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const draggedFiles = [createMockFile('dragged.pdf')];
const result = filterFilesByEndpointConfig(req, {
files: draggedFiles,
endpoint: Providers.OPENAI,
});
expect(result).toEqual([]);
});
it('should filter multiple files when endpoint disabled', () => {
const req = {
config: {
fileConfig: {
endpoints: {
[Providers.OPENAI]: {
disabled: true,
},
},
},
},
} as unknown as ServerRequest;
const files = [
createMockFile('file1.pdf'),
createMockFile('file2.pdf'),
createMockFile('file3.pdf'),
];
const result = filterFilesByEndpointConfig(req, {
files,
endpoint: Providers.OPENAI,
});
expect(result).toEqual([]);
});
});
});

View file

@ -0,0 +1,44 @@
import { getEndpointFileConfig, mergeFileConfig } from 'librechat-data-provider';
import type { IMongoFile } from '@librechat/data-schemas';
import type { ServerRequest } from '~/types';
/**
* Filters out files if the endpoint/provider has file uploads disabled
* @param req - The server request object containing config
* @param params - Object containing files, endpoint, and endpointType
* @param params.files - Array of processed file documents from MongoDB
* @param params.endpoint - The endpoint name to check configuration for
* @param params.endpointType - The endpoint type to check configuration for
* @returns Filtered array of files (empty if disabled)
*/
export function filterFilesByEndpointConfig(
req: ServerRequest,
params: {
files: IMongoFile[] | undefined;
endpoint?: string | null;
endpointType?: string | null;
},
): IMongoFile[] {
const { files, endpoint, endpointType } = params;
if (!files || files.length === 0) {
return [];
}
const fileConfig = mergeFileConfig(req.config?.fileConfig);
const endpointFileConfig = getEndpointFileConfig({
fileConfig,
endpoint,
endpointType,
});
/**
* If endpoint has files explicitly disabled, filter out all files
* Only filter if disabled is explicitly set to true
*/
if (endpointFileConfig?.disabled === true) {
return [];
}
return files;
}

View file

@ -1,6 +1,7 @@
export * from './audio'; export * from './audio';
export * from './context'; export * from './context';
export * from './encode'; export * from './encode';
export * from './filter';
export * from './mistral/crud'; export * from './mistral/crud';
export * from './ocr'; export * from './ocr';
export * from './parse'; export * from './parse';

View file

@ -1,4 +1,3 @@
import { Providers } from '@librechat/agents';
import { AuthType } from 'librechat-data-provider'; import { AuthType } from 'librechat-data-provider';
/** /**
@ -49,11 +48,3 @@ export function optionalChainWithEmptyCheck(
} }
return values[values.length - 1]; return values[values.length - 1];
} }
/**
* Normalize the endpoint name to system-expected value.
* @param name
*/
export function normalizeEndpointName(name = ''): string {
return name.toLowerCase() === Providers.OLLAMA ? Providers.OLLAMA : name;
}

View file

@ -1,12 +1,12 @@
import { z } from 'zod'; import { z } from 'zod';
import type { ZodError } from 'zod'; import type { ZodError } from 'zod';
import type { TModelsConfig } from './types'; import type { TEndpointsConfig, TModelsConfig, TConfig } from './types';
import { EModelEndpoint, eModelEndpointSchema } from './schemas'; import { EModelEndpoint, eModelEndpointSchema } from './schemas';
import { specsConfigSchema, TSpecsConfig } from './models'; import { specsConfigSchema, TSpecsConfig } from './models';
import { fileConfigSchema } from './file-config'; import { fileConfigSchema } from './file-config';
import { apiBaseUrl } from './api-endpoints';
import { FileSources } from './types/files'; import { FileSources } from './types/files';
import { MCPServersSchema } from './mcp'; import { MCPServersSchema } from './mcp';
import { apiBaseUrl } from './api-endpoints';
export const defaultSocialLogins = ['google', 'facebook', 'openid', 'github', 'discord', 'saml']; export const defaultSocialLogins = ['google', 'facebook', 'openid', 'github', 'discord', 'saml'];
@ -1737,3 +1737,24 @@ export const specialVariables = {
}; };
export type TSpecialVarLabel = `com_ui_special_var_${keyof typeof specialVariables}`; export type TSpecialVarLabel = `com_ui_special_var_${keyof typeof specialVariables}`;
/**
* Retrieves a specific field from the endpoints configuration for a given endpoint key.
* Does not infer or default any endpoint type when absent.
*/
export function getEndpointField<
K extends TConfig[keyof TConfig] extends never ? never : keyof TConfig,
>(
endpointsConfig: TEndpointsConfig | undefined | null,
endpoint: EModelEndpoint | string | null | undefined,
property: K,
): TConfig[K] | undefined {
if (!endpointsConfig || endpoint === null || endpoint === undefined) {
return undefined;
}
const config = endpointsConfig[endpoint];
if (!config) {
return undefined;
}
return config[property];
}

File diff suppressed because it is too large Load diff

View file

@ -1,6 +1,7 @@
import { z } from 'zod'; import { z } from 'zod';
import { EModelEndpoint } from './schemas';
import type { EndpointFileConfig, FileConfig } from './types/files'; import type { EndpointFileConfig, FileConfig } from './types/files';
import { EModelEndpoint, isAgentsEndpoint, isDocumentSupportedProvider } from './schemas';
import { normalizeEndpointName } from './utils';
export const supportsFiles = { export const supportsFiles = {
[EModelEndpoint.openAI]: true, [EModelEndpoint.openAI]: true,
@ -331,9 +332,146 @@ export const convertStringsToRegex = (patterns: string[]): RegExp[] =>
return acc; return acc;
}, []); }, []);
/**
* Gets the appropriate endpoint file configuration with standardized lookup logic.
*
* @param params - Object containing fileConfig, endpoint, and optional conversationEndpoint
* @param params.fileConfig - The merged file configuration
* @param params.endpoint - The endpoint name to look up
* @param params.conversationEndpoint - Optional conversation endpoint for additional context
* @returns The endpoint file configuration or undefined
*/
/**
* Merges an endpoint config with the default config to ensure all fields are populated.
* For document-supported providers, uses the comprehensive MIME type list (includes videos/audio).
*/
function mergeWithDefault(
endpointConfig: EndpointFileConfig,
defaultConfig: EndpointFileConfig,
endpoint?: string | null,
): EndpointFileConfig {
/** Use comprehensive MIME types for document-supported providers */
const defaultMimeTypes = isDocumentSupportedProvider(endpoint)
? supportedMimeTypes
: defaultConfig.supportedMimeTypes;
return {
disabled: endpointConfig.disabled ?? defaultConfig.disabled,
fileLimit: endpointConfig.fileLimit ?? defaultConfig.fileLimit,
fileSizeLimit: endpointConfig.fileSizeLimit ?? defaultConfig.fileSizeLimit,
totalSizeLimit: endpointConfig.totalSizeLimit ?? defaultConfig.totalSizeLimit,
supportedMimeTypes: endpointConfig.supportedMimeTypes ?? defaultMimeTypes,
};
}
export function getEndpointFileConfig(params: {
fileConfig?: FileConfig | null;
endpoint?: string | null;
endpointType?: string | null;
}): EndpointFileConfig {
const { fileConfig: mergedFileConfig, endpoint, endpointType } = params;
if (!mergedFileConfig?.endpoints) {
return fileConfig.endpoints.default;
}
/** Compute an effective default by merging user-configured default over the base default */
const baseDefaultConfig = fileConfig.endpoints.default;
const userDefaultConfig = mergedFileConfig.endpoints.default;
const defaultConfig = userDefaultConfig
? mergeWithDefault(userDefaultConfig, baseDefaultConfig, 'default')
: baseDefaultConfig;
const normalizedEndpoint = normalizeEndpointName(endpoint ?? '');
const standardEndpoints = new Set([
'default',
EModelEndpoint.agents,
EModelEndpoint.assistants,
EModelEndpoint.azureAssistants,
EModelEndpoint.openAI,
EModelEndpoint.azureOpenAI,
EModelEndpoint.anthropic,
EModelEndpoint.google,
EModelEndpoint.bedrock,
]);
const normalizedEndpointType = normalizeEndpointName(endpointType ?? '');
const isCustomEndpoint =
endpointType === EModelEndpoint.custom ||
(!standardEndpoints.has(normalizedEndpointType) &&
normalizedEndpoint &&
!standardEndpoints.has(normalizedEndpoint));
if (isCustomEndpoint) {
/** 1. Check direct endpoint lookup (could be normalized or not) */
if (endpoint && mergedFileConfig.endpoints[endpoint]) {
return mergeWithDefault(mergedFileConfig.endpoints[endpoint], defaultConfig, endpoint);
}
/** 2. Check normalized endpoint lookup (skip standard endpoint keys) */
for (const key in mergedFileConfig.endpoints) {
if (!standardEndpoints.has(key) && normalizeEndpointName(key) === normalizedEndpoint) {
return mergeWithDefault(mergedFileConfig.endpoints[key], defaultConfig, key);
}
}
/** 3. Fallback to generic 'custom' config if any */
if (mergedFileConfig.endpoints[EModelEndpoint.custom]) {
return mergeWithDefault(
mergedFileConfig.endpoints[EModelEndpoint.custom],
defaultConfig,
endpoint,
);
}
/** 4. Fallback to 'agents' (all custom endpoints are non-assistants) */
if (mergedFileConfig.endpoints[EModelEndpoint.agents]) {
return mergeWithDefault(
mergedFileConfig.endpoints[EModelEndpoint.agents],
defaultConfig,
endpoint,
);
}
/** 5. Fallback to default */
return defaultConfig;
}
/** Check endpointType first (most reliable for standard endpoints) */
if (endpointType && mergedFileConfig.endpoints[endpointType]) {
return mergeWithDefault(mergedFileConfig.endpoints[endpointType], defaultConfig, endpointType);
}
/** Check direct endpoint lookup */
if (endpoint && mergedFileConfig.endpoints[endpoint]) {
return mergeWithDefault(mergedFileConfig.endpoints[endpoint], defaultConfig, endpoint);
}
/** Check normalized endpoint */
if (normalizedEndpoint && mergedFileConfig.endpoints[normalizedEndpoint]) {
return mergeWithDefault(
mergedFileConfig.endpoints[normalizedEndpoint],
defaultConfig,
normalizedEndpoint,
);
}
/** Fallback to agents if endpoint is explicitly agents */
const isAgents = isAgentsEndpoint(normalizedEndpointType || normalizedEndpoint);
if (isAgents && mergedFileConfig.endpoints[EModelEndpoint.agents]) {
return mergeWithDefault(
mergedFileConfig.endpoints[EModelEndpoint.agents],
defaultConfig,
EModelEndpoint.agents,
);
}
/** Return default config */
return defaultConfig;
}
export function mergeFileConfig(dynamic: z.infer<typeof fileConfigSchema> | undefined): FileConfig { export function mergeFileConfig(dynamic: z.infer<typeof fileConfigSchema> | undefined): FileConfig {
const mergedConfig: FileConfig = { const mergedConfig: FileConfig = {
...fileConfig, ...fileConfig,
endpoints: {
...fileConfig.endpoints,
},
ocr: { ocr: {
...fileConfig.ocr, ...fileConfig.ocr,
supportedMimeTypes: fileConfig.ocr?.supportedMimeTypes || [], supportedMimeTypes: fileConfig.ocr?.supportedMimeTypes || [],
@ -398,8 +536,11 @@ export function mergeFileConfig(dynamic: z.infer<typeof fileConfigSchema> | unde
for (const key in dynamic.endpoints) { for (const key in dynamic.endpoints) {
const dynamicEndpoint = (dynamic.endpoints as Record<string, EndpointFileConfig>)[key]; const dynamicEndpoint = (dynamic.endpoints as Record<string, EndpointFileConfig>)[key];
/** Deep copy the base endpoint config if it exists to prevent mutation */
if (!mergedConfig.endpoints[key]) { if (!mergedConfig.endpoints[key]) {
mergedConfig.endpoints[key] = {}; mergedConfig.endpoints[key] = {};
} else {
mergedConfig.endpoints[key] = { ...mergedConfig.endpoints[key] };
} }
const mergedEndpoint = mergedConfig.endpoints[key]; const mergedEndpoint = mergedConfig.endpoints[key];
@ -428,6 +569,10 @@ export function mergeFileConfig(dynamic: z.infer<typeof fileConfigSchema> | unde
} }
}); });
if (dynamicEndpoint.disabled !== undefined) {
mergedEndpoint.disabled = dynamicEndpoint.disabled;
}
if (dynamicEndpoint.supportedMimeTypes) { if (dynamicEndpoint.supportedMimeTypes) {
mergedEndpoint.supportedMimeTypes = convertStringsToRegex( mergedEndpoint.supportedMimeTypes = convertStringsToRegex(
dynamicEndpoint.supportedMimeTypes as unknown as string[], dynamicEndpoint.supportedMimeTypes as unknown as string[],

View file

@ -52,3 +52,11 @@ export function extractEnvVariable(value: string) {
return result; return result;
} }
/**
* Normalize the endpoint name to system-expected value.
* @param name
*/
export function normalizeEndpointName(name = ''): string {
return name.toLowerCase() === 'ollama' ? 'ollama' : name;
}

View file

@ -1,15 +1,7 @@
import logger from '~/config/winston'; import logger from '~/config/winston';
import { EModelEndpoint } from 'librechat-data-provider'; import { EModelEndpoint, normalizeEndpointName } from 'librechat-data-provider';
import type { TCustomConfig } from 'librechat-data-provider'; import type { TCustomConfig } from 'librechat-data-provider';
/**
* Normalize the endpoint name to system-expected value.
* @param name
*/
function normalizeEndpointName(name = ''): string {
return name.toLowerCase() === 'ollama' ? 'ollama' : name;
}
/** /**
* Sets up Model Specs from the config (`librechat.yaml`) file. * Sets up Model Specs from the config (`librechat.yaml`) file.
* @param [endpoints] - The loaded custom configuration for endpoints. * @param [endpoints] - The loaded custom configuration for endpoints.