diff --git a/content.go b/content.go index 5bd15e6bf4b599d056a7d82de445d3a00bbf0f2a..0edd8410e36057f978791e553ab6bf6e89fc4864 100644 --- a/content.go +++ b/content.go @@ -533,11 +533,8 @@ func (p ProviderDefinedTool) GetName() string { // NewUserMessage creates a new user message with the given prompt and optional files. func NewUserMessage(prompt string, files ...FilePart) Message { - content := []MessagePart{ - TextPart{ - Text: prompt, - }, - } + content := make([]MessagePart, 0, len(files)+1) + content = append(content, TextPart{Text: prompt}) for _, f := range files { content = append(content, f)