Function SummarizeFile

pkg/aiusechat/google/google-summarize.go:166–283

SummarizeFile reads a file and generates a summary using Google's Generative AI. It supports images, PDFs, and text files based on the limits defined in wshcmd-ai.go. Returns the summary text, usage information, and any error encountered.

(ctx context.Context, filename string, opts SummarizeOpts)

// It supports images, PDFs, and text files based on the limits defined in wshcmd-ai.go.
// Returns the summary text, usage information, and any error encountered.
func SummarizeFile(ctx context.Context, filename string, opts SummarizeOpts) (string, *GoogleUsage, error) {
	if opts.Mode == "" {
		return "", nil, fmt.Errorf("mode is required")
	}

	// Read the file
	data, err := os.ReadFile(filename)
	if err != nil {
		return "", nil, fmt.Errorf("reading file: %w", err)
	}

	// Detect MIME type
	mimeType := detectMimeType(data)

	isPDF := mimeType == "application/pdf"
	isImage := strings.HasPrefix(mimeType, "image/")

	if !isPDF && !isImage {
		mimeType = "text/plain"
		if utilfn.ContainsBinaryData(data) {
			return "", nil, fmt.Errorf("file contains binary data and cannot be summarized")
		}
	}

	// Validate file size
	maxSize, sizeStr := getMaxFileSize(mimeType, opts.Mode)
	if len(data) > maxSize {
		return "", nil, fmt.Errorf("file exceeds maximum size of %s for %s files", sizeStr, mimeType)
	}

	// Create client
	client, err := genai.NewClient(ctx, option.WithAPIKey(opts.APIKey))
	if err != nil {
		return "", nil, fmt.Errorf("creating Google AI client: %w", err)
	}
	defer client.Close()

	// Create model
	model := client.GenerativeModel(SummarizeModel)

	// Select prompt based on mode
	var prompt string
	switch opts.Mode {
	case ModeQuickSummary:
		prompt = QuickSummaryPrompt
	case ModeUseful:
		prompt = UsefulSummaryPrompt
	case ModePublicCode:
		prompt = PublicCodeSummaryPrompt
	case ModeHTMLContent:
		prompt = HTMLContentPrompt
	case ModeHTMLFull:
		prompt = HTMLFullPrompt
	default:
		prompt = SummarizePrompt
	}

	// Prepare the content parts