From aa984eb5332c2d24a0b77c36958535540fb4d957 Mon Sep 17 00:00:00 2001
From: shanexi <shane@myshell.ai>
Date: Thu, 17 Oct 2024 10:58:21 +0800
Subject: [PATCH] ShellAgentPluginInputInteger add manage choices and
 client-side form validate

---
 web/shellagent.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/web/shellagent.js b/web/shellagent.js
index 09b277a..d449ebd 100644
--- a/web/shellagent.js
+++ b/web/shellagent.js
@@ -53,7 +53,7 @@ app.registerExtension({
   },
   async beforeRegisterNodeDef(nodeType, nodeData, app) {
 
-    if (["ShellAgentPluginInputText", "ShellAgentPluginInputFloat"].indexOf(nodeData.name) > -1) {
+    if (["ShellAgentPluginInputText", "ShellAgentPluginInputFloat", "ShellAgentPluginInputInteger"].indexOf(nodeData.name) > -1) {
       chainCallback(nodeType.prototype, "onNodeCreated", function () {
         const widget = this.addWidget("string", "choices", presets, () => { }, {
           value: presets,
@@ -85,6 +85,10 @@ app.registerExtension({
           function addRow(p) {
 
             const value = document.createElement("input");
+            if (["ShellAgentPluginInputFloat", "ShellAgentPluginInputInteger"].indexOf(nodeData.name) > -1) {
+              value.type = 'number';
+            }
+
             const valueLbl = document.createElement("label");
             value.value = p;
             Object.assign(value.style, {