feat: use o1-mini

korosuke613 · Jan 15, 2025 · 47b8897 · 47b8897
1 parent 7a4f268
commit 47b8897
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/tools/libs/AiReviewer.ts b/tools/libs/AiReviewer.ts
@@ -49,11 +49,19 @@ export class AiReviewer {
     "gpt-4o": {
       input: 2.5 / (1 * 1000000),
       output: 10 / (1 * 1000000),
-    }
+    },
+    "o1-mini": {
+      input: 3 / (1 * 1000000),
+      output: 12 / (1 * 1000000),
+    },
+    "o1": {
+      input: 15 / (1 * 1000000),
+      output: 60 / (1 * 1000000),
+    },
   };
   static readonly defaultOptions: AiReviewerOptions = {
     max_tokens: 1024,
-    model: "gpt-4o",
+    model: "o1",
     logging: true,
     reviewNum: 5,
   };
@@ -162,7 +170,7 @@ export class AiReviewer {
         },
       ],
       temperature: 1,
-      max_tokens: this.options.max_tokens,
+      max_completion_tokens: this.options.max_tokens,
       top_p: 1,
       frequency_penalty: 0,
       presence_penalty: 0,