fix

Author: jonit-dev

README.md

@@ -61,6 +61,7 @@ jobs:
           # Optional inputs with defaults
           model_id: 'anthropic/claude-2' # Default model
           max_tokens: '2048' # Default max tokens
+          review_label: 'ai-review' # Optional: Only review PRs with this label
 
           # Optional custom prompt
           custom_prompt: |
@@ -137,6 +138,7 @@ _Analyzed using anthropic/claude-2_
 | `model_id`        | Model ID to use             | No       | `anthropic/claude-2`       | See available models below               |
 | `custom_prompt`   | Custom prompt for analysis  | No       | Default code review prompt | Can be multiline YAML                    |
 | `max_tokens`      | Maximum tokens in response  | No       | `2048`                     | Adjust based on review complexity        |
+| `review_label`    | Label to trigger review     | No       | -                          | If not set, reviews all PRs              |
 
 ### Minimal Configuration
 
@@ -194,6 +196,16 @@ jobs:
       Rate the overall quality from 1-5 stars and provide final comments.
 ```
 
+### Label-Based Review Example
+
+```yaml
+- uses: jonit-dev/openrouter-github-action@main
+  with:
+    github_token: ${{ secrets.GITHUB_TOKEN }}
+    open_router_key: ${{ secrets.OPEN_ROUTER_KEY }}
+    review_label: 'ai-review' # Only reviews PRs with this label
+```
+
 ## Available Models
 
 Recommended models:

action.yml

@@ -12,6 +12,9 @@ inputs:
     description: 'The model ID to use for analysis (e.g., anthropic/claude-2, openai/gpt-4)'
     required: true
     default: 'anthropic/claude-2'
+  review_label:
+    description: 'Label that triggers the review. If not set, reviews all PRs.'
+    required: false
   custom_prompt:
     description: 'Custom prompt for the AI analysis. If not provided, a default prompt focusing on code review aspects will be used.'
     required: false

dist/index.js

@@ -41642,6 +41642,24 @@ async function getPRDiff(octokit, context) {
   }
 }
 
+async function shouldReviewPR(octokit, context, requiredLabel) {
+  if (!requiredLabel) {
+    return true;
+  }
+
+  try {
+    const { data: labels } = await octokit.rest.issues.listLabelsOnIssue({
+      owner: context.repo.owner,
+      repo: context.repo.repo,
+      issue_number: context.payload.pull_request.number,
+    });
+
+    return labels.some((label) => label.name === requiredLabel);
+  } catch (error) {
+    throw new Error(`Failed to fetch PR labels: ${error.message}`);
+  }
+}
+
 async function analyzeDiff(diff, modelId, openRouterKey, customPrompt) {
   const defaultPrompt = `You are a highly skilled staff software engineer reviewing a pull request. 
 
@@ -41740,11 +41758,23 @@ async function run() {
     const openRouterKey = core.getInput('open_router_key', { required: true });
     const modelId = core.getInput('model_id', { required: true });
     const customPrompt = core.getInput('custom_prompt');
+    const reviewLabel = core.getInput('review_label');
 
     // Get GitHub token and create octokit client
     const token = core.getInput('github_token', { required: true });
     const octokit = github.getOctokit(token);
 
+    // Check if we should review this PR based on label
+    const shouldReview = await shouldReviewPR(
+      octokit,
+      github.context,
+      reviewLabel
+    );
+    if (!shouldReview) {
+      core.info('Skipping review - required label not found on PR');
+      return;
+    }
+
     // Get PR diff
     const diff = await getPRDiff(octokit, github.context);
 

src/index.js

@@ -19,6 +19,24 @@ async function getPRDiff(octokit, context) {
   }
 }
 
+async function shouldReviewPR(octokit, context, requiredLabel) {
+  if (!requiredLabel) {
+    return true;
+  }
+
+  try {
+    const { data: labels } = await octokit.rest.issues.listLabelsOnIssue({
+      owner: context.repo.owner,
+      repo: context.repo.repo,
+      issue_number: context.payload.pull_request.number,
+    });
+
+    return labels.some((label) => label.name === requiredLabel);
+  } catch (error) {
+    throw new Error(`Failed to fetch PR labels: ${error.message}`);
+  }
+}
+
 async function analyzeDiff(diff, modelId, openRouterKey, customPrompt) {
   const defaultPrompt = `You are a highly skilled staff software engineer reviewing a pull request. 
 
@@ -117,11 +135,23 @@ async function run() {
     const openRouterKey = core.getInput('open_router_key', { required: true });
     const modelId = core.getInput('model_id', { required: true });
     const customPrompt = core.getInput('custom_prompt');
+    const reviewLabel = core.getInput('review_label');
 
     // Get GitHub token and create octokit client
     const token = core.getInput('github_token', { required: true });
     const octokit = github.getOctokit(token);
 
+    // Check if we should review this PR based on label
+    const shouldReview = await shouldReviewPR(
+      octokit,
+      github.context,
+      reviewLabel
+    );
+    if (!shouldReview) {
+      core.info('Skipping review - required label not found on PR');
+      return;
+    }
+
     // Get PR diff
     const diff = await getPRDiff(octokit, github.context);