@probelabs/probe-chat 0.6.0-rc197 → 0.6.0-rc199

package/ChatSessionManager.js

@@ -63,15 +63,18 @@ export class ChatSessionManager {
   }
   
   /**
-   * Initialize the ChatSessionManager by loading history
+   * Initialize the ChatSessionManager by initializing ProbeAgent and loading history
    * This must be called before using chat()
    */
   async initialize() {
     if (this._ready) return; // Already initialized
-    
+
+    // Initialize ProbeAgent (handles API key detection and CLI fallback)
+    await this.agent.initialize();
+
     await this.loadHistory();
     this._ready = true;
-    
+
     if (this.debug) {
       console.log(`[ChatSessionManager] Initialized session ${this.sessionId} with ${this.agent.history.length} messages from storage`);
     }

package/index.js

@@ -346,12 +346,6 @@ export function main() {
     process.env.PORT = options.port;
   }
 
-  // Check for API keys
-  const anthropicApiKey = process.env.ANTHROPIC_API_KEY;
-  const openaiApiKey = process.env.OPENAI_API_KEY;
-  const googleApiKey = process.env.GOOGLE_API_KEY;
-  const hasApiKeys = !!(anthropicApiKey || openaiApiKey || googleApiKey);
-
   // --- Bash Configuration Processing ---
   let bashConfig = null;
   if (options.enableBash) {
@@ -406,16 +400,13 @@ export function main() {
 
   // --- Web Mode (check before non-interactive to override) ---
   if (options.web) {
-    if (!hasApiKeys) {
-      // Use logWarn for web mode warning
-      logWarn(chalk.yellow('Warning: No API key provided. The web interface will show instructions on how to set up API keys.'));
-    }
-    // Import and start web server
+    // Note: API key / CLI availability is checked lazily in ProbeAgent.initialize()
+    // when the first chat message is sent. This allows fallback to Claude Code or Codex CLI.
     import('./webServer.js')
       .then(async module => {
         const { startWebServer } = module;
         logInfo(`Starting web server on port ${process.env.PORT || 8080}...`);
-        await startWebServer(version, hasApiKeys, { allowEdit: options.allowEdit });
+        await startWebServer(version, { allowEdit: options.allowEdit });
       })
       .catch(error => {
         logError(chalk.red(`Error starting web server: ${error.message}`));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@probelabs/probe-chat",
-  "version": "0.6.0-rc197",
+  "version": "0.6.0-rc199",
   "description": "CLI and web interface for Probe code search (formerly @probelabs/probe-web and @probelabs/probe-chat)",
   "main": "index.js",
   "type": "module",

package/test/unit/chatSessionManager.test.js

@@ -0,0 +1,86 @@
+import { describe, it, beforeEach, afterEach, mock } from 'node:test';
+import assert from 'node:assert';
+import { ChatSessionManager } from '../../ChatSessionManager.js';
+
+// Store original environment
+let originalEnv;
+
+describe('ChatSessionManager Tests', () => {
+  beforeEach(() => {
+    // Store original environment
+    originalEnv = { ...process.env };
+
+    // Set test mode
+    process.env.NODE_ENV = 'test';
+    process.env.USE_MOCK_AI = 'true';
+  });
+
+  afterEach(() => {
+    // Restore original environment
+    Object.keys(process.env).forEach(key => {
+      if (!(key in originalEnv)) {
+        delete process.env[key];
+      }
+    });
+    Object.assign(process.env, originalEnv);
+  });
+
+  describe('initialize()', () => {
+    it('should call agent.initialize() when initializing', async () => {
+      // Create a ChatSessionManager
+      const session = new ChatSessionManager({
+        sessionId: 'test-session-123',
+        debug: false
+      });
+
+      // Track if agent.initialize was called
+      let agentInitializeCalled = false;
+      const originalInitialize = session.agent.initialize.bind(session.agent);
+      session.agent.initialize = async function() {
+        agentInitializeCalled = true;
+        return originalInitialize();
+      };
+
+      // Initialize the session
+      await session.initialize();
+
+      // Verify agent.initialize was called
+      assert.strictEqual(agentInitializeCalled, true, 'agent.initialize() should be called during ChatSessionManager.initialize()');
+    });
+
+    it('should only initialize once even if called multiple times', async () => {
+      const session = new ChatSessionManager({
+        sessionId: 'test-session-456',
+        debug: false
+      });
+
+      let initializeCallCount = 0;
+      const originalInitialize = session.agent.initialize.bind(session.agent);
+      session.agent.initialize = async function() {
+        initializeCallCount++;
+        return originalInitialize();
+      };
+
+      // Call initialize multiple times
+      await session.initialize();
+      await session.initialize();
+      await session.initialize();
+
+      // Should only be called once due to _ready flag
+      assert.strictEqual(initializeCallCount, 1, 'agent.initialize() should only be called once');
+    });
+
+    it('should set _ready flag after successful initialization', async () => {
+      const session = new ChatSessionManager({
+        sessionId: 'test-session-789',
+        debug: false
+      });
+
+      assert.strictEqual(session._ready, false, '_ready should be false before initialize');
+
+      await session.initialize();
+
+      assert.strictEqual(session._ready, true, '_ready should be true after initialize');
+    });
+  });
+});

package/webServer.js

@@ -98,13 +98,12 @@ function getOrCreateChat(sessionId, apiCredentials = null) {
 /**
  * Start the web server
  * @param {string} version - The version of the application
- * @param {boolean} hasApiKeys - Whether any API keys are configured
  * @param {Object} options - Additional options
  * @param {boolean} options.allowEdit - Whether to allow editing files via the implement tool
  */
-export async function startWebServer(version, hasApiKeys = true, options = {}) {
+export async function startWebServer(version, options = {}) {
 	const allowEdit = options?.allowEdit || false;
-	
+
 	if (allowEdit) {
 		console.log('Edit mode enabled: implement tool is available');
 	}
@@ -120,11 +119,11 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 	}
 
 	// Initialize persistent storage for web mode
-	globalStorage = new JsonChatStorage({ 
-		webMode: true, 
-		verbose: process.env.DEBUG_CHAT === '1' 
+	globalStorage = new JsonChatStorage({
+		webMode: true,
+		verbose: process.env.DEBUG_CHAT === '1'
 	});
-	
+
 	// Initialize storage synchronously before server starts
 	try {
 		await globalStorage.initialize();
@@ -144,13 +143,9 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 		? process.env.ALLOWED_FOLDERS.split(',').map(folder => folder.trim()).filter(Boolean)
 		: [];
 
-
-	let noApiKeysMode = !hasApiKeys;
-	if (noApiKeysMode) {
-		console.log('Running in No API Keys mode - will show setup instructions to users');
-	} else {
-		console.log('API keys detected. Chat functionality enabled.');
-	}
+	// Note: API key / CLI availability is checked lazily in ProbeAgent.initialize()
+	// when the first chat message is sent. This allows fallback to Claude Code or Codex CLI.
+	console.log('Chat functionality enabled (API availability checked on first message)');
 
 
 	// Define the tools available for direct API calls (bypassing LLM loop)
@@ -419,7 +414,7 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 			// UI Routes
 			'GET /': (req, res) => {
 				const htmlPath = join(__dirname, 'index.html');
-				serveHtml(res, htmlPath, { 'data-no-api-keys': noApiKeysMode ? 'true' : 'false' });
+				serveHtml(res, htmlPath, { 'data-no-api-keys': 'false' });
 			},
 
 			// Chat session route - serves HTML with injected session ID
@@ -428,15 +423,15 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 				if (!sessionId) {
 					return sendError(res, 400, 'Invalid session ID in URL');
 				}
-				
+
 				// Validate that session exists or at least has a valid UUID format
 				if (!isValidUUID(sessionId)) {
 					return sendError(res, 400, 'Invalid session ID format');
 				}
 
 				const htmlPath = join(__dirname, 'index.html');
-				serveHtml(res, htmlPath, { 
-					'data-no-api-keys': noApiKeysMode ? 'true' : 'false',
+				serveHtml(res, htmlPath, {
+					'data-no-api-keys': 'false',
 					'data-session-id': sessionId
 				});
 			},
@@ -451,7 +446,7 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 				sendJson(res, 200, {
 					folders: folders,
 					currentDir: currentDir,
-					noApiKeysMode: noApiKeysMode
+					noApiKeysMode: false
 				});
 			},
 
@@ -700,12 +695,6 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 					// Update last activity timestamp
 					chatInstance.lastActivity = Date.now();
 
-					// Check if API keys are needed but missing
-					if (chatInstance.noApiKeysMode) {
-						console.warn(`[WARN] Chat request for session ${chatSessionId} cannot proceed: No API keys configured.`);
-						return sendError(res, 503, 'Chat service unavailable: API key not configured on server.');
-					}
-
 					// Register this request as active for cancellation
 					registerRequest(chatSessionId, { abort: () => chatInstance.abort() });
 					if (DEBUG) console.log(`[DEBUG] Registered cancellable request for session: ${chatSessionId}`);
@@ -918,9 +907,6 @@ export async function startWebServer(version, hasApiKeys = true, options = {}) {
 		console.log(`Probe Web Interface v${version}`);
 		console.log(`Server running on http://localhost:${PORT}`);
 		console.log(`Environment: ${process.env.NODE_ENV || 'development'}`);
-		if (noApiKeysMode) {
-			console.log('*** Running in NO API KEYS mode. Chat functionality disabled. ***');
-		}
 	});
 }