add EcaChatSelectModel and EcaChatSelectBehavior commands

Author: joaopluigi

lua/eca/commands.lua

@@ -346,4 +346,34 @@ function M.setup()
   Logger.debug("ECA commands registered")
 end
 
+vim.api.nvim_create_user_command("EcaChatSelectModel", function()
+  local state = require("eca.state"):new()
+  local models = state.config.models.list
+
+  vim.ui.select(models, {
+    prompt = "Select ECA Chat Model:",
+  }, function(choice)
+    if choice then
+      state:update_selected_model(choice)
+    end
+  end)
+end, {
+  desc = "Select Current ECA Chat model",
+})
+
+vim.api.nvim_create_user_command("EcaChatSelectBehavior", function()
+  local state = require("eca.state"):new()
+  local behaviors = state.config.behaviors.list
+
+  vim.ui.select(behaviors, {
+    prompt = "Select ECA Chat Behavior:",
+  }, function(choice)
+    if choice then
+      state:update_selected_behavior(choice)
+    end
+  end)
+end, {
+  desc = "Select Current ECA Chat behavior",
+})
+
 return M

lua/eca/sidebar.lua

@@ -1279,6 +1279,8 @@ function M:_send_message(message)
     requestId = tostring(os.time()),
     message = message,
     contexts = contexts or {},
+    model = self.mediator:selected_model(),
+    behavior = self.mediator:selected_behavior(),
   }, function(err, result)
     if err then
       print("err is " .. err)

lua/eca/state.lua

@@ -168,7 +168,7 @@ end
 function State:_update_usage(usage)
   self.usage = {
     tokens = {
-      limit = (usage.limit and usage.limit.output) or self.usage.tokens.limit,
+      limit = (usage.limit and usage.limit.context) or self.usage.tokens.limit,
       session = usage.sessionTokens or self.usage.tokens.session,
     },
     costs = {
@@ -198,4 +198,28 @@ function State:_update_tools(tool)
   end)
 end
 
+function State:update_selected_model(model)
+  if not model or type(model) ~= "string" then
+    return
+  end
+
+  self.config.models.selected = model
+
+  vim.schedule(function()
+    require("eca.observer").notify({ type = "state/updated", content = { config = vim.deepcopy(self.config) } })
+  end)
+end
+
+function State:update_selected_behavior(behavior)
+  if not behavior or type(behavior) ~= "string" then
+    return
+  end
+
+  self.config.behaviors.selected = behavior
+
+  vim.schedule(function()
+    require("eca.observer").notify({ type = "state/updated", content = { config = vim.deepcopy(self.config) } })
+  end)
+end
+
 return State