@librechat/data-schemas 0.0.19 → 0.0.20

package/dist/index.cjs

@@ -219,6 +219,10 @@ const agentCategorySchema = new mongoose.Schema({
         default: true,
         index: true,
     },
+    custom: {
+        type: Boolean,
+        default: false,
+    },
 }, {
     timestamps: true,
 });
@@ -1811,7 +1815,7 @@ const transports$1 = [
         zippedArchive: true,
         maxSize: '20m',
         maxFiles: '14d',
-        format: fileFormat$1,
+        format: winston.format.combine(fileFormat$1, winston.format.json()),
     }),
 ];
 if (useDebugLogging$1) {
@@ -3545,6 +3549,7 @@ function createAgentCategoryMethods(mongoose) {
                         description: category.description || '',
                         order: category.order || index,
                         isActive: true,
+                        custom: category.custom || false,
                     },
                 },
                 upsert: true,
@@ -3609,60 +3614,96 @@ function createAgentCategoryMethods(mongoose) {
         return await AgentCategory.find({}).sort({ order: 1, label: 1 }).lean();
     }
     /**
-     * Ensure default categories exist, seed them if none are present
-     * @returns Promise<boolean> - true if categories were seeded, false if they already existed
+     * Ensure default categories exist and update them if they don't have localization keys
+     * @returns Promise<boolean> - true if categories were created/updated, false if no changes
      */
     async function ensureDefaultCategories() {
-        const existingCategories = await getAllCategories();
-        if (existingCategories.length > 0) {
-            return false; // Categories already exist
-        }
+        const AgentCategory = mongoose.models.AgentCategory;
         const defaultCategories = [
             {
                 value: 'general',
-                label: 'General',
-                description: 'General purpose agents for common tasks and inquiries',
+                label: 'com_agents_category_general',
+                description: 'com_agents_category_general_description',
                 order: 0,
             },
             {
                 value: 'hr',
-                label: 'Human Resources',
-                description: 'Agents specialized in HR processes, policies, and employee support',
+                label: 'com_agents_category_hr',
+                description: 'com_agents_category_hr_description',
                 order: 1,
             },
             {
                 value: 'rd',
-                label: 'Research & Development',
-                description: 'Agents focused on R&D processes, innovation, and technical research',
+                label: 'com_agents_category_rd',
+                description: 'com_agents_category_rd_description',
                 order: 2,
             },
             {
                 value: 'finance',
-                label: 'Finance',
-                description: 'Agents specialized in financial analysis, budgeting, and accounting',
+                label: 'com_agents_category_finance',
+                description: 'com_agents_category_finance_description',
                 order: 3,
             },
             {
                 value: 'it',
-                label: 'IT',
-                description: 'Agents for IT support, technical troubleshooting, and system administration',
+                label: 'com_agents_category_it',
+                description: 'com_agents_category_it_description',
                 order: 4,
             },
             {
                 value: 'sales',
-                label: 'Sales',
-                description: 'Agents focused on sales processes, customer relations.',
+                label: 'com_agents_category_sales',
+                description: 'com_agents_category_sales_description',
                 order: 5,
             },
             {
                 value: 'aftersales',
-                label: 'After Sales',
-                description: 'Agents specialized in post-sale support, maintenance, and customer service',
+                label: 'com_agents_category_aftersales',
+                description: 'com_agents_category_aftersales_description',
                 order: 6,
             },
         ];
-        await seedCategories(defaultCategories);
-        return true; // Categories were seeded
+        const existingCategories = await getAllCategories();
+        const existingCategoryMap = new Map(existingCategories.map((cat) => [cat.value, cat]));
+        const updates = [];
+        let created = 0;
+        for (const defaultCategory of defaultCategories) {
+            const existingCategory = existingCategoryMap.get(defaultCategory.value);
+            if (existingCategory) {
+                const isNotCustom = !existingCategory.custom;
+                const needsLocalization = !existingCategory.label.startsWith('com_');
+                if (isNotCustom && needsLocalization) {
+                    updates.push({
+                        value: defaultCategory.value,
+                        label: defaultCategory.label,
+                        description: defaultCategory.description,
+                    });
+                }
+            }
+            else {
+                await createCategory({
+                    ...defaultCategory,
+                    isActive: true,
+                    custom: false,
+                });
+                created++;
+            }
+        }
+        if (updates.length > 0) {
+            const bulkOps = updates.map((update) => ({
+                updateOne: {
+                    filter: { value: update.value, custom: { $ne: true } },
+                    update: {
+                        $set: {
+                            label: update.label,
+                            description: update.description,
+                        },
+                    },
+                },
+            }));
+            await AgentCategory.bulkWrite(bulkOps, { ordered: false });
+        }
+        return updates.length > 0 || created > 0;
     }
     return {
         getActiveCategories,