Compare commits

..

3 Commits

View File

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