nvim/nvim-config/ai.lua

local ok, gen = pcall(require, 'gen')
if not ok then
  return
end

gen.prompts['DevOps me!'] = {
  prompt = "You are a senior devops engineer, acting as an assistant. You offer help with cloud technologies like: Terraform, AWS, kubernetes, python. You answer with code examples when possible. $input:\n$text",
  replace = true
}

gen.prompts['Code helper'] = {
  prompt = "You are a senior code engineer, acting as an assistant. You offer help with coding technologies like: Kotlin, Java, kubernetes, quarkus, spring, spring-boot. You answer with code examples when possible. $input:\n$text",
  replace = true,
  model = 'codellama'
}