chore: cleanup

Author: PhantomInTheWire

backend/ai/app/services/summarization.py

@@ -24,12 +24,9 @@ def __init__(self) -> None:
         self.api_key = os.getenv("OPENAI_API_KEY", "")
         self.base_url = os.getenv("OPENAI_BASE_URL", "https://openrouter.ai/api/v1")
 
-        # Primary model from env, plus fallback
         primary_model = os.getenv("OPENAI_MODEL", "z-ai/glm-4.5-air:free")
         self.models = [primary_model]
 
-        # Add fallback/alternative models for rotation
-        # Including openai and google free models as requested
         alternatives = [
             "meta-llama/llama-3.1-405b-instruct:free",
             "openai/gpt-oss-120b:free",
@@ -58,7 +55,6 @@ def _call_llm(
             raise ValueError("LLM client not initialized")
 
         for attempt in range(max_retries):
-            # Rotate models: try primary, then secondary, etc.
             model = self.models[attempt % len(self.models)]
 
             try:
@@ -139,9 +135,7 @@ def summarize_issues(
             )
 
             content = response.choices[0].message.content or "{}"
-            # Try to parse JSON from response
             try:
-                # Handle markdown code blocks
                 if "```json" in content:
                     content = content.split("```json")[1].split("```")[0]
                 elif "```" in content:
@@ -209,15 +203,13 @@ def generate_commands(
 
             content = response.choices[0].message.content or "{}"
             try:
-                # Handle markdown code blocks
                 if "```json" in content:
                     content = content.split("```json")[1].split("```")[0]
                 elif "```" in content:
                     content = content.split("```")[1].split("```")[0]
 
                 result = json.loads(content.strip())
                 commands = result.get("commands", [])
-                # Ensure all commands are strings
                 return [str(cmd) for cmd in commands if cmd]
             except json.JSONDecodeError:
                 logger.warning(f"Failed to parse commands JSON: {content}")
@@ -277,10 +269,8 @@ def SummarizeFindings(
 
         logger.info(f"Summarizing {len(findings)} findings for account {account_id}")
 
-        # Group findings by check_id and service
         grouped = self._group_findings(findings)
 
-        # Create finding groups with LLM summaries
         finding_groups = []
         action_items = []
 
@@ -313,7 +303,6 @@ def SummarizeFindings(
 
         logger.info("Finished processing all groups.")
 
-        # Calculate risk summary
         risk_summary = self._calculate_risk_summary(findings)
 
         return summarization_pb2.SummarizeFindingsResponse(
@@ -388,37 +377,29 @@ def _create_finding_group(
         service, check_id = group_key.split(":", 1)
         region = first.region or "us-east-1"
 
-        # Count failed findings
         failed_findings = [
             f for f in findings if f.status == summarization_pb2.FINDING_STATUS_FAIL
         ]
         failed_count = len(failed_findings)
 
-        # Determine highest severity
         severities = [f.severity for f in findings]
         max_severity = max(severities) if severities else 0
 
-        # Collect resource IDs
         resource_ids = [f.resource_id for f in findings]
 
-        # Generate title
         if failed_count > 0:
             title = f"{failed_count} {service.upper()} resources failed {check_id}"
         else:
             title = f"All {len(findings)} {service.upper()} resources passed {check_id}"
 
-        # Calculate risk score
         risk_score = self._calculate_group_risk_score(findings, max_severity)
 
-        # Determine recommended action
         action = self._determine_action(max_severity, failed_count)
 
-        # Collect compliance frameworks
         compliance = set()
         for f in findings:
             compliance.update(f.compliance)
 
-        # Generate LLM summary and remedy for failed findings
         summary = ""
         remedy = ""
         if failed_count > 0:
@@ -492,7 +473,6 @@ def _create_action_item(
         """Create an action item with LLM-generated CLI commands."""
         region = failed_findings[0].region if failed_findings else "us-east-1"
 
-        # Generate remediation commands using LLM
         commands = self.llm.generate_commands(
             service=group.service,
             region=region,

backend/api/graph/schema.resolvers.go

@@ -140,8 +140,6 @@ func (c *CredentialCache) refreshExpiring() {
 	for _, cred := range expiring {
 		ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
 		if _, err := c.RefreshCredentials(ctx, cred.accountID, cred.externalID); err != nil {
-			// TODO: Add proper logging when logger is available
-			// For now, silently continue to avoid crashes
 			_ = err
 		}
 		cancel()

backend/api/internal/awsauth/credentials.go

@@ -57,13 +57,7 @@ func Middleware() gin.HandlerFunc {
 		}
 
 		clientKey := os.Getenv("CLERK_SECRET_KEY")
-		// if clientKey == "" {
-		// We don't want to panic here in case of misconfiguration in dev, just warn
-		// log.Println("WARNING: CLERK_SECRET_KEY is missing")
-		// }
 
-		// If no client available (no key), or no header, just finish
-		// Ideally we should block if key is present but header missing
 		if clientKey == "" {
 			c.Next()
 			return
@@ -78,7 +72,6 @@ func Middleware() gin.HandlerFunc {
 
 		header := r.Header.Get("Authorization")
 		if header == "" {
-			// Unauthenticated
 			c.Next()
 			return
 		}
@@ -128,7 +121,6 @@ func EmailFromContext(ctx context.Context) (string, error) {
 	if user == nil {
 		return "", fmt.Errorf("not logged in")
 	}
-	// Simplified email retrieval
 	if len(user.EmailAddresses) > 0 {
 		return user.EmailAddresses[0].EmailAddress, nil
 	}

backend/api/internal/middleware/auth/auth.go

@@ -52,7 +52,6 @@ type ScanTaskResult struct {
 func (c *Coordinator) StartScan(ctx context.Context, config ScanConfig) (*ScanResult, error) {
 	startedAt := time.Now().UTC()
 
-	// Build list of scan tasks
 	var tasks []ScanTask
 	for _, region := range config.Regions {
 		for _, service := range config.Services {
@@ -68,10 +67,8 @@ func (c *Coordinator) StartScan(ctx context.Context, config ScanConfig) (*ScanRe
 		return nil, fmt.Errorf("no valid scan tasks: check that services have registered scanners")
 	}
 
-	// Execute tasks in parallel
 	results := c.executeParallel(ctx, tasks)
 
-	// Aggregate results
 	var allFindings []Finding
 	var scanErrors []error
 
@@ -83,7 +80,6 @@ func (c *Coordinator) StartScan(ctx context.Context, config ScanConfig) (*ScanRe
 		allFindings = append(allFindings, result.Findings...)
 	}
 
-	// Count passed and failed checks
 	passedChecks := 0
 	failedChecks := 0
 	for _, f := range allFindings {
@@ -120,14 +116,12 @@ func (c *Coordinator) executeParallel(ctx context.Context, tasks []ScanTask) []S
 	resultsChan := make(chan ScanTaskResult, len(tasks))
 	tasksChan := make(chan ScanTask, len(tasks))
 
-	// Start worker pool
 	for i := 0; i < maxWorkers; i++ {
 		wg.Add(1)
 		go func() {
 			defer wg.Done()
 
 			for task := range tasksChan {
-				// Check for context cancellation before processing
 				select {
 				case <-ctx.Done():
 					resultsChan <- ScanTaskResult{
@@ -140,21 +134,18 @@ func (c *Coordinator) executeParallel(ctx context.Context, tasks []ScanTask) []S
 
 				result := ScanTaskResult{Task: task}
 
-				// Create scanner for this service/region
 				factory, exists := c.scanners[task.Service]
 				if !exists {
 					result.Error = fmt.Errorf("no scanner registered for service %s", task.Service)
 					resultsChan <- result
 					continue
 				}
 
-				// Create regional config
 				regionalCfg := c.cfg.Copy()
 				regionalCfg.Region = task.Region
 
 				scanner := factory(regionalCfg, task.Region, c.accountID)
 
-				// Execute scan with context
 				findings, err := scanner.Scan(ctx, task.Region)
 				if err != nil {
 					result.Error = err
@@ -168,21 +159,18 @@ func (c *Coordinator) executeParallel(ctx context.Context, tasks []ScanTask) []S
 		}()
 	}
 
-	// Send tasks to workers
 	go func() {
 		for _, task := range tasks {
 			tasksChan <- task
 		}
 		close(tasksChan)
 	}()
 
-	// Wait for all workers to complete
 	go func() {
 		wg.Wait()
 		close(resultsChan)
 	}()
 
-	// Collect results
 	var results []ScanTaskResult
 	for result := range resultsChan {
 		results = append(results, result)

backend/api/internal/scanner/coordinator.go

@@ -34,15 +34,13 @@ export function AddAccountDialog() {
 
   const handleVerify = async () => {
     setIsVerifying(true);
-    // TODO: Call GraphQL mutation to verify account
     setTimeout(() => {
       setIsVerifying(false);
       setIsVerified(true);
     }, 2000);
   };
 
   const handleConnect = async () => {
-    // TODO: Call GraphQL mutation to connect account
     setOpen(false);
     // Reset state
     setStep(1);

frontend/app/(dashboard)/accounts/_components/add-account-dialog.tsx

@@ -66,7 +66,6 @@ export function NewScanForm() {
 
   const handleStartScan = async () => {
     setIsScanning(true);
-    // TODO: Call GraphQL mutation to start scan
     console.log("Starting scan with:", {
       account: selectedAccount,
       services: selectedServices,

frontend/app/(dashboard)/scans/_components/new-scan-form.tsx

@@ -68,8 +68,6 @@ function SidebarProvider({
   const isMobile = useIsMobile();
   const [openMobile, setOpenMobile] = React.useState(false);
 
-  // This is the internal state of the sidebar.
-  // We use openProp and setOpenProp for control from outside the component.
   const [_open, _setOpen] = React.useState(defaultOpen);
   const open = openProp ?? _open;
   const setOpen = React.useCallback(
@@ -81,18 +79,15 @@ function SidebarProvider({
         _setOpen(openState);
       }
 
-      // This sets the cookie to keep the sidebar state.
       document.cookie = `${SIDEBAR_COOKIE_NAME}=${openState}; path=/; max-age=${SIDEBAR_COOKIE_MAX_AGE}`;
     },
     [setOpenProp, open],
   );
 
-  // Helper to toggle the sidebar.
   const toggleSidebar = React.useCallback(() => {
     return isMobile ? setOpenMobile((open) => !open) : setOpen((open) => !open);
   }, [isMobile, setOpen, setOpenMobile]);
 
-  // Adds a keyboard shortcut to toggle the sidebar.
   React.useEffect(() => {
     const handleKeyDown = (event: KeyboardEvent) => {
       if (
@@ -108,8 +103,6 @@ function SidebarProvider({
     return () => window.removeEventListener("keydown", handleKeyDown);
   }, [toggleSidebar]);
 
-  // We add a state so that we can do data-state="expanded" or "collapsed".
-  // This makes it easier to style the sidebar with Tailwind classes.
   const state = open ? "expanded" : "collapsed";
 
   const contextValue = React.useMemo<SidebarContextProps>(