From 917f4b6a012f19d40e8b847dab7ad228f633b41e Mon Sep 17 00:00:00 2001 From: tmokmss Date: Tue, 28 Nov 2023 20:59:21 +0900 Subject: [PATCH] hard code value --- pr_agent/algo/ai_handler.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/pr_agent/algo/ai_handler.py b/pr_agent/algo/ai_handler.py index 24273db6..7f071ef3 100644 --- a/pr_agent/algo/ai_handler.py +++ b/pr_agent/algo/ai_handler.py @@ -63,9 +63,7 @@ class AiHandler: "VERTEXAI.VERTEX_LOCATION", None ) if get_settings().get("AWS.BEDROCK_REGION", None): - litellm.AmazonAnthropicConfig.max_tokens_to_sample = int(get_settings().get( - "AWS.CLAUDE_MAX_TOKENS_TO_SAMPLE", '2000' - )) + litellm.AmazonAnthropicConfig.max_tokens_to_sample = 2000 self.aws_bedrock_client = boto3.client( service_name="bedrock-runtime", region_name=get_settings().aws.bedrock_region,