Compare commits
3 Commits
df9d207913
...
29ef1cfdf5
Author | SHA1 | Date | |
---|---|---|---|
29ef1cfdf5 | |||
2f471d03a5 | |||
130bd7e4d4 |
|
@ -6,7 +6,6 @@ It assumes that the default branch (default "main") exists and that you have a v
|
|||
|
||||
import argparse
|
||||
import logging
|
||||
import sys
|
||||
import time
|
||||
|
||||
from . import (
|
||||
|
@ -53,14 +52,14 @@ def parse_args():
|
|||
help='Interval in seconds between checks in daemon mode (default: 300)',
|
||||
)
|
||||
parser.add_argument(
|
||||
'--code-model',
|
||||
'--aider-model',
|
||||
default=CODE_MODEL,
|
||||
help='Aider code model (default: %(default)s)',
|
||||
help='Aider model to use (default: %(default)s)',
|
||||
)
|
||||
parser.add_argument(
|
||||
'--evaluator-model',
|
||||
default=EVALUATOR_MODEL,
|
||||
help='Ollama evaluator model (default: %(default)s)',
|
||||
help='Evaluator model to use (default: %(default)s)',
|
||||
)
|
||||
return parser.parse_args()
|
||||
|
||||
|
@ -68,9 +67,11 @@ def parse_args():
|
|||
def main():
|
||||
logging.basicConfig(level='INFO')
|
||||
args = parse_args()
|
||||
# Override models based on CLI args
|
||||
pkg = sys.modules[__package__]
|
||||
pkg.CODE_MODEL = args.code_model
|
||||
# Override default models based on CLI flags
|
||||
import importlib
|
||||
|
||||
pkg = importlib.import_module(__package__)
|
||||
pkg.CODE_MODEL = args.aider_model
|
||||
pkg.EVALUATOR_MODEL = args.evaluator_model
|
||||
|
||||
seen_issues_db = SeenIssuesDB()
|
||||
|
|
Loading…
Reference in New Issue
Block a user