Skip to content

Commit 274b2c6

Browse files
authored
fix(chat): remove images from title prompt (#2586)
Co-authored-by: Oli Morris <[email protected]>
1 parent 92dc15b commit 274b2c6

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

lua/codecompanion/interactions/background/builtin/chat_make_title.lua

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,9 +7,25 @@ local M = {}
77
---Format the messages from a chat buffer
88
---@param messages CodeCompanion.Chat.Messages
99
function M.format_messages(messages)
10+
local exclude_tags = {
11+
["image"] = "[Image content omitted]",
12+
["rules"] = "",
13+
["system_prompt_from_config"] = "",
14+
}
15+
1016
local chat_messages = {}
11-
for _, message in ipairs(messages or {}) do
12-
table.insert(chat_messages, fmt("## %s\n%s", message.role, message.content))
17+
for _, m in ipairs(messages or {}) do
18+
local tag = m._meta and m._meta.tag
19+
local replacement = exclude_tags[tag]
20+
21+
if replacement == "" then
22+
goto continue
23+
end
24+
25+
local content = replacement or m.content
26+
table.insert(chat_messages, fmt("## %s\n%s", m.role, content))
27+
28+
::continue::
1329
end
1430
return table.concat(chat_messages, "\n")
1531
end

0 commit comments

Comments
 (0)