From 2fc1aec8ab357e70ffdbba31aa85d435b37e5b7a Mon Sep 17 00:00:00 2001
From: Vanessa <lly219@gmail.com>
Date: Tue, 21 Mar 2023 19:18:44 +0800
Subject: [PATCH] :art: https://github.com/siyuan-note/siyuan/issues/7714

---
 app/src/config/AI.ts     | 52 +++++++++++++++++++++++++++++++++++-----
 app/src/types/index.d.ts |  9 +++++++
 2 files changed, 55 insertions(+), 6 deletions(-)

diff --git a/app/src/config/AI.ts b/app/src/config/AI.ts
index af4b098bc..32da67e22 100644
--- a/app/src/config/AI.ts
+++ b/app/src/config/AI.ts
@@ -1,24 +1,64 @@
+import {fetchPost} from "../util/fetch";
 
 export const AI = {
     element: undefined as Element,
     genHTML: () => {
         return `<label class="fn__flex b3-label">
     <div class="fn__flex-1">
-        ${window.siyuan.languages.selectOpen}
-        <div class="b3-label__text">${window.siyuan.languages.fileTree2}</div>
+        APIKey
+        <div class="b3-label__text">APIKeyAPIKeyAPIKey</div>
     </div>
     <span class="fn__space"></span>
-    <input class="b3-switch fn__flex-center" id="alwaysSelectOpenedFile" type="checkbox"${window.siyuan.config.fileTree.alwaysSelectOpenedFile ? " checked" : ""}/>
+    <input class="b3-text-field fn__flex-center" id="apiKey" value="${window.siyuan.config.ai.openAI.apiKey}"/>
 </label>
 <label class="fn__flex b3-label">
     <div class="fn__flex-1">
-        ${window.siyuan.languages.fileTree7}
-        <div class="b3-label__text">${window.siyuan.languages.fileTree8}</div>
+        APITimeout
+        <div class="b3-label__text">APITimeoutAPITimeoutAPITimeout</div>
     </div>
     <span class="fn__space"></span>
-    <input class="b3-switch fn__flex-center" id="openFilesUseCurrentTab" type="checkbox"${window.siyuan.config.fileTree.openFilesUseCurrentTab ? " checked" : ""}/>
+    <input class="b3-text-field fn__flex-center" type="number" step="1" min="1" id="openFilesUseCurrentTab" value="${window.siyuan.config.ai.openAI}"/>
+</label>
+<label class="fn__flex b3-label">
+    <div class="fn__flex-1">
+        APIProxy
+        <div class="b3-label__text">APIProxyAPIProxyAPIProxy</div>
+    </div>
+    <span class="fn__space"></span>
+    <input class="b3-text-field fn__flex-center" id="apiProxy" value="${window.siyuan.config.ai.openAI.apiProxy}"/>
+</label>
+<label class="fn__flex b3-label">
+    <div class="fn__flex-1">
+        APIMaxTokens
+        <div class="b3-label__text">APIMaxTokensAPIMaxTokensAPIMaxTokens</div>
+    </div>
+    <span class="fn__space"></span>
+    <input class="b3-text-field fn__flex-center" type="number" step="1" min="0" id="apiMaxTokens" value="${window.siyuan.config.ai.openAI.apiMaxTokens}"/>
+</label>
+<label class="fn__flex b3-label">
+    <div class="fn__flex-1">
+        APIBaseURL
+        <div class="b3-label__text">APIBaseURLAPIBaseURLAPIBaseURL</div>
+    </div>
+    <span class="fn__space"></span>
+    <input class="b3-text-field fn__flex-center" id="apiBaseURL" value="${window.siyuan.config.ai.openAI.apiBaseURL}"/>
 </label>`;
     },
     bindEvent: () => {
+        AI.element.querySelectorAll("input").forEach((item) => {
+            item.addEventListener("change", () => {
+                fetchPost("/api/setting/setAI", {
+                    openAI: {
+                        apiBaseURL: (AI.element.querySelector("#apiBaseURL") as HTMLInputElement).checked,
+                        apiKey: (AI.element.querySelector("#apiKey") as HTMLInputElement).value,
+                        apiMaxTokens: (AI.element.querySelector("#apiMaxTokens") as HTMLInputElement).value,
+                        apiProxy: (AI.element.querySelector("#apiProxy") as HTMLInputElement).checked,
+                        apiTimeout: (AI.element.querySelector("#apiTimeout") as HTMLInputElement).checked,
+                    }
+                }, response => {
+                    window.siyuan.config.ai = response.data;
+                });
+            });
+        });
     },
 };
diff --git a/app/src/types/index.d.ts b/app/src/types/index.d.ts
index fc4570115..a18373a27 100644
--- a/app/src/types/index.d.ts
+++ b/app/src/types/index.d.ts
@@ -392,6 +392,15 @@ declare interface IConfig {
     repo: {
         key: string
     },
+    ai: {
+        openAI: {
+            apiBaseURL: string
+            apiKey: string
+            apiMaxTokens: number
+            apiProxy: string
+            apiTimeout: number
+        },
+    }
     sync: {
         generateConflictDoc: boolean
         enabled: boolean