Add support for kimi.moonshot.cn (#656)

* Add support for kimi.moonshot.cn * some improvements --------- Co-authored-by: josc146 <[email protected]>
josStorer · Mar 22, 2024 · 21b8468 · 21b8468
1 parent c00b8ff
commit 21b8468
Show file tree

Hide file tree

Showing 7 changed files with 659 additions and 1 deletion.
diff --git a/src/background/index.mjs b/src/background/index.mjs
@@ -23,6 +23,7 @@ import {
   chatgptApiModelKeys,
   chatgptWebModelKeys,
   claudeWebModelKeys,
+  moonshotWebModelKeys,
   customApiModelKeys,
   defaultConfig,
   getUserConfig,
@@ -46,6 +47,7 @@ import { registerCommands } from './commands.mjs'
 import { generateAnswersWithBardWebApi } from '../services/apis/bard-web.mjs'
 import { generateAnswersWithClaudeWebApi } from '../services/apis/claude-web.mjs'
 import { generateAnswersWithMoonshotCompletionApi } from '../services/apis/moonshot-api.mjs'
+import { generateAnswersWithMoonshotWebApi } from '../services/apis/moonshot-web.mjs'
 
 function setPortProxy(port, proxyTabId) {
   port.proxy = Browser.tabs.connect(proxyTabId)
@@ -161,6 +163,14 @@ async function executeApi(session, port, config) {
       config.moonshotApiKey,
       session.modelName,
     )
+  } else if (moonshotWebModelKeys.includes(session.modelName)) {
+    await generateAnswersWithMoonshotWebApi(
+      port,
+      session.question,
+      session,
+      config,
+      session.modelName,
+    )
   }
 }
 

diff --git a/src/config/index.mjs b/src/config/index.mjs
@@ -31,6 +31,7 @@ export const chatgptWebModelKeys = [
 export const bingWebModelKeys = ['bingFree4', 'bingFreeSydney']
 export const bardWebModelKeys = ['bardWebFree']
 export const claudeWebModelKeys = ['claude2WebFree']
+export const moonshotWebModelKeys = ['moonshotWebFree']
 export const gptApiModelKeys = ['gptApiInstruct', 'gptApiDavinci']
 export const chatgptApiModelKeys = [
   'chatgptApi35',
@@ -105,6 +106,8 @@ export const Models = {
   bingFree4: { value: '', desc: 'Bing (Web, GPT-4)' },
   bingFreeSydney: { value: '', desc: 'Bing (Web, GPT-4, Sydney)' },
 
+  moonshotWebFree: { value: '', desc: 'Kimi.Moonshot (Web, 100k)' },
+
   bardWebFree: { value: '', desc: 'Gemini (Web)' },
 
   chatglmTurbo: { value: 'chatglm_turbo', desc: 'ChatGLM (ChatGLM-Turbo)' },
@@ -230,6 +233,7 @@ export const defaultConfig = {
     'chatgptApi4_8k',
     'claude2WebFree',
     'bingFree4',
+    'moonshotWebFree',
     'chatglmTurbo',
     'customModel',
     'azureOpenAi',
@@ -255,6 +259,8 @@ export const defaultConfig = {
   chatgptTabId: 0,
   chatgptArkoseReqUrl: '',
   chatgptArkoseReqForm: '',
+  kimiMoonShotRefreshToken: '',
+  kimiMoonShotAccessToken: '',
 
   // unchangeable
 
@@ -339,6 +345,11 @@ export function isUsingAzureOpenAi(configOrSession) {
 export function isUsingClaude2Api(configOrSession) {
   return claudeApiModelKeys.includes(configOrSession.modelName)
 }
+
+export function isUsingMoonshotWeb(configOrSession) {
+  return moonshotWebModelKeys.includes(configOrSession.modelName)
+}
+
 export function isUsingGithubThirdPartyApi(configOrSession) {
   return githubThirdPartyApiModelKeys.includes(configOrSession.modelName)
 }

diff --git a/src/content-script/index.jsx b/src/content-script/index.jsx
@@ -10,6 +10,7 @@ import {
   getPreferredLanguageKey,
   getUserConfig,
   setAccessToken,
+  setUserConfig,
 } from '../config/index.mjs'
 import {
   createElementAtPosition,
@@ -289,7 +290,14 @@ async function prepareForStaticCard() {
 }
 
 async function overwriteAccessToken() {
-  if (location.hostname !== 'chat.openai.com') return
+  if (location.hostname !== 'chat.openai.com') {
+    if (location.hostname === 'kimi.moonshot.cn') {
+      setUserConfig({
+        kimiMoonShotRefreshToken: window.localStorage.refresh_token,
+      })
+    }
+    return
+  }
 
   let data
   if (location.pathname === '/api/auth/session') {

diff --git a/src/manifest.json b/src/manifest.json
@@ -15,6 +15,7 @@
     "https://*.poe.com/*",
     "https://*.google.com/*",
     "https://claude.ai/*",
+    "https://*.moonshot.cn/*",
     "<all_urls>"
   ],
   "permissions": [

diff --git a/src/manifest.v2.json b/src/manifest.v2.json
@@ -22,6 +22,7 @@
     "https://*.poe.com/",
     "https://*.google.com/",
     "https://claude.ai/",
+    "https://*.moonshot.cn/*",
     "<all_urls>"
   ],
   "background": {