add in firecrawl search

2025-09-09 08:14:54 -04:00
parent cb1b0a9f64
commit 599e209888
11 changed files with 559 additions and 119 deletions
@@ -7,7 +7,6 @@ export class SandboxFactory {
    // Use environment variable if provider not specified
    const selectedProvider = provider || process.env.SANDBOX_PROVIDER || 'e2b';
    
-    console.log(`[SandboxFactory] Creating ${selectedProvider} provider`);
    
    switch (selectedProvider.toLowerCase()) {
      case 'e2b':
@@ -11,7 +11,6 @@ export class E2BProvider extends SandboxProvider {
   */
  async reconnect(sandboxId: string): Promise<boolean> {
    try {
-      console.log(`[E2BProvider] Attempting to reconnect to sandbox ${sandboxId}...`);
      
      // Try to connect to existing sandbox
      // Note: E2B SDK doesn't directly support reconnection, but we can try to recreate
@@ -27,11 +26,9 @@ export class E2BProvider extends SandboxProvider {

  async createSandbox(): Promise<SandboxInfo> {
    try {
-      console.log('[E2BProvider] Creating sandbox...');
      
      // Kill existing sandbox if any
      if (this.sandbox) {
-        console.log('[E2BProvider] Killing existing sandbox...');
        try {
          await this.sandbox.kill();
        } catch (e) {
@@ -44,7 +41,6 @@ export class E2BProvider extends SandboxProvider {
      this.existingFiles.clear();

      // Create base sandbox
-      console.log(`[E2BProvider] Creating E2B sandbox with ${appConfig.e2b.timeoutMinutes} minute timeout...`);
      this.sandbox = await Sandbox.create({ 
        apiKey: this.config.e2b?.apiKey || process.env.E2B_API_KEY,
        timeoutMs: this.config.e2b?.timeoutMs || appConfig.e2b.timeoutMs
@@ -53,8 +49,6 @@ export class E2BProvider extends SandboxProvider {
      const sandboxId = (this.sandbox as any).sandboxId || Date.now().toString();
      const host = (this.sandbox as any).getHost(appConfig.e2b.vitePort);
      
-      console.log(`[E2BProvider] Sandbox created: ${sandboxId}`);
-      console.log(`[E2BProvider] Sandbox host: ${host}`);

      this.sandboxInfo = {
        sandboxId,
@@ -66,7 +60,6 @@ export class E2BProvider extends SandboxProvider {
      // Set extended timeout on the sandbox instance if method available
      if (typeof this.sandbox.setTimeout === 'function') {
        this.sandbox.setTimeout(appConfig.e2b.timeoutMs);
-        console.log(`[E2BProvider] Set sandbox timeout to ${appConfig.e2b.timeoutMinutes} minutes`);
      }

      return this.sandboxInfo;
@@ -82,7 +75,6 @@ export class E2BProvider extends SandboxProvider {
      throw new Error('No active sandbox');
    }

-    console.log(`[E2BProvider] Executing: ${command}`);
    
    const result = await this.sandbox.runCode(`
      import subprocess
@@ -125,7 +117,6 @@ export class E2BProvider extends SandboxProvider {
    if ((this.sandbox as any).files && typeof (this.sandbox as any).files.write === 'function') {
      // Use the files.write API if available
      await (this.sandbox as any).files.write(fullPath, Buffer.from(content));
-      console.log(`[E2BProvider] Written file using files.write: ${fullPath}`);
    } else {
      // Fallback to Python code execution
      await this.sandbox.runCode(`
@@ -140,7 +131,6 @@ export class E2BProvider extends SandboxProvider {
            f.write(${JSON.stringify(content)})
        print(f"✓ Written: ${fullPath}")
      `);
-      console.log(`[E2BProvider] Written file using Python: ${fullPath}`);
    }
    
    this.existingFiles.add(path);
@@ -200,7 +190,6 @@ export class E2BProvider extends SandboxProvider {
    const packageList = packages.join(' ');
    const flags = appConfig.packages.useLegacyPeerDeps ? '--legacy-peer-deps' : '';
    
-    console.log(`[E2BProvider] Installing packages: ${packageList}`);
    
    const result = await this.sandbox.runCode(`
      import subprocess
@@ -244,7 +233,6 @@ export class E2BProvider extends SandboxProvider {
      throw new Error('No active sandbox');
    }

-    console.log('[E2BProvider] Setting up Vite React app...');
    
    // Write all files in a single Python script
    const setupScript = `
@@ -405,7 +393,6 @@ print('\\nAll files created successfully!')
    await this.sandbox.runCode(setupScript);
    
    // Install dependencies
-    console.log('[E2BProvider] Installing dependencies...');
    await this.sandbox.runCode(`
 import subprocess

@@ -424,7 +411,6 @@ else:
    `);
    
    // Start Vite dev server
-    console.log('[E2BProvider] Starting Vite dev server...');
    await this.sandbox.runCode(`
 import subprocess
 import os
@@ -470,7 +456,6 @@ print('Waiting for server to be ready...')
      throw new Error('No active sandbox');
    }

-    console.log('[E2BProvider] Restarting Vite server...');
    
    await this.sandbox.runCode(`
 import subprocess
@@ -511,7 +496,6 @@ print(f'✓ Vite restarted with PID: {process.pid}')

  async terminate(): Promise<void> {
    if (this.sandbox) {
-      console.log('[E2BProvider] Terminating sandbox...');
      try {
        await this.sandbox.kill();
      } catch (e) {
@@ -7,11 +7,9 @@ export class VercelProvider extends SandboxProvider {

  async createSandbox(): Promise<SandboxInfo> {
    try {
-      console.log('[VercelProvider] Creating sandbox...');
      
      // Kill existing sandbox if any
      if (this.sandbox) {
-        console.log('[VercelProvider] Stopping existing sandbox...');
        try {
          await this.sandbox.stop();
        } catch (e) {
@@ -24,7 +22,6 @@ export class VercelProvider extends SandboxProvider {
      this.existingFiles.clear();

      // Create Vercel sandbox
-      console.log('[VercelProvider] Creating Vercel sandbox...');
      
      const sandboxConfig: any = {
        timeout: 300000, // 5 minutes in ms
@@ -34,21 +31,13 @@ export class VercelProvider extends SandboxProvider {

      // Add authentication based on environment variables
      if (process.env.VERCEL_TOKEN && process.env.VERCEL_TEAM_ID && process.env.VERCEL_PROJECT_ID) {
-        console.log('[VercelProvider] Using personal access token authentication');
-        console.log('[VercelProvider] Team ID:', process.env.VERCEL_TEAM_ID);
-        console.log('[VercelProvider] Project ID:', process.env.VERCEL_PROJECT_ID);
-        console.log('[VercelProvider] Token present:', !!process.env.VERCEL_TOKEN);
        sandboxConfig.teamId = process.env.VERCEL_TEAM_ID;
        sandboxConfig.projectId = process.env.VERCEL_PROJECT_ID;
        sandboxConfig.token = process.env.VERCEL_TOKEN;
      } else if (process.env.VERCEL_OIDC_TOKEN) {
-        console.log('[VercelProvider] Using OIDC token authentication');
      } else {
-        console.log('[VercelProvider] No authentication found - relying on default Vercel authentication');
-        console.log('[VercelProvider] Available env vars:', Object.keys(process.env).filter(k => k.startsWith('VERCEL')));
      }

-      console.log('[VercelProvider] Creating sandbox with config:', {
        runtime: sandboxConfig.runtime,
        timeout: sandboxConfig.timeout,
        ports: sandboxConfig.ports,
@@ -60,14 +49,12 @@ export class VercelProvider extends SandboxProvider {
      this.sandbox = await Sandbox.create(sandboxConfig);
      
      const sandboxId = this.sandbox.sandboxId;
-      console.log(`[VercelProvider] Sandbox created successfully:`, {
        sandboxId: sandboxId,
        status: this.sandbox.status
      });
      
      // Get the sandbox URL using the correct Vercel Sandbox API
      const sandboxUrl = this.sandbox.domain(5173);
-      console.log(`[VercelProvider] Sandbox URL: ${sandboxUrl}`);

      this.sandboxInfo = {
        sandboxId,
@@ -89,7 +76,6 @@ export class VercelProvider extends SandboxProvider {
      throw new Error('No active sandbox');
    }

-    console.log(`[VercelProvider] Executing: ${command}`);
    
    try {
      // Parse command into cmd and args (matching PR syntax)
@@ -129,7 +115,6 @@ export class VercelProvider extends SandboxProvider {
    // Vercel sandbox default working directory is /vercel/sandbox
    const fullPath = path.startsWith('/') ? path : `/vercel/sandbox/${path}`;
    
-    console.log(`[VercelProvider] writeFile called:`, {
      originalPath: path,
      fullPath: fullPath,
      contentLength: content.length,
@@ -141,7 +126,6 @@ export class VercelProvider extends SandboxProvider {
    // Based on Vercel SDK docs, writeFiles expects path and Buffer content
    try {
      const buffer = Buffer.from(content, 'utf-8');
-      console.log(`[VercelProvider] Calling sandbox.writeFiles with:`, {
        path: fullPath,
        bufferLength: buffer.length,
        isBuffer: Buffer.isBuffer(buffer)
@@ -152,7 +136,6 @@ export class VercelProvider extends SandboxProvider {
        content: buffer
      }]);
      
-      console.log(`[VercelProvider] Successfully written: ${fullPath}`);
      this.existingFiles.add(path);
    } catch (writeError: any) {
      // Log detailed error information
@@ -165,17 +148,14 @@ export class VercelProvider extends SandboxProvider {
      });
      
      // Fallback to command-based approach if writeFiles fails
-      console.log(`[VercelProvider] Attempting command fallback for ${fullPath}`);
      
      // Ensure directory exists
      const dir = fullPath.substring(0, fullPath.lastIndexOf('/'));
      if (dir) {
-        console.log(`[VercelProvider] Creating directory: ${dir}`);
        const mkdirResult = await this.sandbox.runCommand({
          cmd: 'mkdir',
          args: ['-p', dir]
        });
-        console.log(`[VercelProvider] mkdir result:`, {
          exitCode: mkdirResult.exitCode,
          stdout: mkdirResult.stdout,
          stderr: mkdirResult.stderr
@@ -190,20 +170,17 @@ export class VercelProvider extends SandboxProvider {
        .replace(/`/g, '\\`')
        .replace(/\n/g, '\\n');
      
-      console.log(`[VercelProvider] Writing file via echo command to: ${fullPath}`);
      const writeResult = await this.sandbox.runCommand({
        cmd: 'sh',
        args: ['-c', `echo "${escapedContent}" > "${fullPath}"`]
      });
      
-      console.log(`[VercelProvider] Write command result:`, {
        exitCode: writeResult.exitCode,
        stdout: writeResult.stdout,
        stderr: writeResult.stderr
      });
      
      if (writeResult.exitCode === 0) {
-        console.log(`[VercelProvider] Successfully written via command: ${fullPath}`);
        this.existingFiles.add(path);
      } else {
        throw new Error(`Failed to write file via command: ${writeResult.stderr}`);
@@ -256,7 +233,6 @@ export class VercelProvider extends SandboxProvider {

    const flags = process.env.NPM_FLAGS || '';
    
-    console.log(`[VercelProvider] Installing packages: ${packages.join(' ')}`);
    
    // Build args array
    const args = ['install'];
@@ -289,19 +265,15 @@ export class VercelProvider extends SandboxProvider {
      throw new Error('No active sandbox');
    }

-    console.log('[VercelProvider] Setting up Vite React app...');
-    console.log('[VercelProvider] Sandbox details:', {
      sandboxId: this.sandbox.sandboxId,
      status: this.sandbox.status
    });
    
    // Create directory structure
-    console.log('[VercelProvider] Creating directory structure...');
    const mkdirResult = await this.sandbox.runCommand({
      cmd: 'mkdir',
      args: ['-p', '/vercel/sandbox/src']
    });
-    console.log('[VercelProvider] mkdir /vercel/sandbox/src result:', {
      exitCode: mkdirResult.exitCode,
      stdout: mkdirResult.stdout,
      stderr: mkdirResult.stderr
@@ -330,7 +302,6 @@ export class VercelProvider extends SandboxProvider {
      }
    };
    
-    console.log('[VercelProvider] Writing package.json...');
    await this.writeFile('package.json', JSON.stringify(packageJson, null, 2));
    
    // Create vite.config.js
@@ -442,11 +413,8 @@ body {
    
    await this.writeFile('src/index.css', indexCss);
    
-    console.log('[VercelProvider] All files created successfully!');
    
    // Install dependencies
-    console.log('[VercelProvider] Installing dependencies...');
-    console.log('[VercelProvider] Running npm install in /vercel/sandbox');
    try {
      const installResult = await this.sandbox.runCommand({
        cmd: 'npm',
@@ -454,14 +422,12 @@ body {
        cwd: '/vercel/sandbox'
      });
      
-      console.log('[VercelProvider] npm install result:', {
        exitCode: installResult.exitCode,
        stdout: typeof installResult.stdout === 'function' ? 'function' : installResult.stdout,
        stderr: typeof installResult.stderr === 'function' ? 'function' : installResult.stderr
      });
      
      if (installResult.exitCode === 0) {
-        console.log('[VercelProvider] Dependencies installed successfully');
      } else {
        console.warn('[VercelProvider] npm install had issues:', installResult.stderr);
      }
@@ -472,7 +438,6 @@ body {
        responseText: error?.text
      });
      // Try alternative approach - run as shell command
-      console.log('[VercelProvider] Trying alternative npm install approach...');
      try {
        const altResult = await this.sandbox.runCommand({
          cmd: 'sh',
@@ -480,7 +445,6 @@ body {
          cwd: '/vercel/sandbox'
        });
        if (altResult.exitCode === 0) {
-          console.log('[VercelProvider] Dependencies installed successfully (alternative method)');
        } else {
          console.warn('[VercelProvider] Alternative npm install also had issues:', altResult.stderr);
        }
@@ -491,7 +455,6 @@ body {
    }
    
    // Start Vite dev server
-    console.log('[VercelProvider] Starting Vite dev server...');
    
    // Kill any existing Vite processes
    await this.sandbox.runCommand({
@@ -507,7 +470,6 @@ body {
      cwd: '/vercel/sandbox'
    });
    
-    console.log('[VercelProvider] Vite dev server started');
    
    // Wait for Vite to be ready
    await new Promise(resolve => setTimeout(resolve, 7000));
@@ -528,7 +490,6 @@ body {
      throw new Error('No active sandbox');
    }

-    console.log('[VercelProvider] Restarting Vite server...');
    
    // Kill existing Vite process
    await this.sandbox.runCommand({
@@ -547,7 +508,6 @@ body {
      cwd: '/vercel/sandbox'
    });
    
-    console.log('[VercelProvider] Vite restarted');
    
    // Wait for Vite to be ready
    await new Promise(resolve => setTimeout(resolve, 7000));
@@ -563,7 +523,6 @@ body {

  async terminate(): Promise<void> {
    if (this.sandbox) {
-      console.log('[VercelProvider] Terminating sandbox...');
      try {
        await this.sandbox.stop();
      } catch (e) {
@@ -24,7 +24,6 @@ class SandboxManager {
    }

    // Try to reconnect to existing sandbox
-    console.log(`[SandboxManager] Attempting to reconnect to sandbox ${sandboxId}`);
    
    try {
      const provider = SandboxFactory.create();
@@ -41,14 +40,12 @@ class SandboxManager {
            lastAccessed: new Date()
          });
          this.activeSandboxId = sandboxId;
-          console.log(`[SandboxManager] Successfully reconnected to sandbox ${sandboxId}`);
          return provider;
        }
      }
      
      // For Vercel or if reconnection failed, return the new provider
      // The caller will need to handle creating a new sandbox
-      console.log(`[SandboxManager] Could not reconnect to ${sandboxId}, returning new provider`);
      return provider;
    } catch (error) {
      console.error(`[SandboxManager] Error reconnecting to sandbox ${sandboxId}:`, error);
@@ -67,7 +64,6 @@ class SandboxManager {
      lastAccessed: new Date()
    });
    this.activeSandboxId = sandboxId;
-    console.log(`[SandboxManager] Registered sandbox ${sandboxId}`);
  }

  /**
@@ -160,7 +156,6 @@ class SandboxManager {
    
    for (const id of toDelete) {
      await this.terminateSandbox(id);
-      console.log(`[SandboxManager] Cleaned up old sandbox ${id}`);
    }
  }
 }