berget 2.2.6 → 2.2.8

package/templates/agents/security.md

@@ -12,6 +12,7 @@ permission:
 Voice: Scandinavian calm—precise, concise, confident. You are Berget Code Security agent. Expert in application security, penetration testing, and OWASP standards. Core responsibilities: Conduct security assessments and penetration tests, Validate OWASP Top 10 compliance, Review code for security vulnerabilities, Implement security headers and Content Security Policy (CSP), Audit API security, Check for sensitive data exposure, Validate input sanitization and output encoding, Assess dependency security and supply chain risks. Tools and techniques: OWASP ZAP, Burp Suite, security linters, dependency scanners, manual code review. Always provide specific, actionable security recommendations with priority levels.
 
 GIT WORKFLOW RULES (CRITICAL):
+
 - NEVER push directly to main branch - ALWAYS use pull requests
 - NEVER use 'git add .' - ALWAYS add specific files with 'git add path/to/file'
 - ALWAYS clean up test files, documentation files, and temporary artifacts before committing

package/tests/commands/chat.test.ts

@@ -1,83 +1,76 @@
-import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest'
-import { Command } from 'commander'
-import { registerChatCommands } from '../../src/commands/chat'
-import { ChatService } from '../../src/services/chat-service'
-import { DefaultApiKeyManager } from '../../src/utils/default-api-key'
+import { Command } from 'commander';
+import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
+
+import { registerChatCommands } from '../../src/commands/chat';
+import { ChatService } from '../../src/services/chat-service';
+import { DefaultApiKeyManager } from '../../src/utils/default-api-key';
 
 // Mock dependencies
-vi.mock('../../src/services/chat-service')
-vi.mock('../../src/utils/default-api-key')
+vi.mock('../../src/services/chat-service');
+vi.mock('../../src/utils/default-api-key');
 vi.mock('readline', () => ({
   createInterface: vi.fn(() => ({
-    question: vi.fn(),
     close: vi.fn(),
+    question: vi.fn(),
   })),
-}))
+}));
 
 describe('Chat Commands', () => {
-  let program: Command
-  let mockChatService: any
-  let mockDefaultApiKeyManager: any
+  let program: Command;
+  let mockChatService: any;
+  let mockDefaultApiKeyManager: any;
 
   beforeEach(() => {
-    program = new Command()
+    program = new Command();
 
     // Mock ChatService
     mockChatService = {
       createCompletion: vi.fn(),
       listModels: vi.fn(),
-    }
-    vi.mocked(ChatService.getInstance).mockReturnValue(mockChatService)
+    };
+    vi.mocked(ChatService.getInstance).mockReturnValue(mockChatService);
 
     // Mock DefaultApiKeyManager
     mockDefaultApiKeyManager = {
       getDefaultApiKeyData: vi.fn(),
       promptForDefaultApiKey: vi.fn(),
-    }
-    vi.mocked(DefaultApiKeyManager.getInstance).mockReturnValue(
-      mockDefaultApiKeyManager,
-    )
+    };
+    vi.mocked(DefaultApiKeyManager.getInstance).mockReturnValue(mockDefaultApiKeyManager);
 
-    registerChatCommands(program)
-  })
+    registerChatCommands(program);
+  });
 
   afterEach(() => {
-    vi.clearAllMocks()
-  })
+    vi.clearAllMocks();
+  });
 
   describe('chat run command', () => {
     it('should use berget/glm-4.7 as default model', () => {
-      const chatCommand = program.commands.find((cmd) => cmd.name() === 'chat')
-      const runCommand = chatCommand?.commands.find(
-        (cmd) => cmd.name() === 'run',
-      )
+      const chatCommand = program.commands.find((cmd) => cmd.name() === 'chat');
+      const runCommand = chatCommand?.commands.find((cmd) => cmd.name() === 'run');
 
-      expect(runCommand).toBeDefined()
+      expect(runCommand).toBeDefined();
 
       // Check the help text which contains the default model
-      const helpText = runCommand?.helpInformation()
-      expect(helpText).toContain('glm-4.7')
-    })
+      const helpText = runCommand?.helpInformation();
+      expect(helpText).toContain('glm-4.7');
+    });
 
     it('should have streaming enabled by default', () => {
-      const chatCommand = program.commands.find((cmd) => cmd.name() === 'chat')
-      const runCommand = chatCommand?.commands.find(
-        (cmd) => cmd.name() === 'run',
-      )
+      const chatCommand = program.commands.find((cmd) => cmd.name() === 'chat');
+      const runCommand = chatCommand?.commands.find((cmd) => cmd.name() === 'run');
 
-      expect(runCommand).toBeDefined()
+      expect(runCommand).toBeDefined();
 
       // Check that the option is --no-stream (meaning streaming is default)
-      const streamOption = runCommand?.options.find(
-        (opt) => opt.long === '--no-stream',
-      )
-      expect(streamOption).toBeDefined()
-      expect(streamOption?.description).toContain('Disable streaming')
-    })
+      const streamOption = runCommand?.options.find((opt) => opt.long === '--no-stream');
+      expect(streamOption).toBeDefined();
+      expect(streamOption?.description).toContain('Disable streaming');
+    });
 
     it('should create completion with correct default options', async () => {
       // Mock API key
-      process.env.BERGET_API_KEY = 'test-key'
+      process.env.BERGET_API_KEY = 'test-key';
 
       // Mock successful completion
       mockChatService.createCompletion.mockResolvedValue({
@@ -86,44 +79,42 @@ describe('Chat Commands', () => {
             message: { content: 'Test response' },
           },
         ],
-      })
+      });
 
       // This would normally test the actual command execution
       // but since it involves readline interaction, we just verify
       // that the service would be called with correct defaults
-      expect(mockChatService.createCompletion).not.toHaveBeenCalled()
+      expect(mockChatService.createCompletion).not.toHaveBeenCalled();
 
       // Clean up
-      delete process.env.BERGET_API_KEY
-    })
-  })
+      delete process.env.BERGET_API_KEY;
+    });
+  });
 
   describe('chat list command', () => {
     it('should list available models', async () => {
       const mockModels = {
         data: [
           {
-            id: 'gpt-oss',
-            owned_by: 'openai',
             active: true,
             capabilities: {
-              vision: false,
               function_calling: true,
               json_mode: true,
+              vision: false,
             },
+            id: 'gpt-oss',
+            owned_by: 'openai',
           },
         ],
-      }
+      };
 
-      mockChatService.listModels.mockResolvedValue(mockModels)
+      mockChatService.listModels.mockResolvedValue(mockModels);
 
-      const chatCommand = program.commands.find((cmd) => cmd.name() === 'chat')
-      const listCommand = chatCommand?.commands.find(
-        (cmd) => cmd.name() === 'list',
-      )
+      const chatCommand = program.commands.find((cmd) => cmd.name() === 'chat');
+      const listCommand = chatCommand?.commands.find((cmd) => cmd.name() === 'list');
 
-      expect(listCommand).toBeDefined()
-      expect(listCommand?.description()).toBe('List available chat models')
-    })
-  })
-})
+      expect(listCommand).toBeDefined();
+      expect(listCommand?.description()).toBe('List available chat models');
+    });
+  });
+});