store model and use to calculate completion costs

2023-07-14 11:06:07 -07:00
parent 0371dacfca
commit a5378b106b
14 changed files with 67 additions and 73 deletions
--- a/src/utils/calculateTokenCost.ts
+++ b/src/utils/calculateTokenCost.ts
@@ -23,7 +23,7 @@ const openAICompletionTokensToDollars: { [key in OpenAIChatModel]: number } = {
 };

 export const calculateTokenCost = (
-  model: SupportedModel | null,
+  model: SupportedModel | string | null,
  numTokens: number,
  isCompletion = false,
 ) => {