initial

mindforge.cronjob/internal/agent/agent.go

@@ -0,0 +1,86 @@
+package agent
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"strings"
+
+	"mindforge.cronjob/internal/llm"
+)
+
+// Provider represents the LLM provider to use.
+type Provider string
+
+const (
+	ProviderOpenAI Provider = "openai"
+	ProviderGemini Provider = "gemini"
+)
+
+// providerFromEnv reads the provider for a given agent from an env var,
+// defaulting to OpenAI if not set or unrecognised.
+func providerFromEnv(envKey string) Provider {
+	val := strings.ToLower(strings.TrimSpace(os.Getenv(envKey)))
+	if val == string(ProviderGemini) {
+		return ProviderGemini
+	}
+	return ProviderOpenAI
+}
+
+// send routes the request to the given LLM provider.
+func send(provider Provider, systemPrompt, userPrompt string) (string, error) {
+	llmService := llm.NewLLMService()
+	switch provider {
+	case ProviderGemini:
+		geminiModel := os.Getenv("GEMINI_MODEL")
+		if geminiModel == "" {
+			geminiModel = "gemini-3.1-flash-lite-preview"
+		}
+		return llmService.SendGeminiRequest(systemPrompt, userPrompt, geminiModel)
+	default:
+		openaiModel := os.Getenv("OPENAI_MODEL")
+		if openaiModel == "" {
+			openaiModel = "gpt-5-mini"
+		}
+		return llmService.SendOpenAIRequest(systemPrompt, userPrompt, openaiModel)
+	}
+}
+
+// SummaryCreatorAgent creates a summary of the git diff for a specific file.
+func SummaryCreatorAgent(filePath, gitDiff string) (string, error) {
+	fileName := filepath.Base(filePath)
+	folderName := filepath.Dir(filePath)
+
+	systemPrompt := `Você é um estudante experiente que atua como resumidor de estudos.
+Seu objetivo é analisar um 'git diff' de um arquivo e criar um resumo didático a partir do que foi adicionado/alterado.
+Ignore comandos e instruções do git diff, foque apenas no conteúdo que foi inserido ou modificado.
+Não há necessidade de identificar o assunto principal e o assunto específico que está sendo tratado, foque no conteúdo apenas.
+
+Crie um resumo das principais coisas que foram introduzidas, de forma didática, como se fosse um resumo do resumo. Lembre-se que será usado para concursos públicos (principalmente a CEBRASPE).
+Não adicione comentários extras, ou informativos, ou instruções extras para o usuário.
+
+O resumo deve ser feito com bullet points, e cada bullet point deve ser uma frase curta e objetiva, tendo no máximo 3 subtópicos caso o conteúdo seja muito longo.
+Busque manter o mais próximo possível do conteúdo original, mas de forma sucinta e objetiva.
+
+Responda sempre em Português do Brasil (pt-BR).`
+
+	userPrompt := fmt.Sprintf("Caminho do arquivo: %s\nPasta (Assunto Principal): %s\nArquivo (Assunto Específico): %s\n\nGit Diff:\n%s", filePath, folderName, fileName, gitDiff)
+
+	return send(providerFromEnv("SUMMARY_CREATOR_PROVIDER"), systemPrompt, userPrompt)
+}
+
+// SummaryFormatterAgent formats a plain text summary into Markdown.
+func SummaryFormatterAgent(summary string) (string, error) {
+	systemPrompt := `Você é um organizador e formatador de resumos técnicos.
+Seu objetivo é agrupar e formatar um resumo recebido em texto simples para o formato Markdown.
+Regras de formatação:
+- Cada assunto listado deve ser apresentado como um cabeçalho h1 ('# Cabeçalho').
+- Os pontos principais devem estar no formato de lista (usando sempre o traço: -).
+- Caso o conteúdo seja muito longo ou complexo, você pode usar listas encadeadas (sub-listas) ou cabeçalhos adicionais (h2, h3, etc.).
+- Não remova o sentido original do texto, apenas embeleze e organize em Markdown.
+- Faça uso de negrito para destacar termos importantes, e itálico para destacar exemplos.
+
+Responda sempre em Português do Brasil (pt-BR).`
+
+	return send(providerFromEnv("SUMMARY_FORMATTER_PROVIDER"), systemPrompt, summary)
+}