diff --git a/aider_gitea/__main__.py b/aider_gitea/__main__.py index d2e1185..a6f3d49 100644 --- a/aider_gitea/__main__.py +++ b/aider_gitea/__main__.py @@ -45,6 +45,16 @@ def parse_args(): default=300, help='Interval in seconds between checks in daemon mode (default: 300)', ) + parser.add_argument( + '--aider-model', + help='Model to use for generating code (overrides default)', + default=None, + ) + parser.add_argument( + '--evaluator-model', + help='Model to use for evaluating code (overrides default)', + default=None, + ) return parser.parse_args() @@ -52,6 +62,14 @@ def main(): logging.basicConfig(level='INFO') args = parse_args() + # Override default models if provided + import aider_gitea as core + + if args.aider_model: + core.CODE_MODEL = args.aider_model + if args.evaluator_model: + core.EVALUATOR_MODEL = args.evaluator_model + seen_issues_db = SeenIssuesDB() client = GiteaClient(args.gitea_url, secrets.gitea_token()) diff --git a/aider_gitea/gitea_client.py b/aider_gitea/gitea_client.py index 0c2f8e6..7360e05 100644 --- a/aider_gitea/gitea_client.py +++ b/aider_gitea/gitea_client.py @@ -169,7 +169,9 @@ class GiteaClient: # If a pull request for this head/base already exists, return it instead of crashing if response.status_code == 409: logger.warning( - 'Pull request already exists for head %s and base %s', head, base, + 'Pull request already exists for head %s and base %s', + head, + base, ) prs = self.get_pull_requests(owner, repo) for pr in prs: