add test

integrations/amazon_bedrock/tests/test_generator.py

@@ -319,28 +319,37 @@ def test_get_model_adapter(model: str, expected_model_adapter: Optional[Type[Bed
 )
 def test_get_model_adapter_with_model_family(model_family: str, expected_model_adapter: Optional[Type[BedrockModelAdapter]]):
     """
-    Test that the correct model adapter is returned for a given model
+    Test that the correct model adapter is returned for a given model model_family
     """
     model_adapter = AmazonBedrockGenerator.get_model_adapter(model="arn:123435423", model_family=model_family)
     assert model_adapter == expected_model_adapter
 
 
 def test_get_model_adapter_with_invalid_model_family():
     """
-    Test that the correct model adapter is returned for a given model
+    Test that an error is raised when an invalid model_family is provided
     """
     with pytest.raises(AmazonBedrockConfigurationError):
         AmazonBedrockGenerator.get_model_adapter(model="arn:123435423", model_family="invalid")
 
 
 def test_get_model_adapter_auto_detect_family_fails():
     """
-    Test that the correct model adapter is returned for a given model
+    Test that an error is raised when auto-detection of model_family fails
     """
     with pytest.raises(AmazonBedrockConfigurationError):
         AmazonBedrockGenerator.get_model_adapter(model="arn:123435423", model_family=None)
 
 
+def test_get_model_adapter_model_family_over_auto_detection():
+    """
+    Test that the model_family is used over auto-detection
+    """
+    model_adapter = AmazonBedrockGenerator.get_model_adapter(
+        model="cohere.command-text-v14", model_family="anthropic.claude"
+    )
+    assert model_adapter == AnthropicClaudeAdapter
+
 
 class TestAnthropicClaudeAdapter:
     def test_default_init(self) -> None: