OpenAiGptService should map the CLI model param values to the OpenAI …

…API model param values

Prompt:
Modify the OpenAiGptService class.
In the call() method, if model == "gpt3" then set model to "gpt-3.5-turbo"
In the call() method, if model == "gpt4" then set model to "gpt-4"
Add tests for your changes in test/OpenAiGptService.test.js
Use sinon to assert that openai.createChatCompletion receives the correct model value

src/services/OpenAiGptService.js

@@ -5,6 +5,9 @@ import { encode } from "gpt-3-encoder"
 
 export default class OpenAiGptService {
   static async call(prompt, model) {
+    if (model == "gpt3") model = "gpt-3.5-turbo";
+    if (model == "gpt4") model = "gpt-4";
+
     const configuration = new Configuration({
       apiKey: process.env.OPENAI_API_KEY
     })

test/OpenAiGptService.test.js

@@ -75,4 +75,27 @@ describe('OpenAiGptService', () => {
       ])
     }));
   });
+
+  it('should pass the correct model value to openai.createChatCompletion', async () => {
+    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'];
+
+    const configStub = sinon.stub(ConfigService, 'retrieveConfig').resolves({ api: { temperature: 0.5 } });
+    const openaiStub = sinon.stub(OpenAIApi.prototype, 'createChatCompletion').resolves({
+      data: {
+        choices: [
+          { message: { content: expectedResult } }
+        ]
+      }
+    });
+
+    for (let i = 0; i < models.length; i++) {
+      await OpenAiGptService.call(prompt, models[i]);
+      sinon.assert.calledWith(openaiStub.getCall(i), sinon.match({ model: expectedModels[i] }));
+    }
+
+    sinon.assert.callCount(openaiStub, models.length);
+  });
 });