hedgequantx 2.6.0 → 2.6.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "hedgequantx",
-  "version": "2.6.0",
+  "version": "2.6.2",
   "description": "HedgeQuantX - Prop Futures Trading CLI",
   "main": "src/app.js",
   "bin": {

package/src/menus/ai-agent.js

@@ -219,11 +219,13 @@ const showAgentDetails = async (agent) => {
   
   const choice = await prompts.textInput(chalk.cyan('SELECT:'));
   
+  const agentDisplayName = agent.model ? `${agent.name} (${agent.model})` : agent.name;
+  
   switch ((choice || '').toLowerCase()) {
     case 'a':
       if (!agent.isActive) {
         aiService.setActiveAgent(agent.id);
-        console.log(chalk.green(`\n  ${agent.name} IS NOW ACTIVE`));
+        console.log(chalk.green(`\n  ${agentDisplayName} IS NOW ACTIVE`));
         await prompts.waitForEnter();
       }
       return await aiAgentMenu();
@@ -231,7 +233,7 @@ const showAgentDetails = async (agent) => {
       return await selectModel(agent);
     case 'r':
       aiService.removeAgent(agent.id);
-      console.log(chalk.yellow(`\n  ${agent.name} REMOVED`));
+      console.log(chalk.yellow(`\n  ${agentDisplayName} REMOVED`));
       await prompts.waitForEnter();
       return await aiAgentMenu();
     case '<':
@@ -267,8 +269,9 @@ const selectActiveAgent = async () => {
     const providerColor = agent.providerId === 'anthropic' ? chalk.magenta :
                          agent.providerId === 'openai' ? chalk.green : chalk.cyan;
     
+    const modelDisplay = agent.model ? chalk.gray(` (${agent.model})`) : '';
     console.log(makeLine(
-      chalk.white(`[${i + 1}] `) + providerColor(agent.name) + activeMarker
+      chalk.white(`[${i + 1}] `) + providerColor(agent.name) + modelDisplay + activeMarker
     ));
   }
   
@@ -289,7 +292,9 @@ const selectActiveAgent = async () => {
   }
   
   aiService.setActiveAgent(agents[index].id);
-  console.log(chalk.green(`\n  ${agents[index].name} IS NOW ACTIVE`));
+  const selectedAgent = agents[index];
+  const modelInfo = selectedAgent.model ? ` (${selectedAgent.model})` : '';
+  console.log(chalk.green(`\n  ${selectedAgent.name}${modelInfo} IS NOW ACTIVE`));
   await prompts.waitForEnter();
   return await aiAgentMenu();
 };
@@ -371,8 +376,9 @@ const selectAgentToRemove = async () => {
   
   for (let i = 0; i < agents.length; i++) {
     const agent = agents[i];
+    const modelDisplay = agent.model ? chalk.gray(` (${agent.model})`) : '';
     console.log(makeLine(
-      chalk.white(`[${i + 1}] `) + chalk.red(agent.name)
+      chalk.white(`[${i + 1}] `) + chalk.red(agent.name) + modelDisplay
     ));
   }
   
@@ -392,8 +398,10 @@ const selectAgentToRemove = async () => {
     return await aiAgentMenu();
   }
   
-  aiService.removeAgent(agents[index].id);
-  console.log(chalk.yellow(`\n  ${agents[index].name} REMOVED`));
+  const removedAgent = agents[index];
+  const modelInfo = removedAgent.model ? ` (${removedAgent.model})` : '';
+  aiService.removeAgent(removedAgent.id);
+  console.log(chalk.yellow(`\n  ${removedAgent.name}${modelInfo} REMOVED`));
   await prompts.waitForEnter();
   return await aiAgentMenu();
 };

package/src/services/session.js

@@ -141,9 +141,15 @@ const connections = {
 
   /**
    * Saves all sessions to encrypted storage
+   * IMPORTANT: Preserves AI agent sessions when saving trading connections
    */
   saveToStorage() {
-    const sessions = this.services.map(conn => {
+    // Load existing sessions to preserve AI settings
+    const existingSessions = storage.load();
+    const aiSession = existingSessions.find(s => s.type === 'ai');
+    
+    // Build trading sessions
+    const tradingSessions = this.services.map(conn => {
       const session = {
         type: conn.type,
         propfirm: conn.propfirm,
@@ -163,7 +169,12 @@ const connections = {
       return session;
     });
     
-    storage.save(sessions);
+    // Combine: trading sessions + preserved AI session
+    const allSessions = aiSession 
+      ? [...tradingSessions, aiSession]
+      : tradingSessions;
+    
+    storage.save(allSessions);
   },
 
   /**
@@ -360,9 +371,13 @@ const connections = {
   },
 
   /**
-   * Disconnects all connections and clears sessions
+   * Disconnects all trading connections (preserves AI agents)
    */
   disconnectAll() {
+    // Preserve AI session before clearing
+    const existingSessions = storage.load();
+    const aiSession = existingSessions.find(s => s.type === 'ai');
+    
     for (const conn of this.services) {
       try {
         if (conn.service?.logout) {
@@ -381,8 +396,15 @@ const connections = {
     }
     
     this.services = [];
-    storage.clear();
-    log.info('All connections disconnected');
+    
+    // Save with preserved AI session (don't use storage.clear())
+    if (aiSession) {
+      storage.save([aiSession]);
+      log.info('Trading connections disconnected (AI agents preserved)');
+    } else {
+      storage.clear();
+      log.info('All connections disconnected');
+    }
   },
 
   /**