Remove prefixes for ids of aws inferences profiles (#945)

Author: kforjan

lib/langchain/llm/aws_bedrock.rb

@@ -167,7 +167,7 @@ def chat(params = {}, &block)
 
     def parse_model_id(model_id)
       model_id
-        .gsub("us.", "") # Meta append "us." to their model ids
+        .gsub(/^(us|eu|apac|us-gov)\./, "") # Meta append "us." to their model ids, and AWS region prefixes are used by Bedrock cross-region model IDs.
         .split(".")
     end
 

spec/langchain/llm/aws_bedrock_spec.rb

@@ -605,4 +605,36 @@
       end
     end
   end
+
+  describe "#parse_model_id" do
+    it "strips the 'us.' prefix" do
+      input = "us.anthropic.claude-3-5-sonnet-20240620-v1:0"
+      result = subject.send(:parse_model_id, input)
+      expect(result).to eq(["anthropic", "claude-3-5-sonnet-20240620-v1:0"])
+    end
+
+    it "strips the 'eu.' prefix" do
+      input = "eu.anthropic.claude-3-5-sonnet-20240620-v1:0"
+      result = subject.send(:parse_model_id, input)
+      expect(result).to eq(["anthropic", "claude-3-5-sonnet-20240620-v1:0"])
+    end
+
+    it "strips the 'apac.' prefix" do
+      input = "apac.anthropic.claude-3-5-sonnet-20240620-v1:0"
+      result = subject.send(:parse_model_id, input)
+      expect(result).to eq(["anthropic", "claude-3-5-sonnet-20240620-v1:0"])
+    end
+
+    it "strips the 'us-gov.' prefix" do
+      input = "us-gov.anthropic.claude-3-5-sonnet-20240620-v1:0"
+      result = subject.send(:parse_model_id, input)
+      expect(result).to eq(["anthropic", "claude-3-5-sonnet-20240620-v1:0"])
+    end
+
+    it "returns the correct output when no region prefix is present" do
+      input = "anthropic.claude-3-5-sonnet-20240620-v1:0"
+      result = subject.send(:parse_model_id, input)
+      expect(result).to eq(["anthropic", "claude-3-5-sonnet-20240620-v1:0"])
+    end
+  end
 end