diff --git a/.github/mergify.yml b/.github/mergify.yml index 73ea58aed6..0798654d34 100644 --- a/.github/mergify.yml +++ b/.github/mergify.yml @@ -112,6 +112,32 @@ pull_request_rules: add: - frontend +- name: label-pooling + description: Automatically apply pooling label + conditions: + - -closed + - or: + - files~=^vllm/entrypoints/pooling/ + - files~=^vllm/model_executor/layers/pooler/ + - files=vllm/model_executor/models/transformers/pooling.py + - files=vllm/model_executor/models/adapters.py + - files=vllm/config/pooler.py + - files=vllm/pooling_params.py + - files~=^vllm/v1/pool/ + - files~=^vllm/v1/worker/gpu/pool/ + - files~=^tests/entrypoints/pooling/ + - files~=^tests/models/(language|multimodal)/pooling(_mteb_test)?/ + - files~=^tests/model_executor/layers/test_pooler_.*\.py$ + - files=tests/test_pooling_params.py + - files~=^tests/v1/e2e/general/test_pooling_.*\.py$ + - files~=^docs/models/pooling_models/ + - files~=^examples/pooling/ + - title~=(?i)\bpooling\b + actions: + label: + add: + - pooling + - name: label-rust description: Automatically apply rust label conditions: diff --git a/.github/workflows/issue_autolabel.yml b/.github/workflows/issue_autolabel.yml index fc6e4921ef..c3190f8f18 100644 --- a/.github/workflows/issue_autolabel.yml +++ b/.github/workflows/issue_autolabel.yml @@ -255,6 +255,27 @@ jobs: // phrases go in `substrings`: they are specific enough on their // own, and a boundary match would miss ordinary variations -- // "structured outputs", "tool parsers", "prefix caching". + pooling: { + keywords: [ + { term: "pooling", searchIn: "title" }, + { term: "pooler", searchIn: "title" }, + { term: "rerank", searchIn: "title" }, + { term: "reranker", searchIn: "title" }, + { term: "reranking", searchIn: "title" }, + ], + substrings: [ + { term: "embedding model", searchIn: "title" }, + { term: "embedding API", searchIn: "title" }, + { term: "embeddings API", searchIn: "title" }, + { term: "sentence embedding", searchIn: "title" }, + { term: "sequence classification", searchIn: "title" }, + { term: "token classification", searchIn: "title" }, + { term: "cross-encoder", searchIn: "title" }, + { term: "late interaction", searchIn: "title" }, + { term: "late-interaction", searchIn: "title" }, + { term: "reward model", searchIn: "title" }, + ], + }, "speculative-decoding": { keywords: [ { term: "eagle", searchIn: "title" },