DRAFT: Separate section for persona's AI Bot options

Author: romanrizzi

assets/javascripts/discourse/admin/models/ai-persona.js

@@ -1,4 +1,3 @@
-import { tracked } from "@glimmer/tracking";
 import { ajax } from "discourse/lib/ajax";
 import RestModel from "discourse/models/rest";
 
@@ -63,32 +62,39 @@ const SYSTEM_ATTRIBUTES = [
   "allow_chat_direct_messages",
 ];
 
-class ToolOption {
-  @tracked value = null;
-}
-
 export default class AiPersona extends RestModel {
   // this code is here to convert the wire schema to easier to work with object
   // on the wire we pass in/out tools as an Array.
   // [[ToolName, {option1: value, option2: value}, force], ToolName2, ToolName3]
   // So we rework this into a "tools" property and nested toolOptions
   init(properties) {
     this.forcedTools = [];
+    this.toolOptions = {};
+
     if (properties.tools) {
       properties.tools = properties.tools.map((tool) => {
         if (typeof tool === "string") {
           return tool;
         } else {
           let [toolId, options, force] = tool;
-          for (let optionId in options) {
+          const mappedOptions = {};
+
+          for (const optionId in options) {
             if (!options.hasOwnProperty(optionId)) {
               continue;
             }
-            this.getToolOption(toolId, optionId).value = options[optionId];
+
+            mappedOptions[optionId] = options[optionId];
+          }
+
+          if (Object.keys(mappedOptions).length > 0) {
+            this.toolOptions[toolId] = mappedOptions;
           }
+
           if (force) {
             this.forcedTools.push(toolId);
           }
+
           return toolId;
         }
       });
@@ -109,63 +115,41 @@ export default class AiPersona extends RestModel {
     return this.user;
   }
 
-  getToolOption(toolId, optionId) {
-    this.toolOptions ||= {};
-    this.toolOptions[toolId] ||= {};
-    return (this.toolOptions[toolId][optionId] ||= new ToolOption());
-  }
-
-  populateToolOptions(attrs) {
-    if (!attrs.tools) {
-      return;
-    }
-    let toolsWithOptions = [];
-    attrs.tools.forEach((toolId) => {
-      if (typeof toolId !== "string") {
-        toolId = toolId[0];
-      }
-
-      let force = this.forcedTools.includes(toolId);
-      if (this.toolOptions && this.toolOptions[toolId]) {
-        let options = this.toolOptions[toolId];
-        let optionsWithValues = {};
-        for (let optionId in options) {
-          if (!options.hasOwnProperty(optionId)) {
-            continue;
-          }
-          let option = options[optionId];
-          optionsWithValues[optionId] = option.value;
-        }
-        toolsWithOptions.push([toolId, optionsWithValues, force]);
-      } else {
-        toolsWithOptions.push([toolId, {}, force]);
-      }
+  flattenedToolStructure(data) {
+    return data.tools.map((tName) => {
+      return [tName, data.toolOptions[tName], data.forcedTools.includes(tName)];
     });
-    attrs.tools = toolsWithOptions;
   }
 
   updateProperties() {
-    let attrs = this.system
+    const attrs = this.system
       ? this.getProperties(SYSTEM_ATTRIBUTES)
       : this.getProperties(CREATE_ATTRIBUTES);
     attrs.id = this.id;
-    this.populateToolOptions(attrs);
+
     return attrs;
   }
 
   createProperties() {
-    let attrs = this.getProperties(CREATE_ATTRIBUTES);
-    this.populateToolOptions(attrs);
-    return attrs;
+    return this.getProperties(CREATE_ATTRIBUTES);
   }
 
-  workingCopy() {
-    let attrs = this.getProperties(CREATE_ATTRIBUTES);
-    this.populateToolOptions(attrs);
+  fromPOJO(data) {
+    const dataClone = JSON.parse(JSON.stringify(data));
+
+    const persona = AiPersona.create(dataClone);
+    persona.tools = this.flattenedToolStructure(dataClone);
 
-    const persona = AiPersona.create(attrs);
-    persona.forcedTools = (this.forcedTools || []).slice();
-    persona.forced_tool_count = this.forced_tool_count || -1;
     return persona;
   }
+
+  toPOJO() {
+    const attrs = this.getProperties(CREATE_ATTRIBUTES);
+    attrs.tools = this.tools;
+    attrs.forcedTools = (this.forcedTools || []).slice();
+    attrs.toolOptions = Object.assign({}, this.toolOptions);
+    attrs.forced_tool_count = this.forced_tool_count || -1;
+
+    return attrs;
+  }
 }

assets/javascripts/discourse/components/ai-forced-tool-strategy-selector.gjs

@@ -0,0 +1,17 @@
+import { hash } from "@ember/helper";
+import ComboBox from "select-kit/components/combo-box";
+
+const AiLlmSelector = <template>
+  <ComboBox
+    @value={{@value}}
+    @content={{@llms}}
+    @onChange={{@onChange}}
+    @options={{hash
+      filterable=true
+      none="discourse_ai.ai_persona.no_llm_selected"
+    }}
+    class={{@class}}
+  />
+</template>;
+
+export default AiLlmSelector;