update sonnet model names since they are in litellm json now

This commit is contained in:
Paul Gauthier 2024-10-22 12:53:26 -07:00
parent a800d219c7
commit 4a3e6ef1e5
3 changed files with 2 additions and 20 deletions

View file

@ -58,7 +58,7 @@ def get_parser(default_config_files, git_root):
const=opus_model, const=opus_model,
help=f"Use {opus_model} model for the main chat", help=f"Use {opus_model} model for the main chat",
) )
sonnet_model = "anthropic/claude-3-5-sonnet-20241022" sonnet_model = "claude-3-5-sonnet-20241022"
group.add_argument( group.add_argument(
"--sonnet", "--sonnet",
action="store_const", action="store_const",

View file

@ -580,7 +580,7 @@ def main(argv=None, input=None, output=None, force_git_root=None, return_coder=F
if not args.model: if not args.model:
args.model = "gpt-4o-2024-08-06" args.model = "gpt-4o-2024-08-06"
if os.environ.get("ANTHROPIC_API_KEY"): if os.environ.get("ANTHROPIC_API_KEY"):
args.model = "anthropic/claude-3-5-sonnet-20241022" args.model = "claude-3-5-sonnet-20241022"
main_model = models.Model( main_model = models.Model(
args.model, args.model,

View file

@ -1,18 +0,0 @@
{
"anthropic/claude-3-5-sonnet-20241022": {
"max_tokens": 8192,
"max_input_tokens": 200000,
"max_output_tokens": 8192,
"input_cost_per_token": 0.000003,
"output_cost_per_token": 0.000015,
"cache_creation_input_token_cost": 0.00000375,
"cache_read_input_token_cost": 0.0000003,
"litellm_provider": "anthropic",
"mode": "chat",
"supports_function_calling": true,
"supports_vision": true,
"tool_use_system_prompt_tokens": 159,
"supports_assistant_prefill": true,
"supports_prompt_caching": true
}
}