test

Author: chao.an

action/index.cjs

@@ -150397,6 +150397,9 @@ const MAX_PATCH_COUNT = process.env.MAX_PATCH_LENGTH
     : Infinity;
 const robot = (app) => {
     const loadChat = async (context) => {
+        if (process.env.USE_GITHUB_MODELS === 'true' && process.env.GITHUB_TOKEN) {
+            return new chat_js_1.Chat(process.env.GITHUB_TOKEN);
+        }
         if (process.env.OPENAI_API_KEY) {
             return new chat_js_1.Chat(process.env.OPENAI_API_KEY);
         }
@@ -150574,7 +150577,7 @@ class Chat {
         if (this.isAzure) {
             // Azure OpenAI configuration
             this.openai = new openai_1.AzureOpenAI({
-                apiKey: this.isGithubModels ? process.env.GITHUB_TOKEN : apikey,
+                apiKey: apikey,
                 endpoint: process.env.OPENAI_API_ENDPOINT || '',
                 apiVersion: process.env.AZURE_API_VERSION || '',
                 deployment: process.env.AZURE_DEPLOYMENT || '',

src/bot.ts

@@ -11,6 +11,10 @@ const MAX_PATCH_COUNT = process.env.MAX_PATCH_LENGTH
 
 export const robot = (app: Probot) => {
   const loadChat = async (context: Context) => {
+    if (process.env.USE_GITHUB_MODELS === 'true' && process.env.GITHUB_TOKEN) {
+      return new Chat(process.env.GITHUB_TOKEN);
+    }
+
     if (process.env.OPENAI_API_KEY) {
       return new Chat(process.env.OPENAI_API_KEY);
     }

src/chat.ts

@@ -15,7 +15,7 @@ export class Chat {
     if (this.isAzure) {
       // Azure OpenAI configuration
       this.openai = new AzureOpenAI({
-        apiKey: this.isGithubModels ? process.env.GITHUB_TOKEN : apikey,
+        apiKey: apikey,
         endpoint: process.env.OPENAI_API_ENDPOINT || '',
         apiVersion: process.env.AZURE_API_VERSION || '',
         deployment: process.env.AZURE_DEPLOYMENT || '',