Use Gpt4Service in PluginService

Prompt:
Modify the PluginService class.
In the PluginService.executeMode() method, call Gpt4Service.call(prompt, model) if model equals "gpt4" or "gpt3" - don't use Gpt3Service.
Add tests for your changes in test/pluginService.test.js

Author: ferrislucas

src/services/pluginService.js

@@ -86,14 +86,11 @@ export default class PluginService {
       await this.processPipedInput()
       return "Changes applied"
     }
-    if (model === "gpt3") {
-      return await Gpt3Service.call(prompt)
-    }
-    if (model === "gpt4") {
-      return await Gpt4Service.call(prompt)
+    if (model === "gpt3" || model === "gpt4") {
+      return await Gpt4Service.call(prompt, model)
     }
     console.log(`model ${model} is not supported`)
     exit(1)
   }
 
-}
+}

test/pluginService.test.js

@@ -21,17 +21,18 @@ describe('PluginService', () => {
       if (gpt4ServiceStub) gpt4ServiceStub.restore()
     });
 
-    it('should call Gpt3Service when mode is gpt3', async () => {
-      const gpt3ServiceStub = sinon.stub(Gpt3Service, 'call').resolves('GPT3 result');
+    it('should call Gpt4Service when mode is gpt3', async () => {
+      gpt4ServiceStub.resolves('GPT3 result');
       const result = await PluginService.executeMode('gpt3', 'Test prompt');
       assert.strictEqual(result, 'GPT3 result');
-      gpt3ServiceStub.restore();
+      sinon.assert.calledWith(gpt4ServiceStub, 'Test prompt', 'gpt3');
     });
 
     it('should call Gpt4Service when mode is gpt4', async () => {
       gpt4ServiceStub.resolves('GPT4 result');
       const result = await PluginService.executeMode('gpt4', 'Test prompt');
       assert.strictEqual(result, 'GPT4 result');
+      sinon.assert.calledWith(gpt4ServiceStub, 'Test prompt', 'gpt4');
     });
   });