forked from NVIDIA/TensorRT-LLM
-
Notifications
You must be signed in to change notification settings - Fork 0
82 lines (72 loc) · 3.05 KB
/
Copy pathllm-api-compatibility.yml
File metadata and controls
82 lines (72 loc) · 3.05 KB
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
70
71
72
73
74
75
76
77
78
79
80
81
82
# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: Apache-2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
name: LLM API Compatibility
on:
pull_request:
types: [opened, edited, synchronize, reopened, labeled, unlabeled]
jobs:
check-llm-api-compatibility:
name: Check LLM API Compatibility Label
runs-on: ubuntu-latest
permissions:
pull-requests: read
steps:
- name: Validate LLM API compatibility label
uses: actions/github-script@v8
with:
script: |
const referencePrefixes = [
'tests/unittest/api_stability/references/',
'tests/unittest/api_stability/references_committed/',
];
// Keep these names in sync with the PR template and API-change guide.
const compatibleApiLabel = 'api-compatible';
const breakingApiLabel = 'api-breaking';
const files = await github.paginate(github.rest.pulls.listFiles, {
owner: context.repo.owner,
repo: context.repo.repo,
pull_number: context.payload.pull_request.number,
per_page: 100,
});
const changedPaths = files
.flatMap((file) => [file.filename, file.previous_filename].filter(Boolean));
const apiReferenceChanged = changedPaths.some((path) =>
referencePrefixes.some((prefix) => path.startsWith(prefix))
);
if (!apiReferenceChanged) {
core.info('No API stability reference changes detected.');
return;
}
const labelNames = new Set(
(context.payload.pull_request.labels || []).map((label) => label.name)
);
const apiChangeLabels = [compatibleApiLabel, breakingApiLabel].filter((label) =>
labelNames.has(label)
);
if (apiChangeLabels.length !== 1) {
core.setFailed(
'API stability reference files changed. Apply exactly one ' +
`of these labels: '${compatibleApiLabel}' or '${breakingApiLabel}'.`
);
return;
}
if (
apiChangeLabels[0] === breakingApiLabel &&
!context.payload.pull_request.title.includes('BREAKING')
) {
core.setFailed("Breaking LLM API changes must include 'BREAKING' in the PR title.");
return;
}
core.info(`LLM API change label accepted: ${apiChangeLabels[0]}`);