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

add Claude3Config

Browse source
This commit is contained in:
koid 2024-03-13 11:24:51 +09:00
parent 1ed2cd064a
commit f94a0fd704
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
pr_agent/algo/ai_handlers/litellm_ai_handler.py
View file

@ -70,6 +70,7 @@ class LiteLLMAIHandler(BaseAiHandler):
) )
if get_settings().get("AWS.BEDROCK_REGION", None): if get_settings().get("AWS.BEDROCK_REGION", None):
litellm.AmazonAnthropicConfig.max_tokens_to_sample = 2000 litellm.AmazonAnthropicConfig.max_tokens_to_sample = 2000
litellm.AmazonAnthropicClaude3Config.max_tokens = 2000
self.aws_bedrock_client = boto3.client( self.aws_bedrock_client = boto3.client(
service_name="bedrock-runtime", service_name="bedrock-runtime",
region_name=get_settings().aws.bedrock_region, region_name=get_settings().aws.bedrock_region,