feat: add CLI options to specify aider and evaluator models
This commit is contained in:
parent
cf7f784450
commit
fa6c221465
|
@ -45,6 +45,16 @@ def parse_args():
|
||||||
default=300,
|
default=300,
|
||||||
help='Interval in seconds between checks in daemon mode (default: 300)',
|
help='Interval in seconds between checks in daemon mode (default: 300)',
|
||||||
)
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--aider-model',
|
||||||
|
default='o4-mini',
|
||||||
|
help='Model to use for Aider (default: o4-mini)',
|
||||||
|
)
|
||||||
|
parser.add_argument(
|
||||||
|
'--evaluator-model',
|
||||||
|
default='ollama/gemma3:27b',
|
||||||
|
help='Model to use for evaluator (default: ollama/gemma3:27b)',
|
||||||
|
)
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
|
|
||||||
|
@ -52,6 +62,11 @@ def main():
|
||||||
logging.basicConfig(level='INFO')
|
logging.basicConfig(level='INFO')
|
||||||
args = parse_args()
|
args = parse_args()
|
||||||
|
|
||||||
|
import importlib
|
||||||
|
pkg = importlib.import_module(__package__)
|
||||||
|
pkg.CODE_MODEL = args.aider_model
|
||||||
|
pkg.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())
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user