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 argparse
|
||||||
import logging
|
import logging
|
||||||
import sys
|
|
||||||
import time
|
import time
|
||||||
|
|
||||||
from . import (
|
from . import (
|
||||||
|
@ -53,14 +52,14 @@ def parse_args():
|
||||||
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(
|
parser.add_argument(
|
||||||
'--code-model',
|
'--aider-model',
|
||||||
default=CODE_MODEL,
|
default=CODE_MODEL,
|
||||||
help='Aider code model (default: %(default)s)',
|
help='Aider model to use (default: %(default)s)',
|
||||||
)
|
)
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
'--evaluator-model',
|
'--evaluator-model',
|
||||||
default=EVALUATOR_MODEL,
|
default=EVALUATOR_MODEL,
|
||||||
help='Ollama evaluator model (default: %(default)s)',
|
help='Evaluator model to use (default: %(default)s)',
|
||||||
)
|
)
|
||||||
return parser.parse_args()
|
return parser.parse_args()
|
||||||
|
|
||||||
|
@ -68,9 +67,11 @@ def parse_args():
|
||||||
def main():
|
def main():
|
||||||
logging.basicConfig(level='INFO')
|
logging.basicConfig(level='INFO')
|
||||||
args = parse_args()
|
args = parse_args()
|
||||||
# Override models based on CLI args
|
# Override default models based on CLI flags
|
||||||
pkg = sys.modules[__package__]
|
import importlib
|
||||||
pkg.CODE_MODEL = args.code_model
|
|
||||||
|
pkg = importlib.import_module(__package__)
|
||||||
|
pkg.CODE_MODEL = args.aider_model
|
||||||
pkg.EVALUATOR_MODEL = args.evaluator_model
|
pkg.EVALUATOR_MODEL = args.evaluator_model
|
||||||
|
|
||||||
seen_issues_db = SeenIssuesDB()
|
seen_issues_db = SeenIssuesDB()
|
||||||
|
|
Loading…
Reference in New Issue
Block a user