SuperCoder/internal/llm/prompt/prompt.go

package prompt

import (
	"fmt"
	"os"
	"path/filepath"

	"github.com/opencode-ai/opencode/internal/config"
	"github.com/opencode-ai/opencode/internal/llm/models"
)

// contextFiles is a list of potential context files to check for
var contextFiles = []string{
	".github/copilot-instructions.md",
	".cursorrules",
	"CLAUDE.md",
	"CLAUDE.local.md",
	"opencode.md",
	"opencode.local.md",
	"OpenCode.md",
	"OpenCode.local.md",
	"OPENCODE.md",
	"OPENCODE.local.md",
}

func GetAgentPrompt(agentName config.AgentName, provider models.ModelProvider) string {
	basePrompt := ""
	switch agentName {
	case config.AgentCoder:
		basePrompt = CoderPrompt(provider)
	case config.AgentTitle:
		basePrompt = TitlePrompt(provider)
	case config.AgentTask:
		basePrompt = TaskPrompt(provider)
	default:
		basePrompt = "You are a helpful assistant"
	}

	if agentName == config.AgentCoder || agentName == config.AgentTask {
		// Add context from project-specific instruction files if they exist
		contextContent := getContextFromFiles()
		if contextContent != "" {
			return fmt.Sprintf("%s\n\n# Project-Specific Context\n%s", basePrompt, contextContent)
		}
	}
	return basePrompt
}

// getContextFromFiles checks for the existence of context files and returns their content
func getContextFromFiles() string {
	workDir := config.WorkingDirectory()
	var contextContent string

	for _, file := range contextFiles {
		filePath := filepath.Join(workDir, file)
		content, err := os.ReadFile(filePath)
		if err == nil {
			contextContent += fmt.Sprintf("\n%s\n", string(content))
		}
	}

	return contextContent
}