From b76c5005ff41ce70ab7ea82fdfe61eb1bf09ff12 Mon Sep 17 00:00:00 2001
From: Creylay <careyesla@gmail.com>
Date: Mon, 19 Jan 2026 12:07:38 -0300
Subject: [PATCH] fix:  update session handling props

---
 DashAI/front/src/pages/models/ModelsContent.jsx | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/DashAI/front/src/pages/models/ModelsContent.jsx b/DashAI/front/src/pages/models/ModelsContent.jsx
index 9dcea249c..32bb83394 100644
--- a/DashAI/front/src/pages/models/ModelsContent.jsx
+++ b/DashAI/front/src/pages/models/ModelsContent.jsx
@@ -923,10 +923,10 @@ export default function ModelsContent() {
                 ) : step === 2 && selectedDatasetId ? (
                   <DatasetVisualization
                     dataset={datasets.find((d) => d.id === selectedDatasetId)}
-                    onSessionCreated={handleSessionCreated}
-                    onNewSession={handleNewSessionFromDataset}
-                    existingSessions={sessions}
-                    tasks={tasks}
+                    onItemCreated={handleSessionCreated}
+                    onNewItem={handleNewSessionFromDataset}
+                    existingItems={sessions}
+                    newItemButtonText="New Session"
                   />
                 ) : step === 0 ? (
                   <SelectOptionMenu