-
+
🧠 AI Learning System
diff --git a/lib/automation-singleton.js b/lib/automation-singleton.js
index 1c6f605..7e208c5 100644
--- a/lib/automation-singleton.js
+++ b/lib/automation-singleton.js
@@ -4,7 +4,7 @@ let automationInstance = null;
async function createAutomationInstance() {
try {
// Try to import the learning-enhanced automation first
- const AutomationWithLearning = require('./automation-with-learning-v2.js');
+ const AutomationWithLearning = (await import('./automation-with-learning-v2.js')).default;
console.log('✅ Creating automation instance with AI learning system');
return new AutomationWithLearning();
} catch (error) {
@@ -23,24 +23,24 @@ async function createAutomationInstance() {
}
}
-function getAutomationInstance() {
+async function getAutomationInstance() {
if (!automationInstance) {
- automationInstance = createAutomationInstance();
+ automationInstance = await createAutomationInstance();
}
return automationInstance;
}
-function resetAutomationInstance() {
+async function resetAutomationInstance() {
if (automationInstance) {
console.log('🔄 Resetting automation instance');
if (typeof automationInstance.stop === 'function') {
- automationInstance.stop();
+ await automationInstance.stop();
}
}
automationInstance = null;
}
-module.exports = {
+export {
getAutomationInstance,
resetAutomationInstance
};
diff --git a/lib/automation-with-learning-v2.js b/lib/automation-with-learning-v2.js
index a20e622..385ecef 100644
--- a/lib/automation-with-learning-v2.js
+++ b/lib/automation-with-learning-v2.js
@@ -445,4 +445,4 @@ class AutomationWithLearning {
}
}
-module.exports = AutomationWithLearning;
+export default AutomationWithLearning;
diff --git a/lib/db.js b/lib/db.js
index d44d2e4..5be8f47 100644
--- a/lib/db.js
+++ b/lib/db.js
@@ -1,46 +1,12 @@
-/**
- * Database utility for Prisma client
- *
- * Provides a global Prisma instance to avoid connection issues
- */
+import { PrismaClient } from '@prisma/client';
-const { PrismaClient } = require('@prisma/client');
+let prisma;
-// Global Prisma instance
-let prisma = null;
-
-/**
- * Get the global Prisma database instance
- */
-async function getDB() {
+function getDB() {
if (!prisma) {
prisma = new PrismaClient();
-
- // Test the connection
- try {
- await prisma.$connect();
- console.log('✅ Database connection established');
- } catch (error) {
- console.error('❌ Database connection failed:', error.message);
- throw error;
- }
}
-
return prisma;
}
-/**
- * Close the database connection
- */
-async function closeDB() {
- if (prisma) {
- await prisma.$disconnect();
- prisma = null;
- console.log('✅ Database connection closed');
- }
-}
-
-module.exports = {
- getDB,
- closeDB
-};
+export { getDB };
\ No newline at end of file
diff --git a/lib/simplified-stop-loss-learner-fixed.js b/lib/simplified-stop-loss-learner-fixed.js
index 2072655..5d1edde 100644
--- a/lib/simplified-stop-loss-learner-fixed.js
+++ b/lib/simplified-stop-loss-learner-fixed.js
@@ -5,8 +5,8 @@
* without complex statistical analysis.
*/
-const { PrismaClient } = require('@prisma/client');
-const getDB = require('./database-util');
+import { PrismaClient } from '@prisma/client';
+import { getDB } from './db.js';
class SimplifiedStopLossLearner {
constructor() {
@@ -527,4 +527,4 @@ class SimplifiedStopLossLearner {
}
}
-module.exports = SimplifiedStopLossLearner;
+export { SimplifiedStopLossLearner };
diff --git a/prisma/prisma/dev.db b/prisma/prisma/dev.db
index a12c0ec..7a43a44 100644
Binary files a/prisma/prisma/dev.db and b/prisma/prisma/dev.db differ