| 
					
				 | 
			
			
				@@ -1,6 +1,7 @@ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				 import logging 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-from core.model_runtime.entities.model_entities import ModelType 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+import requests 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				 from core.model_runtime.errors.validate import CredentialsValidateFailedError 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				 from core.model_runtime.model_providers.__base.model_provider import ModelProvider 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
	
		
			
				| 
					
				 | 
			
			
				@@ -16,8 +17,18 @@ class GiteeAIProvider(ModelProvider): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         :param credentials: provider credentials, credentials form defined in `provider_credential_schema`. 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         """ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         try: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-            model_instance = self.get_model_instance(ModelType.LLM) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-            model_instance.validate_credentials(model="Qwen2-7B-Instruct", credentials=credentials) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            api_key = credentials.get("api_key") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            if not api_key: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                raise CredentialsValidateFailedError("Credentials validation failed: api_key not given") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            # send a get request to validate the credentials 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            headers = {"Authorization": f"Bearer {api_key}"} 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            response = requests.get("https://ai.gitee.com/api/base/account/me", headers=headers, timeout=(10, 300)) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            if response.status_code != 200: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                raise CredentialsValidateFailedError( 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                    f"Credentials validation failed with status code {response.status_code}" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+                ) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         except CredentialsValidateFailedError as ex: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				             raise ex 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         except Exception as ex: 
			 |