Compare commits

..

3 Commits

View File

@ -47,13 +47,13 @@ def parse_args():
) )
parser.add_argument( parser.add_argument(
'--aider-model', '--aider-model',
help='Model to use for generating code (overrides default)',
default=None, default=None,
help='Aider code model (overrides default CODE_MODEL)',
) )
parser.add_argument( parser.add_argument(
'--evaluator-model', '--evaluator-model',
help='Model to use for evaluating code (overrides default)',
default=None, default=None,
help='Evaluator model for assessment (overrides default EVALUATOR_MODEL)',
) )
return parser.parse_args() return parser.parse_args()
@ -61,14 +61,15 @@ def parse_args():
def main(): def main():
logging.basicConfig(level='INFO') logging.basicConfig(level='INFO')
args = parse_args() args = parse_args()
# Override default models if CLI args provided
# Override default models if provided
import aider_gitea as core
if args.aider_model: if args.aider_model:
core.CODE_MODEL = args.aider_model import aider_gitea
aider_gitea.CODE_MODEL = args.aider_model
if args.evaluator_model: if args.evaluator_model:
core.EVALUATOR_MODEL = args.evaluator_model import aider_gitea
aider_gitea.EVALUATOR_MODEL = args.evaluator_model
seen_issues_db = SeenIssuesDB() seen_issues_db = SeenIssuesDB()
client = GiteaClient(args.gitea_url, secrets.gitea_token()) client = GiteaClient(args.gitea_url, secrets.gitea_token())