Files
mindforge/mindforge.cronjob/internal/agent/agent.go

48 lines
2.5 KiB
Go

package agent
import (
"fmt"
"path/filepath"
"mindforge.cronjob/internal/llm"
)
// 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 llm.NewLLMService().Send(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 llm.NewLLMService().Send(systemPrompt, summary)
}