git.iko.soy/crush
on 94de6bc help sign in
crush/internal/agent/prompts.go
42 lines 1.0 kB history raw
1package agent
3import (
4 "context"
5 _ "embed"
7 "github.com/charmbracelet/crush/internal/agent/prompt"
8 "github.com/charmbracelet/crush/internal/config"
9)
11//go:embed templates/coder.md.tpl
12var coderPromptTmpl []byte
14//go:embed templates/task.md.tpl
15var taskPromptTmpl []byte
17//go:embed templates/initialize.md.tpl
18var initializePromptTmpl []byte
20func coderPrompt(opts ...prompt.Option) (*prompt.Prompt, error) {
21 systemPrompt, err := prompt.NewPrompt("coder", string(coderPromptTmpl), opts...)
22 if err != nil {
23 return nil, err
24 }
25 return systemPrompt, nil
28func taskPrompt(opts ...prompt.Option) (*prompt.Prompt, error) {
29 systemPrompt, err := prompt.NewPrompt("task", string(taskPromptTmpl), opts...)
30 if err != nil {
31 return nil, err
32 }
33 return systemPrompt, nil
36func InitializePrompt(cfg config.Config) (string, error) {
37 systemPrompt, err := prompt.NewPrompt("initialize", string(initializePromptTmpl))
38 if err != nil {
39 return "", err
40 }
41 return systemPrompt.Build(context.Background(), "", "", cfg)