From cfd753f77794f6bdc801ea3b7f62fffef44d1b98 Mon Sep 17 00:00:00 2001
From: Oleg Korshul <Oleg.Korshul@onlyoffice.com>
Date: Tue, 15 Apr 2025 17:32:38 +0300
Subject: [PATCH] Add AI custom function for editors

---
 sdkjs-plugins/content/ai/scripts/code.js | 47 +++++++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/sdkjs-plugins/content/ai/scripts/code.js b/sdkjs-plugins/content/ai/scripts/code.js
index 282478b9..28bac0a8 100644
--- a/sdkjs-plugins/content/ai/scripts/code.js
+++ b/sdkjs-plugins/content/ai/scripts/code.js
@@ -45,8 +45,53 @@ window.Asc.plugin.init = async function() {
 					break;
 			}
 
-			Asc.Editor.callMethod("onAIRequest", [data]);
+			await Asc.Editor.callMethod("onAIRequest", [data]);
 		});
+
+		if ("cell" === window.Asc.plugin.info.editorType) {
+			let CustomFunctions = {
+				current : 0,
+				macrosArray : [
+					{
+						guid : "e8ea2fb288054deaa6b82158c04dee37",
+						name : "AI",
+						value : "\
+(function()\n\
+{\n\
+    /**\n\
+     * Function that returns the AI answer.\n\
+     * @customfunction\n\
+     * @param {string} value Prompt.\n\
+     * @returns {string} Answer value.\n\
+     */\n\
+    async function AI(value) {\n\
+        let systemMessage = \"As an Excel formula expert, your job is to provide advanced Excel formulas that perform complex calculations or data manipulations as described by the user. Keep your answers as brief as possible. If the user asks for formulas, return only the formula. If the user asks for something, answer briefly and only the result, without descriptions or reflections.\";\n\
+        return new Promise(resolve => (function(){\n\
+            Api.AI({ type : \"text\", data : [{role: \"system\", content: systemMessage}, {role:\"user\", content: value}] }, function(data){\n\
+                if (data.error)\n\
+                    return resolve(data.error);\n\
+                switch (data.type) {\n\
+                    case \"text\":\n\
+                    {\n\
+                        resolve(data.text);\n\
+                        break;\n\
+                    }\n\
+                    default:\n\
+                    {\n\
+                        resolve(\"#ERROR\");\n\
+                    }\n\
+                }\n\
+                resolve(data)\n\
+            });\n\
+        })());\n\
+    }\n\
+    Api.AddCustomFunction(AI);\n\
+})();"
+					}
+				]
+			};
+			await Asc.Editor.callMethod("SetCustomFunctions", [JSON.stringify(CustomFunctions)]);
+		}
 	}
 };