Function groupFilesByModelWithSource

cmd/analyze-batch/main.go:211–259

groupFilesByModelWithSource groups result files by model name and determines batch source

(files []string, batchDirs []string)

// groupFilesByModelWithSource groups result files by model name and determines batch source
func groupFilesByModelWithSource(files []string, batchDirs []string) map[string]ModelFileInfo {
	modelFiles := make(map[string]ModelFileInfo)

	// Pattern to extract model name from filename
	pattern := regexp.MustCompile(`^(.+?)_agent_test_results_`)

	for _, file := range files {
		basename := filepath.Base(file)
		matches := pattern.FindStringSubmatch(basename)

		var modelName string
		if len(matches) > 1 {
			modelName = matches[1]
		} else {
			// Fallback: try to extract model name from the middle part
			parts := strings.Split(basename, "_")
			if len(parts) >= 4 {
				modelName = parts[0]
			} else {
				modelName = "unknown"
			}
		}

		// Determine which batch directory this file came from
		var batchSource string
		for _, batchDir := range batchDirs {
			if strings.HasPrefix(file, batchDir) {
				batchSource = batchDir
				break
			}
		}
		if batchSource == "" {
			batchSource = "unknown"
		}

		// Get existing info or create new
		info := modelFiles[modelName]
		info.files = append(info.files, file)
		if info.batchSource == "" {
			info.batchSource = batchSource
		} else if info.batchSource != batchSource {
			// Model appears in multiple batches, combine the sources
			info.batchSource = info.batchSource + "," + batchSource
		}
		modelFiles[modelName] = info
	}

	return modelFiles
}

// analyzeModel analyzes all result files for a single model
func analyzeModel(modelName string, files []string) (*ModelAnalysis, error) {