| 
					
				 | 
			
			
				@@ -5,9 +5,12 @@ import requests 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				 from core.tools.entities.tool_entities import ToolInvokeMessage 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				 from core.tools.tool.builtin_tool import BuiltinTool 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-SDURL = { 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-    "sd_3": "https://api.siliconflow.cn/v1/stabilityai/stable-diffusion-3-medium/text-to-image", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-    "sd_xl": "https://api.siliconflow.cn/v1/stabilityai/stable-diffusion-xl-base-1.0/text-to-image", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+SILICONFLOW_API_URL = "https://api.siliconflow.cn/v1/image/generations" 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+ 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+SD_MODELS = { 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    "sd_3": "stabilityai/stable-diffusion-3-medium", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    "sd_xl": "stabilityai/stable-diffusion-xl-base-1.0", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+    "sd_3.5_large": "stabilityai/stable-diffusion-3-5-large", 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				 } 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
	
		
			
				| 
					
				 | 
			
			
				@@ -22,9 +25,10 @@ class StableDiffusionTool(BuiltinTool): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         } 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         model = tool_parameters.get("model", "sd_3") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-        url = SDURL.get(model) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        sd_model = SD_MODELS.get(model) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         payload = { 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+            "model": sd_model, 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				             "prompt": tool_parameters.get("prompt"), 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				             "negative_prompt": tool_parameters.get("negative_prompt", ""), 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				             "image_size": tool_parameters.get("image_size", "1024x1024"), 
			 | 
		
	
	
		
			
				| 
					
				 | 
			
			
				@@ -34,7 +38,7 @@ class StableDiffusionTool(BuiltinTool): 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				             "num_inference_steps": tool_parameters.get("num_inference_steps", 20), 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         } 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 | 
		
	
		
			
				 | 
				 | 
			
			
				-        response = requests.post(url, json=payload, headers=headers) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				+        response = requests.post(SILICONFLOW_API_URL, json=payload, headers=headers) 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				         if response.status_code != 200: 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				             return self.create_text_message(f"Got Error Response:{response.text}") 
			 | 
		
	
		
			
				 | 
				 | 
			
			
				  
			 |