diff --git a/src/apis/trans.js b/src/apis/trans.js index b332e4bd..f06952fc 100644 --- a/src/apis/trans.js +++ b/src/apis/trans.js @@ -28,6 +28,7 @@ import { INPUT_PLACE_TO, INPUT_PLACE_TEXT, INPUT_PLACE_KEY, + INPUT_PLACE_MODEL, } from "../config"; import { msAuth } from "../libs/auth"; import { genDeeplFree } from "./deepl"; @@ -219,6 +220,9 @@ const genOpenAI = ({ text, from, to, url, key, prompt, model }) => { }; const genGemini = ({ text, from, to, url, key, prompt, model }) => { + url = url + .replaceAll(INPUT_PLACE_MODEL, model) + .replaceAll(INPUT_PLACE_KEY, key); prompt = prompt .replaceAll(INPUT_PLACE_FROM, from) .replaceAll(INPUT_PLACE_TO, to) @@ -237,7 +241,6 @@ const genGemini = ({ text, from, to, url, key, prompt, model }) => { ], }; - const input = `${url}/${model}:generateContent?key=${key}`; const init = { headers: { "Content-type": "application/json", @@ -246,7 +249,7 @@ const genGemini = ({ text, from, to, url, key, prompt, model }) => { body: JSON.stringify(data), }; - return [input, init]; + return [url, init]; }; const genOllama = ({ text, from, to, url, key, prompt, model }) => { diff --git a/src/config/index.js b/src/config/index.js index 02d73889..1c0dcf33 100644 --- a/src/config/index.js +++ b/src/config/index.js @@ -384,6 +384,7 @@ export const INPUT_PLACE_FROM = "{{from}}"; // 占位符 export const INPUT_PLACE_TO = "{{to}}"; // 占位符 export const INPUT_PLACE_TEXT = "{{text}}"; // 占位符 export const INPUT_PLACE_KEY = "{{key}}"; // 占位符 +export const INPUT_PLACE_MODEL = "{{model}}"; // 占位符 export const DEFAULT_COLOR = "#209CEE"; // 默认高亮背景色/线条颜色 @@ -554,7 +555,7 @@ export const DEFAULT_TRANS_APIS = { [OPT_TRANS_OPENAI_2]: defaultOpenaiApi, [OPT_TRANS_OPENAI_3]: defaultOpenaiApi, [OPT_TRANS_GEMINI]: { - url: "https://generativelanguage.googleapis.com/v1/models", + url: `https://generativelanguage.googleapis.com/v1/models/${INPUT_PLACE_MODEL}:generateContent?key=${INPUT_PLACE_KEY}`, key: "", model: "gemini-pro", prompt: `Translate the following text from ${INPUT_PLACE_FROM} to ${INPUT_PLACE_TO}:\n\n${INPUT_PLACE_TEXT}`, @@ -562,7 +563,7 @@ export const DEFAULT_TRANS_APIS = { fetchInterval: 500, }, [OPT_TRANS_CLOUDFLAREAI]: { - url: "https://api.cloudflare.com/client/v4/accounts/{ACCOUNT_ID}/ai/run/@cf/meta/m2m100-1.2b", + url: "https://api.cloudflare.com/client/v4/accounts/{{ACCOUNT_ID}}/ai/run/@cf/meta/m2m100-1.2b", key: "", fetchLimit: 1, fetchInterval: 500,