feat: add proxy prompt

onepointAI · May 12, 2023 · 7d99f7e · 7d99f7e
1 parent 863a1d5
commit 7d99f7e
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 2 deletions.
diff --git a/src/components/Setting/Account/index.tsx b/src/components/Setting/Account/index.tsx
@@ -79,7 +79,7 @@ export default function () {
             }}
           >
             <Form.Item label={t('Base Path')} name="basePath">
-              <Input placeholder="Like https://api.openai.com/v1" />
+              <Input placeholder="Like https://api.openai.com" />
             </Form.Item>
             <Form.Item label="ApiKey" name="apikey">
               <Input.Password

diff --git a/src/electron/apis/prompt.ts b/src/electron/apis/prompt.ts
@@ -64,8 +64,9 @@ export default async (req: any, res: any) => {
       res.end()
     })
   } catch (e) {
+    Logger.error(e)
     if (e instanceof Error) {
-      res.write(e.message)
+      res.write(String(ERR_CODES.NETWORK_CONGESTION))
       res.end()
     } else {
       res.write(JSON.stringify(e))

diff --git a/src/electron/types.ts b/src/electron/types.ts
@@ -17,3 +17,5 @@ export enum ERR_CODES {
   NOT_SET_APIKEY = -998,
   TOKEN_TOO_LONG = -997,
 }
+
+export const CODE_SPLIT = '&^.>'
diff --git a/src/features/chat/chatSlice.ts b/src/features/chat/chatSlice.ts
@@ -135,6 +135,13 @@ export const fetchChatResp = createAsyncThunk(
             reject(new Error('please set your apikey first.'))
             break
           }
+
+          if (Number(value) === ERR_CODES.NETWORK_CONGESTION) {
+            reject(
+              new Error('Network error. Check whether you have set up a proxy')
+            )
+            break
+          }
           str += value
           dispatch(
             saveResp({

diff --git a/src/locales/en.json b/src/locales/en.json
@@ -42,6 +42,7 @@
   "Plugins": "Plugins",
   "please set your apikey first.": "please set your apikey first.",
   "High network latency. Check whether you have set up a proxy": "High network latency. Check whether you have set up a proxy",
+  "Network error. Check whether you have set up a proxy": "Network error. Check whether you have set up a proxy",
   "Prompt Setting": "Prompt Setting",
   "Chat mode, feel free to ask any questions you want.": "Chat mode, feel free to ask any questions you want.",
   "Code Master, generate or refactor the code you want.": "Code Master, generate or refactor the code you want.",

diff --git a/src/locales/zh.json b/src/locales/zh.json
@@ -42,6 +42,7 @@
   "Plugins": "插件",
   "please set your apikey first.": "请先设置你的apikey",
   "High network latency. Check whether you have set up a proxy": "网络拥堵，请检查是否设置了代理。",
+  "Network error. Check whether you have set up a proxy": "网络错误，请检查是否设置了代理。",
   "Prompt Setting": "Prompt设置",
   "Chat mode, feel free to ask any questions you want.": "聊天模式，随意提出任何你想问的问题。",
   "Code Master, generate or refactor the code you want.": "编程模式，生成或重构你想要的代码。",