feat: add source to MongoFile schema, as well as BatchFile, and ExtendedFile types

Author: danny-avila

api/models/schema/fileSchema.js

@@ -1,3 +1,4 @@
+const { FileSources } = require('librechat-data-provider');
 const mongoose = require('mongoose');
 
 /**
@@ -12,6 +13,7 @@ const mongoose = require('mongoose');
  * @property {'file'} object - Type of object, always 'file'
  * @property {string} type - Type of file
  * @property {number} usage - Number of uses of the file
+ * @property {string} [source] - The source of the file
  * @property {number} [width] - Optional width of the file
  * @property {number} [height] - Optional height of the file
  * @property {Date} [expiresAt] - Optional height of the file
@@ -42,11 +44,6 @@ const fileSchema = mongoose.Schema(
       type: Number,
       required: true,
     },
-    usage: {
-      type: Number,
-      required: true,
-      default: 0,
-    },
     filename: {
       type: String,
       required: true,
@@ -64,6 +61,15 @@ const fileSchema = mongoose.Schema(
       type: String,
       required: true,
     },
+    usage: {
+      type: Number,
+      required: true,
+      default: 0,
+    },
+    source: {
+      type: String,
+      default: FileSources.local,
+    },
     width: Number,
     height: Number,
     expiresAt: {

client/src/common/types.ts

@@ -1,3 +1,4 @@
+import { FileSources } from 'librechat-data-provider';
 import type {
   TConversation,
   TMessage,
@@ -230,6 +231,7 @@ export interface ExtendedFile {
   size: number;
   preview: string;
   progress: number;
+  source?: FileSources;
 }
 
 export type ContextType = { navVisible: boolean; setNavVisible: (visible: boolean) => void };

client/src/components/Chat/Input/Files/Images.tsx

@@ -1,5 +1,6 @@
 import debounce from 'lodash/debounce';
 import { useState, useEffect, useCallback } from 'react';
+import { FileSources } from 'librechat-data-provider';
 import type { BatchFile } from 'librechat-data-provider';
 import { useDeleteFilesMutation } from '~/data-provider';
 import { useSetFilesToDelete } from '~/hooks';
@@ -70,13 +71,20 @@ export default function Images({
   }
 
   const deleteFile = (_file: ExtendedFile) => {
-    const { file_id, progress, temp_file_id = '', filepath = '' } = _file;
+    const {
+      file_id,
+      progress,
+      temp_file_id = '',
+      filepath = '',
+      source = FileSources.local,
+    } = _file;
     if (progress < 1) {
       return;
     }
-    const file = {
+    const file: BatchFile = {
       file_id,
       filepath,
+      source,
     };
 
     setFiles((currentFiles) => {

client/src/hooks/useFileHandling.ts

@@ -133,6 +133,7 @@ const useFileHandling = () => {
           height: data.height,
           width: data.width,
           filename: data.filename,
+          source: data.source,
         });
       }, 300);
     },

packages/data-provider/src/types/files.ts

@@ -51,6 +51,7 @@ export type DeleteFilesResponse = {
 export type BatchFile = {
   file_id: string;
   filepath: string;
+  source: FileSources;
 };
 
 export type DeleteFilesBody = {