release bug fix

Author: j-mchugh

src/main/preload.ts

@@ -93,6 +93,7 @@ contextBridge.exposeInMainWorld('electronAPI', {
   testModelConnection: (model: any) => ipcRenderer.invoke('test-model-connection', model),
   executeRedTeamAttack: (model: any, attack: any) => ipcRenderer.invoke('execute-red-team-attack', model, attack),
   executeBenchmarkTest: (model: any, benchmark: any, payload: any) => ipcRenderer.invoke('execute-benchmark-test', model, benchmark, payload),
+  getAttackPayloads: () => ipcRenderer.invoke('get-attack-payloads'),
 })
 
 // Type declaration for the exposed API
@@ -124,6 +125,7 @@ declare global {
       testModelConnection: (model: any) => Promise<any>
       executeRedTeamAttack: (model: any, attack: any) => Promise<any>
       executeBenchmarkTest: (model: any, benchmark: any, payload: any) => Promise<any>
+      getAttackPayloads: () => Promise<AttackPayload[]>
     }
   }
 } 

src/main/services/attack-engine.ts

@@ -8,16 +8,18 @@ import { AIRequest } from '../../types';
 // Import attack payloads with proper path resolution for packaged apps
 const getPayloadsPath = () => {
   if (app.isPackaged) {
-    // In packaged app, use the resources directory
-    return path.join(process.resourcesPath, 'app.asar', 'public/assets/payloads/all-attack-payloads.json')
+    // In packaged app, use the asar path
+    return path.join(process.resourcesPath, 'app.asar', 'public', 'assets', 'payloads', 'all-attack-payloads.json');
   } else {
     // In development, use the normal path
-    return path.join(__dirname, '../../../public/assets/payloads/all-attack-payloads.json')
+    return path.join(__dirname, '../../../public/assets/payloads/all-attack-payloads.json');
   }
 }
 
-const payloadsPath = getPayloadsPath()
-const attackPayloads = JSON.parse(fs.readFileSync(payloadsPath, 'utf8'))
+const payloadsPath = getPayloadsPath();
+console.log('Payloads path:', payloadsPath);
+console.log('Exists:', fs.existsSync(payloadsPath));
+const attackPayloads = JSON.parse(fs.readFileSync(payloadsPath, 'utf8'));
 
 export interface AttackPayload {
   id: string
@@ -293,5 +295,9 @@ export class AttackEngine extends EventEmitter {
     ipcMain.handle('get-test', (_event, testId: string) => {
       return this.tests.get(testId)
     })
+
+    ipcMain.handle('get-attack-payloads', async () => {
+      return attackPayloads;
+    });
   }
 } 

src/services/payload-manager.ts

@@ -382,9 +382,16 @@ export class PayloadManager {
   public async loadAllPayloads(): Promise<AttackPayload[]> {
     const allPayloads: AttackPayload[] = []
     try {
-      const response = await fetch(this.sources[0])
-      if (!response.ok) throw new Error('Failed to load all-attack-payloads.json')
-      const payloads = await response.json()
+      let payloads: AttackPayload[] = [];
+      if (window.electronAPI && typeof window.electronAPI.getAttackPayloads === 'function') {
+        // Electron production: use IPC
+        payloads = await window.electronAPI.getAttackPayloads();
+      } else {
+        // Dev mode: use fetch
+        const response = await fetch(this.sources[0])
+        if (!response.ok) throw new Error('Failed to load all-attack-payloads.json')
+        payloads = await response.json()
+      }
       allPayloads.push(...payloads)
       // Index payloads by ID
       allPayloads.forEach(payload => {

vite.config.ts

@@ -4,6 +4,7 @@ import path from 'path'
 
 // https://vitejs.dev/config/
 export default defineConfig({
+  base: './',
   plugins: [react()],
   resolve: {
     alias: {