data-primals-engine 1.6.5 → 1.7.1

package/src/migrate.js

@@ -24,7 +24,7 @@ engine.start(port, async () => {
     const target = process.argv[3]; // Le nom du fichier de migration cible
 
     try {
-        const db = MongoDatabase;
+        const db = MongoDatabase();
         const { executedNames, allMigrationFiles } = await getMigrationStatus(db);
 
         switch (command) {

package/src/modules/assistant/assistant.js

@@ -8,7 +8,7 @@ import {Config} from "../../config.js";
 import { Event } from "../../events.js";
 import rateLimit from "express-rate-limit";
 import {generateLimiter} from "../user.js";
-import {assistantConfig, maxAIReflectiveSteps} from "../../constants.js";
+import {maxAIReflectiveSteps} from "../../constants.js";
 import i18n from "../../i18n.js";
 import {parseSafeJSON} from "../../core.js";
 
@@ -80,16 +80,15 @@ REGLE FONDATRICE : suis les règles et ne dévie pas du chemin.
 STYLE UTILISE : apporte l'information au plus rapide, sans détours, ni sollicitation à l'utilisateur, ou à des tiers.
 
 FORMAT DE RÉPONSE OBLIGATOIRE :
-Ta réponse DOIT être un objet JSON unique OU un tableau d'objets JSON. Chaque objet doit contenir exactement 2 champs :
+Ta réponse DOIT être un tableau d'objets JSON (Même pour une seule action). Chaque objet doit contenir exactement 2 champs :
 1. "action" (string) 
 2. "params" (object)
 
-- Pour une action simple, retourne un objet : { "action": "...", "params": {...} }
-- Pour enchaîner plusieurs actions, retourne un tableau :
-  [
-    { "action": "search_models", "params": {"query": "..."} },
-    { "action": "displayMessage", "params": { "message": "Je cherche les modèles..." } }
-  ]
+Exemple :
+[
+  { "action": "search_models", "params": {"query": "..."} },
+  { "action": "displayMessage", "params": { "message": "Je cherche les modèles..." } }
+]
 
 Tu as accès aux outils et actions suivants.
 
@@ -529,7 +528,7 @@ export async function handleChatRequest(params, user, sendEvent = null) {
         }
         
         const toolResults = [];
-        let finalActionResult = null;
+        const finalActionResults = [];
 
         for (const command of commands) {
             logger.debug(`[Assistant] Action décidée par l'IA: ${command.action}`, command);
@@ -554,27 +553,31 @@ export async function handleChatRequest(params, user, sendEvent = null) {
             const res = await Event.Trigger('OnChatAction', 'event', 'user', action, parsedParams, command, llmOptions, user, params);
 
             if (res) {
-                // On a trouvé une action finale. On la stocke et on arrête de chercher.
-                finalActionResult = res;
-                break; // Sort de la boucle des commandes
+                if (res.isToolResult) {
+                    toolResults.push({ action, result: res.toolResult || res.displayMessage });
+                } else {
+                    // On a trouvé une action finale. On la stocke.
+                    finalActionResults.push(res);
+                }
             } else {
                 // Si l'action n'est reconnue par aucune des logiques ci-dessus
                 logger.warn(`[Assistant] Action non reconnue reçue de l'IA: ${action}`);
-                finalActionResult = {
+                finalActionResults.push({
                     success: true,
                     displayMessage: i18n.t('assistant.unknownAction', "Désolé, je ne comprends pas la commande '{{action}}'.", { action })
-                };
-                break; // Sortir de la boucle des commandes
+                });
             }
         }
 
-        // Si une action finale a été exécutée, on retourne son résultat.
-        if (finalActionResult) {
+        // Si une ou plusieurs actions finales ont été exécutées, on retourne le résultat.
+        if (finalActionResults.length > 0) {
+            // Rétrocompatibilité : si un seul résultat, on renvoie l'objet directement. Sinon, on renvoie un tableau 'results'.
+            const response = finalActionResults.length === 1 ? finalActionResults[0] : { success: true, results: finalActionResults };
             if (sendEvent) {
-                sendEvent('final_result', finalActionResult);
+                sendEvent('final_result', response);
                 return;
             }
-            return finalActionResult;
+            return response;
         }
 
         // Si on a uniquement des résultats d'outils, on les ajoute à l'historique et on continue la boucle.
@@ -673,7 +676,14 @@ async function handleFinalChatAction(action, params, parsedResponse, llmOptions,
 
     // Actions nécessitant une confirmation de l'utilisateur
     if (['post', 'update', 'delete'].includes(action)) {
-        return { success: true, displayMessage: i18n.t('assistant.confirmActionPrompt', "Veuillez confirmer l'action suivante :"), confirmationRequest: parsedResponse };
+        return {
+            success: true,
+            displayMessage: i18n.t('assistant.confirmActionPrompt', "Veuillez confirmer l'action suivante :"),
+            confirmationRequest: parsedResponse,
+            model: params.model,
+            filter: params.filter,
+            data: params.data
+        };
     }
 
     // Action finale pour afficher un message

package/src/modules/data/data.backup.js

@@ -8,7 +8,7 @@ import {downloadFromS3, getUserS3Config, listS3Backups, uploadToS3} from "../buc
 import fs from "node:fs";
 import {modelsCache, runCryptoWorkerTask} from "./data.core.js";
 import * as tar from "tar";
-import {getCollection, getUserCollectionName, modelsCollection} from "../mongodb.js";
+import {getCollection, getDatabase, getUserCollectionName, modelsCollection} from "../mongodb.js";
 import {removeFile} from "../file.js";
 import {cancelAlerts, scheduleAlerts} from "./data.scheduling.js";
 import {dbName} from "../../constants.js";
@@ -136,7 +136,7 @@ export const loadFromDump = async (user, options = {}) => {
         await tar.extract({file: backupFilePath, gzip: true, C: tmpRestoreDir, sync: true});
 
         // ... (Cleaning logic: deleteMany, removeFile, cancelAlerts) ...
-        const datasCollection = getCollection("datas");
+        const datasCollection = getCollection(Config.Get('dataCollection', 'datas'));
         if (modelsOnly) {
             await modelsCollection.deleteMany({_user: user.username});
         } else {
@@ -169,7 +169,7 @@ export const loadFromDump = async (user, options = {}) => {
             args.push('--nsInclude', `${d}.models`);
         } else {
             // mongorestore accepte plusieurs fois l'option --nsInclude
-            args.push('--nsInclude', `${d}.datas`);
+            args.push('--nsInclude', `${d}.${Config.Get('dataCollection', 'datas')}`);
             args.push('--nsInclude', `${d}.models`);
         }
         // Le répertoire source est le dernier argument
@@ -245,7 +245,7 @@ export const dumpUserData = async (user) => {
 
 
         const d = Config.Get('dbName', dbName);
-        const collections = await MongoDatabase.listCollections().toArray();
+        const collections = await getDatabase().listCollections().toArray();
         for (const collection of collections) {
             const collsToBackup = [await getUserCollectionName(user), 'models'];
             if (collsToBackup.includes(collection.name)) {