Skip to content

Commit 0783dbf

Browse files
committed
pw_air_poller: support llm-mode
Signed-off-by: Jakub Kicinski <kuba@kernel.org>
1 parent c5e3e08 commit 0783dbf

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

pw_air_poller.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ def __init__(self, config_path: str):
127127
self.air_token = self.config.get('air', 'token')
128128
self.air_tree = self.config.get('air', 'tree')
129129
self.air_branch = self.config.get('air', 'branch', fallback=None)
130+
self.air_llm_mode = self.config.get('air', 'llm_mode', fallback=None)
130131

131132
# Patchwork configuration
132133
self.check_name = self.config.get('patchwork', 'check_name', fallback='ai-review')
@@ -241,6 +242,9 @@ def submit_to_air(self, series_id: int) -> Optional[str]:
241242
if self.air_branch:
242243
payload['branch'] = self.air_branch
243244

245+
if self.air_llm_mode:
246+
payload['llm_mode'] = self.air_llm_mode
247+
244248
try:
245249
response = requests.post(api_url, json=payload, timeout=30)
246250
response.raise_for_status()
@@ -473,6 +477,7 @@ def run(self):
473477
log(f"Starting pw_air_poller")
474478
log(f" AIR URL: {self.air_url}")
475479
log(f" Tree: {self.air_tree}")
480+
log(f" LLM mode: {self.air_llm_mode or 'default'}")
476481
log(f" Rate limit: {self.rate_limiter.max_patches} patches / {self.rate_limiter.window_days} days")
477482
log(f" Poll interval: {self.poll_interval}s")
478483
log(f" Review timeout: {self.review_timeout}s")
@@ -507,6 +512,7 @@ def main():
507512
token = your_air_token
508513
tree = netdev/net-next
509514
branch = main # optional
515+
llm_mode = classic # optional, classic or orc
510516
511517
[patchwork]
512518
# Standard NIPA patchwork config

0 commit comments

Comments
 (0)