mlow/lmcli
Private
Public Access
1
0
Fork 0
Code Activity

Add conversations.titleGenerationPrompt configuration

Browse Source
This commit is contained in:
Matt Low
2025-07-29 15:13:53 +00:00
parent 0cf0a4ff0d
commit a945223037
2 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pkg/cmd/util/util.go
View File

@@ -162,8 +162,7 @@ func FormatForExternalPrompt(messages []conversation.Message, system bool) strin
return sb.String()
}
func GenerateTitle(ctx *lmcli.Context, messages []conversation.Message) (string, error) {
const systemPrompt = `You will be shown a conversation between a user and an AI assistant. Your task is to generate a short title (8 words or less) for the provided conversation that reflects the conversation's topic. Your response is expected to be in JSON in the format shown below.
const defaultTitleGenerationPrompt = `You will be shown a conversation between a user and an AI assistant. Your task is to generate a short title (8 words or less) for the provided conversation that reflects the conversation's topic. Your response is expected to be in JSON in the format shown below.
Example conversation:
@@ -173,6 +172,19 @@ Example response:
{"title": "Help with math homework"}
`
func GenerateTitle(ctx *lmcli.Context, messages []conversation.Message) (string, error) {
if ctx.Config.Conversations.TitleGenerationModel == nil {
return "", nil
}
var systemPrompt string
if ctx.Config.Conversations.TitleGenerationPrompt != nil {
systemPrompt = *ctx.Config.Conversations.TitleGenerationPrompt
} else {
systemPrompt = defaultTitleGenerationPrompt
}
type msg struct {
Role string
Content string