endpoints/.github/workflows/auto-review.yml at f2ec2f8a0ebc8a4e2d8395b4f0981dd61456ebab · mlcommons/endpoints · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
name: Auto Request Review

on:
  pull_request:
    types: [opened, ready_for_review]

jobs:
  auto-review:
    runs-on: ubuntu-latest
    steps:
      - name: Auto request review
        uses: actions/github-script@v7
        with:
          script: |
            const { owner, repo, number } = context.issue;

            // Default reviewers for all components
            const defaultReviewers = ['nvzhihanj','arekay-nv'];

            // Component-specific reviewers (you can expand this later)
            const componentReviewers = {
              'src/inference_endpoint/load_generator/': ['nvzhihanj','arekay-nv'],
              'src/inference_endpoint/dataset_manager/': ['nvzhihanj','arekay-nv'],
              'src/inference_endpoint/endpoint_client/': ['nvzhihanj','arekay-nv'],
              'src/inference_endpoint/metrics/': ['nvzhihanj','arekay-nv'],
              'src/inference_endpoint/config/': ['nvzhihanj','arekay-nv'],
              'src/inference_endpoint/plugins/': ['nvzhihanj','arekay-nv'],
            };

            // Get changed files
            const { data: files } = await github.rest.pulls.listFiles({
              owner,
              repo,
              pull_number: number,
            });

            // Get PR author
            const { data: pullRequest } = await github.rest.pulls.get({
              owner,
              repo,
              pull_number: number,
            });
            const prAuthor = pullRequest.user.login;

            // Determine which reviewers to request based on changed files
            let reviewersToRequest = new Set(defaultReviewers);

            for (const file of files) {
              for (const [component, reviewers] of Object.entries(componentReviewers)) {
                if (file.filename.startsWith(component)) {
                  reviewers.forEach(reviewer => reviewersToRequest.add(reviewer));
                }
              }
            }

            // Exclude PR author from reviewers
            reviewersToRequest.delete(prAuthor);

            // Request reviews
            if (reviewersToRequest.size > 0) {
              await github.rest.pulls.requestReviewers({
                owner,
                repo,
                pull_number: number,
                reviewers: Array.from(reviewersToRequest),
              });

              console.log(`Requested review from: ${Array.from(reviewersToRequest).join(', ')}`);
            }