feat: 本地存储chatGPT令牌

orangelckc · Jan 25, 2023 · 2584e53 · 2584e53
1 parent 724c688
commit 2584e53
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 3 deletions.
diff --git a/src/api/chatGTP.ts b/src/api/chatGTP.ts
@@ -1,8 +1,10 @@
 
+import { MANAGE } from "@/constants";
+import { getStore } from "@/store";
 import { Body } from "@tauri-apps/api/http";
 import { getQueryData } from ".";
 
-const API_KEY = "";
+const API_KEY = await getStore(MANAGE.gptToken) || ''
 const MAX_TOKENS = 100;
 const MODEL = "text-davinci-003"; // 功能最全的模型
 // const MODEL = "text-curie-001	"; // 适合聊天的模型

diff --git a/src/constants/index.ts b/src/constants/index.ts
@@ -14,6 +14,7 @@ export const MANAGE = {
   follow: 'manage.follow',
   gift: 'manage.gift',
   welcome: 'manage.welcome',
+  gptToken: 'manage.gptToken'
 };
 
 export const MESSAGE_TYPE = {

diff --git a/src/views/Robot/message/robot.ts b/src/views/Robot/message/robot.ts
@@ -22,6 +22,7 @@ export const manage = reactive({
   follow: await (getStore(MANAGE.follow)) || false,
   gift: await (getStore(MANAGE.gift)) || false,
   welcome: await (getStore(MANAGE.welcome)) || false,
+  gptToken: await getStore(MANAGE.gptToken) || ''
 });
 
 // 更新store，做持久化
@@ -169,6 +170,7 @@ const init_listener = async () => {
             messages.push(...autoSlice(`@${manage.hostName} 当前粉丝数${curFans}，今日已增长${curFans - todayFans}，冲～`));
             return;
           }
+          if (manage.gptToken === '') return
           const result = await chatGTPApi(question);
           if (!result) return;
           messages.push(...autoSlice(`@${formatUname(uname)}:${result}`));
@@ -242,7 +244,7 @@ watch(active, async (value) => {
         console.log("已得到直播间信息");
         clearInterval(liveInfoIntever);
         msgInterval = setInterval(() => {
-          if (!active.value && messages.length === 0) { 
+          if (!active.value && messages.length === 0) {
             clearInterval(msgInterval);
             return;
           }

diff --git a/src/views/Robot/room.vue b/src/views/Robot/room.vue
@@ -117,10 +117,18 @@ const stopRecord = async () => {
             <q-checkbox v-model="manage.welcome" label="欢迎词" />
             <q-toggle v-model="active" size="md" :disable="!connected">
               <div class="i-carbon-machine-learning text-2xl font-bold" :class="active ? 'text-green' : 'text-gray'" />
-              <q-tooltip v-if="active">
+              <q-tooltip v-if="connected">
                 开启机器人
               </q-tooltip>
             </q-toggle>
+            <q-icon class="i-carbon-add-alt" color="primary">
+              <q-tooltip>
+                修改GPT-Token
+              </q-tooltip>
+              <q-popup-proxy>
+                <q-input v-model="manage.gptToken" dense autofocus />
+              </q-popup-proxy>
+            </q-icon>
           </div>
         </div>
       </q-card-section>