opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-08-02 22:58:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: add gpt4 turbo and 4o

Browse Source
This commit is contained in:
tianzhou
2024-05-17 12:20:38 +08:00
parent d32212776c
commit 2293727e31
2 changed files with 41 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
src/components/OpenAIApiConfigView.tsx
View File

@ -3,7 +3,7 @@ import { useTranslation } from "react-i18next";
import { useDebounce } from "react-use";
import { useSettingStore } from "@/store";
import { OpenAIApiConfig } from "@/types";
import { allowSelfOpenAIKey } from "@/utils";
import { allowSelfOpenAIKey, hasFeature } from "@/utils";
import Radio from "./kit/Radio";
import TextField from "./kit/TextField";
import Tooltip from "./kit/Tooltip";
@ -17,14 +17,29 @@ const OpenAIApiConfigView = () => {
const models = [
{
id: "gpt-3.5-turbo",
title: `GPT-3.5 (${t("setting.openai-api-configuration.quota-per-ask", { count: 1 })})`,
title: `GPT-3.5 Turbo`,
cost: 1,
disabled: false,
tooltip: "",
},
{
id: "gpt-4-turbo",
title: `GPT-4 Turbo`,
cost: 20,
disabled: false,
tooltip: "",
},
// Disable GPT-4 if user doesn't provide key (because SQL Chat own key hasn't been whitelisted yet).
{
id: "gpt-4",
title: `GPT-4 (${t("setting.openai-api-configuration.quota-per-ask", { count: 10 })})`,
title: `GPT-4`,
cost: 60,
disabled: false,
tooltip: "",
},
{
id: "gpt-4o",
title: `GPT-4o`,
cost: 10,
disabled: false,
tooltip: "",
},
@ -66,7 +81,7 @@ const OpenAIApiConfigView = () => {
onChange={(value) => handleSetOpenAIApiConfig({ model: value })}
/>
<label htmlFor={model.id} className="ml-3 block text-sm font-medium leading-6 text-gray-900">
{model.title}
{model.title} {hasFeature("quota") ? `(${t("setting.openai-api-configuration.quota-per-ask", { count: model.cost })})` : ""}
</label>
</div>
);

24
src/utils/model.ts
View File

@ -1,4 +1,4 @@
const gpt35 = {
const gpt35turbo = {
name: "gpt-3.5-turbo",
temperature: 0,
frequency_penalty: 0.0,
@ -13,10 +13,28 @@ const gpt4 = {
frequency_penalty: 0.0,
presence_penalty: 0.0,
max_token: 8000,
cost_per_call: 60,
};
const gpt4turbo = {
name: "gpt-4-turbo",
temperature: 0,
frequency_penalty: 0.0,
presence_penalty: 0.0,
max_token: 4000,
cost_per_call: 20,
};
const gpt4ho = {
name: "gpt-4o",
temperature: 0,
frequency_penalty: 0.0,
presence_penalty: 0.0,
max_token: 4000,
cost_per_call: 10,
};
export const models = [gpt35, gpt4];
export const models = [gpt35turbo, gpt4, gpt4turbo, gpt4ho];
export const getModel = (name: string) => {
for (const model of models) {
@ -24,5 +42,5 @@ export const getModel = (name: string) => {
return model;
}
}
return gpt35;
return gpt35turbo;
};