| import { useState, useRef, useEffect } from 'react'; |
| import { EToolResources, mergeFileConfig, getEndpointFileConfig } from 'librechat-data-provider'; |
| import type { AssistantsEndpoint } from 'librechat-data-provider'; |
| import type { ExtendedFile } from '~/common'; |
| import FileRow from '~/components/Chat/Input/Files/FileRow'; |
| import { useGetFileConfig } from '~/data-provider'; |
| import { useFileHandling } from '~/hooks/Files'; |
| import { useChatContext } from '~/Providers'; |
| import { useLocalize } from '~/hooks'; |
|
|
| const tool_resource = EToolResources.code_interpreter; |
|
|
| export default function CodeFiles({ |
| endpoint, |
| assistant_id, |
| files: _files, |
| }: { |
| version: number | string; |
| endpoint: AssistantsEndpoint; |
| assistant_id: string; |
| files?: [string, ExtendedFile][]; |
| }) { |
| const localize = useLocalize(); |
| const { setFilesLoading } = useChatContext(); |
| const fileInputRef = useRef<HTMLInputElement>(null); |
| const [files, setFiles] = useState<Map<string, ExtendedFile>>(new Map()); |
| const { data: fileConfig = null } = useGetFileConfig({ |
| select: (data) => mergeFileConfig(data), |
| }); |
| const { handleFileChange } = useFileHandling({ |
| additionalMetadata: { assistant_id, tool_resource }, |
| fileSetter: setFiles, |
| }); |
|
|
| useEffect(() => { |
| if (_files) { |
| setFiles(new Map(_files)); |
| } |
| }, [_files]); |
|
|
| const endpointFileConfig = getEndpointFileConfig({ |
| fileConfig, |
| endpoint, |
| endpointType: endpoint, |
| }); |
| const isUploadDisabled = endpointFileConfig?.disabled ?? false; |
|
|
| if (isUploadDisabled) { |
| return null; |
| } |
|
|
| const handleButtonClick = () => { |
| |
| if (fileInputRef.current) { |
| fileInputRef.current.value = ''; |
| } |
| fileInputRef.current?.click(); |
| }; |
|
|
| return ( |
| <div className="mb-2 w-full"> |
| <div className="flex flex-col gap-4"> |
| <div className="rounded-lg text-xs text-text-secondary"> |
| {localize('com_assistants_code_interpreter_files')} |
| </div> |
| <FileRow |
| files={files} |
| setFiles={setFiles} |
| assistant_id={assistant_id} |
| tool_resource={tool_resource} |
| setFilesLoading={setFilesLoading} |
| Wrapper={({ children }) => <div className="flex flex-wrap gap-2">{children}</div>} |
| /> |
| <div> |
| <button |
| type="button" |
| disabled={!assistant_id} |
| className="btn btn-neutral border-token-border-light relative h-9 w-full rounded-lg font-medium" |
| onClick={handleButtonClick} |
| > |
| <div className="flex w-full items-center justify-center gap-2"> |
| <input |
| multiple={true} |
| type="file" |
| style={{ display: 'none' }} |
| tabIndex={-1} |
| ref={fileInputRef} |
| disabled={!assistant_id} |
| onChange={handleFileChange} |
| /> |
| {localize('com_ui_upload_files')} |
| </div> |
| </button> |
| </div> |
| </div> |
| </div> |
| ); |
| } |
|
|