Added --opus

This commit is contained in:
Paul Gauthier 2024-04-18 17:01:37 -07:00
parent 3e153cc7fa
commit 93bd187bf3
4 changed files with 15 additions and 19 deletions

View file

@ -170,6 +170,14 @@ def main(argv=None, input=None, output=None, force_git_root=None):
default=default_model,
help=f"Specify the model to use for the main chat (default: {default_model})",
)
opus_model = "claude-3-opus-20240229"
core_group.add_argument(
"--opus",
action="store_const",
dest="model",
const=opus_model,
help=f"Use {opus_model} model for the main chat",
)
default_4_model = "gpt-4-0613"
core_group.add_argument(
"--4",
@ -238,13 +246,6 @@ def main(argv=None, input=None, output=None, force_git_root=None):
env_var="OPENAI_ORGANIZATION_ID",
help="Specify the OpenAI organization ID",
)
model_group.add_argument(
"--openrouter",
dest="openai_api_base",
action="store_const",
const="https://openrouter.ai/api/v1",
help="Specify the api base url as https://openrouter.ai/api/v1",
)
model_group.add_argument(
"--edit-format",
metavar="EDIT_FORMAT",

View file

@ -101,7 +101,7 @@ MODEL_SETTINGS = [
# Claude
ModelSettings(
"claude-3-opus-20240229",
"udiff",
"diff",
weak_model_name="claude-3-haiku-20240307",
use_repo_map=True,
send_undo_reply=True,