Merge pull request #47 from kingchan818/feat

feat: add supportModelType, gpt-4o, gpt4o-2024-05-13
Cainier · May 15, 2024 · f9c7a5a · f9c7a5a
2 parents 4e44d44 + c0b908b
commit f9c7a5a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/dist/pricing.js b/dist/pricing.js
@@ -21,13 +21,15 @@ Pricing.generalModelMapping = {
     'gpt-4-32k': 'gpt-4-32k-0613',
     'gpt-4-turbo-preview': 'gpt-4-0125-preview',
     'gpt-4-turbo': 'gpt-4-turbo-2024-04-09',
+    'gpt-4o': 'gpt-4o-2024-05-13',
 };
 /**
  * Incremental Models
  * @public
  * Model: [Input , Output, Train (If support)] ($/1K Tokens)
  */
 Pricing.incrementalModels = {
+    'gpt-4o-2024-05-13': [0.005, 0.015], // 2024-05-13
     'gpt-4-turbo-2024-04-09': [0.01, 0.03], // 2024-04-09
     'gpt-4-0314': [0.03, 0.06], // 2023-03-14
     'gpt-4-32k-0314': [0.06, 0.12], // 2023-03-14

diff --git a/src/pricing.ts b/src/pricing.ts
@@ -18,6 +18,7 @@ export class Pricing {
         'gpt-4-32k'          : 'gpt-4-32k-0613',
         'gpt-4-turbo-preview': 'gpt-4-0125-preview',
         'gpt-4-turbo'        : 'gpt-4-turbo-2024-04-09',
+        'gpt-4o'             : 'gpt-4o-2024-05-13',
     }
 
     /**
@@ -28,6 +29,7 @@ export class Pricing {
     public static readonly incrementalModels: {
         [model: string]: [number, number, number?]
     } = {
+        'gpt-4o-2024-05-13'        : [0.005, 0.015],                // 2024-05-13
         'gpt-4-turbo-2024-04-09'   : [0.01, 0.03],                  // 2024-04-09
         'gpt-4-0314'               : [0.03, 0.06],                  // 2023-03-14
         'gpt-4-32k-0314'           : [0.06, 0.12],                  // 2023-03-14