|  | @@ -17,10 +17,9 @@ class BedrockProvider(ModelProvider):
 | 
	
		
			
				|  |  |          """
 | 
	
		
			
				|  |  |          try:
 | 
	
		
			
				|  |  |              model_instance = self.get_model_instance(ModelType.LLM)
 | 
	
		
			
				|  |  | -
 | 
	
		
			
				|  |  | -            # Use `gemini-pro` model for validate,
 | 
	
		
			
				|  |  | +            bedrock_validate_model_name = credentials.get('model_for_validation', 'amazon.titan-text-lite-v1')
 | 
	
		
			
				|  |  |              model_instance.validate_credentials(
 | 
	
		
			
				|  |  | -                model='amazon.titan-text-lite-v1',
 | 
	
		
			
				|  |  | +                model=bedrock_validate_model_name,
 | 
	
		
			
				|  |  |                  credentials=credentials
 | 
	
		
			
				|  |  |              )
 | 
	
		
			
				|  |  |          except CredentialsValidateFailedError as ex:
 |