feat: add support for new GPT-4o model

apify-projects · May 20, 2024 · 392828f · 392828f
1 parent 1665f52
commit 392828f
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 4 deletions.
diff --git a/actors/extended-gpt-scraper/.actor/input_schema.json b/actors/extended-gpt-scraper/.actor/input_schema.json
@@ -32,14 +32,23 @@
             "editor": "select",
             "default": "gpt-3.5-turbo",
             "prefill": "gpt-3.5-turbo",
-            "enum": ["gpt-3.5-turbo", "gpt-3.5-turbo-16k", "gpt-4", "gpt-4-32k", "text-davinci-003", "gpt-4-turbo"],
+            "enum": [
+                "gpt-3.5-turbo",
+                "gpt-3.5-turbo-16k",
+                "gpt-4",
+                "gpt-4-32k",
+                "text-davinci-003",
+                "gpt-4-turbo",
+                "gpt-4o"
+            ],
             "enumTitles": [
                 "GPT-3.5 Turbo",
                 "GPT-3.5 Turbo 16k",
                 "GPT-4",
                 "GPT-4 32k",
                 "GTP-3 (davinci)",
-                "GPT-4 Turbo (Preview)"
+                "GPT-4 Turbo (Preview)",
+                "GPT-4o"
             ]
         },
         "includeUrlGlobs": {

diff --git a/code/src/models/models.ts b/code/src/models/models.ts
@@ -75,4 +75,13 @@ const OPEN_AI_MODELS: { [modelKey: string]: ModelConfig } = {
             output: 0.03,
         },
     },
+    'gpt-4o': {
+        modelName: 'gpt-4o',
+        maxTokens: 128_000,
+        interface: 'chat',
+        cost: {
+            input: 0.005,
+            output: 0.015,
+        },
+    },
 } as const;
diff --git a/code/src/types/model.ts b/code/src/types/model.ts
@@ -16,8 +16,10 @@ export interface Usage {
 }
 
 export interface Cost {
-    input: number; // USD cost per 1000 tokens
-    output: number; // USD cost per 1000 tokens
+    /** USD cost per 1000 tokens */
+    input: number;
+    /** USD cost per 1000 tokens */
+    output: number;
 }
 
 export interface ProcessInstructionsOptions<ModelSettings extends object> {