heimoshuiyu/chatgpt-api-web
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases 1 Wiki Activity

prevent undefined on new models

Browse Source
This commit is contained in:
黑墨水鱼
2023-03-31 15:00:57 +08:00
parent 464e417537
commit 11d9b09e36
2 changed files with 7 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/chatbox.tsx
View File

@@ -87,7 +87,8 @@ export default function ChatBOX(props: {
// estimate cost
if (chatStore.responseModelName) {
chatStore.cost +=
token * models[chatStore.responseModelName].price.completion;
token *
(models[chatStore.responseModelName]?.price?.completion ?? 0);
let sum = 0;
for (const msg of chatStore.history
.filter(({ hide }) => !hide)
@@ -95,7 +96,7 @@ export default function ChatBOX(props: {
sum += msg.token;
}
chatStore.cost +=
sum * models[chatStore.responseModelName].price.prompt;
sum * (models[chatStore.responseModelName]?.price?.prompt ?? 0);
}
chatStore.history.push({
role: "assistant",
@@ -119,10 +120,11 @@ export default function ChatBOX(props: {
chatStore.responseModelName = data.model ?? "";
if (data.model) {
chatStore.cost +=
(data.usage.prompt_tokens ?? 0) * models[data.model].price.prompt;
(data.usage.prompt_tokens ?? 0) *
(models[data.model]?.price?.prompt ?? 0);
chatStore.cost +=
(data.usage.completion_tokens ?? 0) *
models[data.model].price.completion;
(models[data.model]?.price?.completion ?? 0);
}
const content = client.processFetchResponse(data);
chatStore.history.push({