diff --git a/src/components/TokenCount/TokenCount.tsx b/src/components/TokenCount/TokenCount.tsx index af47c1b..2c1921a 100644 --- a/src/components/TokenCount/TokenCount.tsx +++ b/src/components/TokenCount/TokenCount.tsx @@ -1,8 +1,9 @@ -import React, { useEffect, useState } from 'react'; +import React, { useEffect, useMemo, useState } from 'react'; import useStore from '@store/store'; import { shallow } from 'zustand/shallow'; import countTokens from '@utils/messageUtils'; +import { modelCost } from '@constants/chat'; const TokenCount = React.memo(() => { const [tokenCount, setTokenCount] = useState(0); @@ -19,6 +20,11 @@ const TokenCount = React.memo(() => { : 'gpt-3.5-turbo' ); + const cost = useMemo(() => { + const price = modelCost[model].price * (tokenCount / modelCost[model].unit); + return price.toPrecision(3); + }, [model, tokenCount]); + useEffect(() => { if (!generating) setTokenCount(countTokens(messages, model)); }, [messages, generating]); @@ -26,7 +32,7 @@ const TokenCount = React.memo(() => { return (
- Tokens: {tokenCount} + Tokens: {tokenCount} (${cost})
); diff --git a/src/constants/chat.ts b/src/constants/chat.ts index f310552..f2a3d0a 100644 --- a/src/constants/chat.ts +++ b/src/constants/chat.ts @@ -34,6 +34,15 @@ export const modelMaxToken = { 'gpt-4-32k-0314': 32768, }; +export const modelCost = { + 'gpt-3.5-turbo': { price: 0.002, unit: 1000 }, + 'gpt-3.5-turbo-0301': { price: 0.002, unit: 1000 }, + 'gpt-4': { price: 0.03, unit: 1000 }, + 'gpt-4-0314': { price: 0.03, unit: 1000 }, + 'gpt-4-32k': { price: 0.06, unit: 1000 }, + 'gpt-4-32k-0314': { price: 0.06, unit: 1000 }, +}; + export const defaultUserMaxToken = 4000; export const _defaultChatConfig: ConfigInterface = {