feat: test

Author: willy-dunamu

packages/code-review-gpt/src/args.ts

@@ -49,7 +49,7 @@ export const getYargs = async (): Promise<ReviewArgs> => {
     .option("model", {
       description: "The model to use for generating the review.",
       type: "string",
-      default: "gpt-4",
+      default: "gpt-4o",
     })
     .option("reviewType", {
       description:
@@ -79,7 +79,7 @@ export const getYargs = async (): Promise<ReviewArgs> => {
       description: "Provider to use for AI",
       choices: ["openai", "bedrock", "azure"],
       type: "string",
-      default: "openai",
+      default: "azure",
     })
     .command("review", "Review the pull request.")
     .command("configure", "Configure the script.")

packages/code-review-gpt/src/common/model/AIModel.ts

@@ -4,7 +4,6 @@ import { retryAsync } from "ts-retry";
 import { IFeedback } from "../types";
 import { logger } from "../utils/logger";
 import { parseAttributes } from "../utils/parseAttributes";
-import { AzureGptChat } from './AzureGptChat';
 
 interface IAIModel {
   modelName: string;
@@ -18,7 +17,7 @@ interface IAIModel {
 const defaultRetryCount = 3;
 
 class AIModel {
-  private model: OpenAIChat | AzureGptChat;
+  private model: OpenAIChat;
   private retryCount: number;
 
   constructor(options: IAIModel) {
@@ -32,11 +31,17 @@ class AIModel {
         });
         break;
       case "azure":
-        this.model = new AzureGptChat({
-          apiKey: options.apiKey,
+        this.model = new OpenAIChat({
           modelName: options.modelName,
-          temperature: options.temperature
-        })
+          temperature: options.temperature,
+          configuration: { organization: options.organization },
+          azureOpenAIApiVersion: '2024-04-01-preview',
+          azureOpenAIApiKey: options.apiKey,
+          // azureOpenAIApiInstanceName
+          azureOpenAIApiDeploymentName: 'D-OAI-model-deploy',
+          azureOpenAIBasePath: 'https://d-oai-dev.openai.azure.com',
+        });
+        break;
       case "bedrock":
         throw new Error("Bedrock provider not implemented");
       default: