autocrossbow/cmd/acb/summarize.go

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"os"
	"time"

	"o5r.ca/autocrossbow/contributions"
	"o5r.ca/autocrossbow/issues"
	"o5r.ca/autocrossbow/issues/vikunja"
)

const defaultPrompt = `I will provide you, for a given period, with an employee name and a list of Pull Request titles and summaries split by repository, and a list of Jira Issues an employee has worked on. I may also provide, optionally, the employee's self-assessment. If I do, integrate that.

I'd like you to summarize the employee's accomplishments for the quarter
I'd like the summary for the accomplishments to be in prose form, in a few paragraphs separated based on areas of work. Keep answers to 500 words for the summary.`

// Summarizer interface defines the contract for summarization implementations
type Summarizer interface {
	Summarize(prompt string, endpoint string, token string, model string) (string, error)
}

// OpenAISummarizer implements the Summarizer interface for OpenAI-compatible endpoints
type OpenAISummarizer struct{}

// Summarize sends the prompt to an OpenAI-compatible endpoint for summarization
func (o *OpenAISummarizer) Summarize(fullPrompt string, openaiEndpoint string, openaiToken string, openaiModel string) (string, error) {
	// Check if required environment variables are set
	if openaiEndpoint == "" || openaiToken == "" {
		return "", fmt.Errorf("OpenAI endpoint or token not set")
	}

	// Create a JSON payload for the OpenAI API
	payload := struct {
		Model    string `json:"model"`
		Messages []struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		} `json:"messages"`
	}{
		Model: openaiModel,
		Messages: []struct {
			Role    string `json:"role"`
			Content string `json:"content"`
		}{{Role: "system", Content: fullPrompt}},
	}

	jsonPayload, err := json.Marshal(payload)
	fmt.Println(string(jsonPayload))
	if err != nil {
		return "", err
	}

	// Create a POST request to the OpenAI endpoint with JSON body
	req, err := http.NewRequest("POST", openaiEndpoint, bytes.NewBuffer(jsonPayload))
	if err != nil {
		return "", err
	}
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", openaiToken))

	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		return "", err
	}
	defer resp.Body.Close()

	body, err := io.ReadAll(resp.Body)
	if err != nil {
		return "", err
	}

	return string(body), nil
}

// buildPrompt constructs the prompt string from PRs, issues, and tasks
func buildPrompt(employeename string, prs map[string][]contributions.PullRequest, issues []issues.Issue, tasks []vikunja.Task, prompt string) string {
	// Build a prompt string
	fullPrompt := prompt + fmt.Sprintf("\n\nHere's the PRs and Tickets for the employee %s:\n\n", employeename)
	for repo, prList := range prs {
		fullPrompt += fmt.Sprintf("Repository: %s\n", repo)
		for _, pr := range prList {
			fullPrompt += fmt.Sprintf("- Title: %s\n", pr.Title)
			fullPrompt += fmt.Sprintf("  Body: %s\n", pr.Body)
		}
	}
	fullPrompt += "Issues:\n"
	for _, issue := range issues {
		fullPrompt += fmt.Sprintf("Summary: %s\n", issue.Summary)
		fullPrompt += fmt.Sprintf("Description: %s\n", issue.Description)
		fullPrompt += "--------\n"
	}
	
	// Save prompt to file for debugging
	promptf, err := os.Create(fmt.Sprintf("prompt-%s-%d.json", employeename, time.Now().Unix()))
	if err != nil {
		fmt.Println(fmt.Errorf("error creating PR file: %w", err))
		os.Exit(1)
	}
	promptf.WriteString(fullPrompt)
	defer promptf.Close()
	
	return fullPrompt
}

// SummarizeData builds the prompt and calls the summarization endpoint
func SummarizeData(employeename string, prs map[string][]contributions.PullRequest, issues []issues.Issue, tasks []vikunja.Task, prompt string, openaiEndpoint string, openaiToken string, openaiModel string, summarizer Summarizer) (string, error) {
	// Build the prompt
	fullPrompt := buildPrompt(employeename, prs, issues, tasks, prompt)
	
	// Always call the summarizer's Summarize method
	result, err := summarizer.Summarize(fullPrompt, openaiEndpoint, openaiToken, openaiModel)
	if err != nil {
		return "", err
	}
	
	return result, nil
}