opensource/sqlchat
1
0
Fork 0
mirror of https://github.com/sqlchat/sqlchat.git synced 2025-07-25 16:13:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: pass model to calculate usage

Browse Source
This commit is contained in:
Tianzhou Chen
2023-12-17 19:49:10 +08:00
parent 319c12a4c6
commit b5c017461f
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/pages/api/chat.ts
View File

@ -40,6 +40,10 @@ const handler = async (req: NextRequest) => {
const requestHeaders: any = {
Authorization: `Bearer ${sessionToken}`,
};
if (req.headers.get("x-openai-model")) {
// Pass model to usage API to calculate usage
requestHeaders["x-openai-model"] = req.headers.get("x-openai-model");
}
if (useServerKey) {
if (hasFeature("account") && !sessionToken) {

2
src/pages/api/usage.ts
View File

@ -26,7 +26,7 @@ const handler = async (req: NextApiRequest, res: NextApiResponse) => {
usage = await getCurrentMonthUsage(endUser);
} else if (req.method === "POST") {
const model = getModel((req.headers["x-openai-model"] as string) || "");
usage = await addUsage(endUser);
usage = await addUsage(endUser, model.cost_per_call);
}
const quota: Quota = {