Update Fork

README.md

@@ -117,7 +117,7 @@ They can be accessed in the workflow file using `${{ secrets.YOUR_KEY_NAME }}`.
 
 ### 4. `ai-model-name` (Optional)
 
-Specify the name of the model you want to use to generate suggestions. Fallbacks to `gpt-4o-2024-08-06` for OpenAI, `claude-3-7-sonnet-latest` for Anthropic, `gemini-2.5-pro-preview-05-06` for Google's Gemini, `pixtral-12b-2409` for Mistral, and `deepseek/deepseek-r1` for OpenRouter if not specified. Here's a list of supported models:
+Specify the name of the model you want to use to generate suggestions. Fallbacks to `gpt-5` for OpenAI, `claude-3-7-sonnet-latest` for Anthropic, `gemini-2.5-pro-preview-06-05` for Google's Gemini, `pixtral-12b-2409` for Mistral, and `google/gemini-2.5-pro-preview-06-05` for OpenRouter if not specified. Here's a list of supported models:
 
 For OpenAI:
 
@@ -141,7 +141,7 @@ For Mistral:
 
 For OpenRouter:
 
-- [deepseek/deepseek-r1](https://openrouter.ai/deepseek/deepseek-r1).
+- Any of the models listed [here in OpenRouter](https://openrouter.ai/models?fmt=table&supported_parameters=response_format) that support `structured outputs`/`response_format`.
 
 For Google:
 

index.js

@@ -174,7 +174,7 @@ function getUserPrompt(rules, rawComments, pullRequestContext) {
 async function useOpenAI({ rawComments, openAI, rules, modelName, pullRequestContext, platform }) {
     const modelDeepseek = /deepseek/i.test(getModelName(modelName, platform));
     const result = !modelDeepseek
-        ? await openAI.beta.chat.completions.parse({
+        ? await openAI.chat.completions.parse({
               model: getModelName(modelName, platform),
               messages: [
                   {
@@ -356,14 +356,6 @@ async function retry(
         } catch (error) {
             lastError = error;
 
-            const shouldNotRetry = nonRetryableErrors.some(errMsg =>
-                error.message.toLowerCase().includes(errMsg.toLowerCase())
-            );
-
-            if (shouldNotRetry || attempt === retries - 1) {
-                throw error;
-            }
-
             if (onRetry) {
                 onRetry({
                     error,
@@ -373,6 +365,14 @@ async function retry(
                 });
             }
 
+            const shouldNotRetry = nonRetryableErrors.some(errMsg =>
+                error.message.toLowerCase().includes(errMsg.toLowerCase())
+            );
+
+            if (shouldNotRetry || attempt === retries - 1) {
+                throw error;
+            }
+
             await new Promise(resolve => setTimeout(resolve, delay));
             delay = Math.min(delay * backoffFactor, maxDelay);
         }
@@ -444,7 +444,10 @@ async function getSuggestions({
                 retries: maxRetries ?? 3,
                 onRetry: ({ error: retryError, attempt, remainingAttempts, delay }) => {
                     error(`Attempt ${attempt} failed: ${retryError.message}.`);
-                    warning(`Retrying in ${delay}ms. Remaining attempts: ${remainingAttempts}.`);
+
+                    if (remainingAttempts !== 0) {
+                        warning(`Retrying in ${delay}ms. Remaining attempts: ${remainingAttempts}.`);
+                    }
                 },
             }
         );
@@ -519,6 +522,7 @@ async function getAllReviewsForPullRequest(octokit) {
         owner: github.context.repo.owner,
         repo: github.context.repo.repo,
         pull_number: github.context.payload.pull_request.number,
+        per_page: 500,
     });
 }
 
@@ -532,6 +536,7 @@ async function getAllCommentsUnderAReview(octokit, review_id) {
         repo: github.context.repo.repo,
         pull_number: github.context.payload.pull_request.number,
         review_id,
+        per_page: 500,
     });
 }
 
@@ -672,7 +677,7 @@ async function run() {
                 platformSDK,
                 rules,
                 modelName,
-                maxRetries: Number(maxRetries),
+                maxRetries: maxRetries === "" ? null : Number(maxRetries),
                 pullRequestContext: {
                     body: pullRequestData.data.body,
                 },