From 5178d232260f8d0d690d70ed568108bd820572b5 Mon Sep 17 00:00:00 2001 From: semio Date: Sat, 31 Aug 2024 10:51:18 +0800 Subject: [PATCH] fix claude region --- .../custom_configuration/model_compare.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/automation-api/yival_experiments/custom_configuration/model_compare.py b/automation-api/yival_experiments/custom_configuration/model_compare.py index d6b96bc..4a4fb80 100644 --- a/automation-api/yival_experiments/custom_configuration/model_compare.py +++ b/automation-api/yival_experiments/custom_configuration/model_compare.py @@ -109,12 +109,8 @@ def model_compare( # google allows changing content filters. We will disable all litellm_params["safety_settings"] = safety_settings elif model["vendor"] == "Anthropic": - if "opus" in model["model_id"]: - # there is only one location where claude Opus is available. - litellm.vertex_location = "us-east5" - else: - litellm.vertex_location = "us-central1" - + # all Anthropic models are abailable in us-east5 + litellm.vertex_location = "us-east5" try: if model["vendor"] == "Alibaba": # FIXME: alibaba's complete function doesn't support system prompt.