Skip to content

Commit 4b17557

Browse files
committed
fix: olimorris#117 not resolving adapter role
1 parent c19fe7b commit 4b17557

File tree

2 files changed

+2
-10
lines changed

2 files changed

+2
-10
lines changed

lua/codecompanion/adapters.lua

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -234,17 +234,9 @@ end
234234
---@param messages table
235235
---@return table
236236
function Adapter:map_roles(messages)
237-
local roles = config.strategies.chat.roles
238-
local map = {
239-
[roles.llm:lower()] = self.args.roles.llm,
240-
[roles.user:lower()] = self.args.roles.user,
241-
}
242-
243237
for _, message in ipairs(messages) do
244238
if message.role then
245-
message.role = message.role:lower()
246-
-- Pass through the role if it doesn't exist in the map
247-
message.role = map[message.role] or message.role
239+
message.role = self.args.roles[message.role:lower()] or message.role
248240
end
249241
end
250242

lua/codecompanion/strategies/chat.lua

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -843,7 +843,7 @@ function Chat:submit()
843843

844844
lock_buf(bufnr)
845845
log:info("Chat request started")
846-
self.current_request = client.new():stream(settings, self.messages, function(err, data)
846+
self.current_request = client.new():stream(settings, self.adapter:map_roles(self.messages), function(err, data)
847847
if err then
848848
log:error("Error: %s", err)
849849
return self:reset()

0 commit comments

Comments
 (0)