Merge pull request #63 from ferrislucas/use_gpt-4o

Use gpt-4o as the default model
ferrislucas · May 18, 2024 · 538a841 · 538a841
2 parents 7afa19c + fe97cf6
commit 538a841
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/CliState.js b/src/CliState.js
@@ -13,7 +13,7 @@ export default class CliState {
     this.program.option('-t, --template <template>', 'Teplate name, template path, or a url for a template file')
     this.program.option('-o, --output-path <outputPath>', 'Path to output file. If no path is specified, output will be printed to stdout.')
     this.program.option('-v, --verbose', 'Verbose output')
-    this.program.option('-m, --model <model>', 'Specify the model to use', 'gpt-4-turbo-preview')
+    this.program.option('-m, --model <model>', 'Specify the model to use', 'gpt-4o')
     this.program.option('-dac, --disable-auto-context', 'Prevents files referenced in the prompt from being automatically included in the context sent to the model.');
 
     this.program.version(version, '--version', 'Display the current version')

diff --git a/src/services/OpenAiGptService.js b/src/services/OpenAiGptService.js
@@ -8,7 +8,7 @@ export default class OpenAiGptService {
 
   static async call(prompt, model, requestJsonOutput = true) {
     if (model == "gpt3") model = "gpt-3.5-turbo";
-    if (model == "gpt4") model = "gpt-4-turbo-preview";
+    if (model == "gpt4") model = "gpt-4o";
 
     const configuration = new Configuration({
       apiKey: process.env.OPENAI_API_KEY

diff --git a/test/OpenAiGptService.test.js b/test/OpenAiGptService.test.js
@@ -104,7 +104,7 @@ describe('OpenAiGptService', () => {
     const prompt = 'What is the capital of France?';
     const expectedResult = 'The capital of France is Paris.';
     const models = ['gpt3', 'gpt4'];
-    const expectedModels = ['gpt-3.5-turbo', 'gpt-4-turbo-preview'];
+    const expectedModels = ['gpt-3.5-turbo', 'gpt-4o'];
 
     const openaiStub = sinon.stub(OpenAIApi.prototype, 'createChatCompletion').resolves({
       data: {