fix(openai): user and assistant roles should be alternating (#859)

2024-11-17 03:49:02 +08:00
parent ff85b9c1e2
commit 9891b03656
2 changed files with 25 additions and 6 deletions
--- a/lua/avante/providers/gemini.lua
+++ b/lua/avante/providers/gemini.lua
@@ -19,12 +19,15 @@ M.parse_messages = function(opts)
  vim.iter(opts.messages):each(function(message)
    local role = message.role
    if role == prev_role then
-      if role == "user" then
-        table.insert(contents, { role = "model", parts = {
-          { text = "Ok, I understand." },
-        } })
+      if role == M.role_map["user"] then
+        table.insert(
+          contents,
+          { role = M.role_map["assistant"], parts = {
+            { text = "Ok, I understand." },
+          } }
+        )
      else
-        table.insert(contents, { role = "user", parts = {
+        table.insert(contents, { role = M.role_map["user"], parts = {
          { text = "Ok" },
        } })
      end