From 7438190ed100f5ea0911b8ad06371489837b1d3c Mon Sep 17 00:00:00 2001 From: mrT23 Date: Sun, 11 Aug 2024 11:43:57 +0300 Subject: [PATCH] set_claude_model --- pr_agent/git_providers/utils.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/pr_agent/git_providers/utils.py b/pr_agent/git_providers/utils.py index a0d65b66..8a9579cf 100644 --- a/pr_agent/git_providers/utils.py +++ b/pr_agent/git_providers/utils.py @@ -47,3 +47,17 @@ def apply_repo_settings(pr_url): os.remove(repo_settings_file) except Exception as e: get_logger().error(f"Failed to remove temporary settings file {repo_settings_file}", e) + + # enable switching models with a short definition + if get_settings().config.model.lower()=='claude-3-5-sonnet': + set_claude_model() + + +def set_claude_model(): + """ + set the claude-sonnet-3.5 model easily (even by users), just by stating: --config.model='claude-3-5-sonnet' + """ + model_claude = "bedrock/anthropic.claude-3-5-sonnet-20240620-v1:0" + get_settings().set('config.model', model_claude) + get_settings().set('config.model_turbo', model_claude) + get_settings().set('config.fallback_models', [model_claude])