ti.sil/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-12-12 10:55:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

hard code value

Browse source
This commit is contained in:
tmokmss 2023-11-28 20:59:21 +09:00
parent 97d6fb999a
commit 917f4b6a01
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pr_agent/algo/ai_handler.py
View file

@ -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,