From 714caf34fe380a2b9118b7125beabca28e9928c0 Mon Sep 17 00:00:00 2001 From: Rafael dos Santos Silva Date: Thu, 20 Jun 2024 17:51:46 -0300 Subject: [PATCH] FEATURE: Support for Claude 3.5 Sonnet via AWS Bedrock (#680) --- lib/completions/dialects/claude.rb | 4 +--- lib/completions/endpoints/aws_bedrock.rb | 2 ++ 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/completions/dialects/claude.rb b/lib/completions/dialects/claude.rb index 2c6fd131..bf43d9b7 100644 --- a/lib/completions/dialects/claude.rb +++ b/lib/completions/dialects/claude.rb @@ -6,9 +6,7 @@ module DiscourseAi class Claude < Dialect class << self def can_translate?(model_name) - %w[claude-instant-1 claude-2 claude-3-haiku claude-3-sonnet claude-3-opus].include?( - model_name, - ) + model_name.start_with?("claude") || model_name.start_with?("anthropic") end end diff --git a/lib/completions/endpoints/aws_bedrock.rb b/lib/completions/endpoints/aws_bedrock.rb index d0ef6274..0087a981 100644 --- a/lib/completions/endpoints/aws_bedrock.rb +++ b/lib/completions/endpoints/aws_bedrock.rb @@ -78,6 +78,8 @@ module DiscourseAi "anthropic.claude-instant-v1" when "claude-3-opus" "anthropic.claude-3-opus-20240229-v1:0" + else + model end api_url =