fix: trailing whitespace

Author: yanbasic

src/emd/cfn/sagemaker_async/post_build.py

@@ -41,7 +41,7 @@ def post_build():
         use_spot = True
     else:
         use_spot = False
-        
+
     deploy_ecs_cluster(args.region, vpc_id, subnets, use_spot)
 
 

src/emd/cfn/sagemaker_realtime/post_build.py

@@ -41,7 +41,7 @@ def post_build():
         use_spot = True
     else:
         use_spot = False
-        
+
     deploy_ecs_cluster(args.region, vpc_id, subnets, use_spot)
 
 

src/emd/cfn/shared/ecs_cluster.py

@@ -11,7 +11,7 @@
 
 def run_command(cmd):
     try:
-        result = subprocess.run(cmd, shell=True, check=True, 
+        result = subprocess.run(cmd, shell=True, check=True,
                               stdout=subprocess.PIPE, stderr=subprocess.PIPE,
                               text=True)
         return result.stdout.strip()
@@ -23,15 +23,15 @@ def run_command(cmd):
 def build_router_image(region):
     repo_name = "emd-openai-api-router"
     image_tag = f"{repo_name}:latest"
-    
+
     # Check if repo exists
     try:
         run_command(f"aws ecr describe-repositories --repository-names {repo_name} --region {region}")
         print(f"ECR repository {repo_name} already exists")
     except:
         print(f"Creating ECR repository: {repo_name}")
         run_command(f"aws ecr create-repository --repository-name {repo_name} --region {region}")
-    
+
     # Get ECR login
     account_id = json.loads(run_command(f"aws sts get-caller-identity --region {region}"))['Account']
     domain = f"dkr.ecr.{region}.amazonaws.com.cn" if region.startswith("cn-") else f"dkr.ecr.{region}.amazonaws.com"
@@ -45,16 +45,16 @@ def build_router_image(region):
         f"-t {image_tag} ."
     )
     run_command(build_cmd)
-    
+
     # Tag and push image
     full_image_tag = f"{registry}/{image_tag}"
-    
+
     print(f"Tagging image as: {full_image_tag}")
     run_command(f"docker tag {image_tag} {full_image_tag}")
-    
+
     print(f"Pushing image to ECR: {full_image_tag}")
     run_command(f"docker push {full_image_tag}")
-    
+
     return full_image_tag
 
 def wait_for_stack_completion(client, stack_name):
@@ -207,23 +207,23 @@ def deploy_ecs_cluster_template(region, vpc_id, subnets, api_router_uri, use_spo
 
 def deploy_ecs_cluster(region, vpc_id=None, subnets=None, use_spot=False):
     """Deploy ECS cluster with specified VPC and subnets.
-    
+
     Args:
         region (str): AWS region to deploy to
         vpc_id (str, optional): VPC ID to use for deployment. If None, deploys new VPC.
         subnets (str, optional): Comma-separated list of subnet IDs. If None, deploys new VPC.
     """
-    
+
     # Deploy new VPC if no vpc_id/subnets provided
     if not vpc_id or not subnets:
         vpc_id, subnets = deploy_vpc_template(region)
-    
+
     # Update parameters with networking info
     update_parameters_file("parameters.json", {"VPCID": vpc_id, "Subnets": subnets})
-    
+
     # Build and push Fargate image to ECR as the OpenAI compatible API router
     api_router_uri = build_router_image(region)
-    
+
     # Deploy the ECS cluster
     deploy_ecs_cluster_template(region, vpc_id, subnets, api_router_uri, use_spot)
 

src/emd/cfn/shared/filter_parameters.py

@@ -7,73 +7,73 @@ def read_template_parameters(template_path):
     """Read CloudFormation template and extract parameter names"""
     with open(template_path) as f:
         content = f.read()
-    
+
     # Find Parameters section (match until next section or end of file)
     params_section = re.search(
-        r'^Parameters:\s*$(.*?)(?=^\w|\Z)', 
-        content, 
+        r'^Parameters:\s*$(.*?)(?=^\w|\Z)',
+        content,
         re.MULTILINE | re.DOTALL
     )
     if not params_section:
         print(f"Warning: No Parameters section found in {template_path}")
         return set()
-    
+
     # Extract parameter names (must be valid CloudFormation parameter names)
     param_names = re.findall(
         r'^\s+([A-Za-z0-9]+)\s*:',  # Only allow alphanumeric parameter names
-        params_section.group(1), 
+        params_section.group(1),
         re.MULTILINE
     )
-    
+
     if not param_names:
         print(f"Warning: Parameters section empty in {template_path}")
-    
+
     print(f"Found parameters in {template_path}: {', '.join(param_names)}")
     return set(param_names)
 
 def filter_parameters_file(template_path, parameters_path):
     """Filter parameters.json to only keep parameters defined in the template"""
     # Get allowed parameters from template
     allowed_params = read_template_parameters(template_path)
-    
+
     # Read current parameters
     with open(parameters_path) as f:
         params_data = json.load(f)
-    
+
     if 'Parameters' not in params_data:
         print(f"Error: {parameters_path} has no 'Parameters' object")
         return False
-    
+
     # Filter parameters
     original_count = len(params_data['Parameters'])
     params_data['Parameters'] = {
-        k: v for k, v in params_data['Parameters'].items() 
+        k: v for k, v in params_data['Parameters'].items()
         if k in allowed_params
     }
     removed_count = original_count - len(params_data['Parameters'])
-    
+
     # Write cleaned parameters back
     with open(parameters_path, 'w') as f:
         json.dump(params_data, f, indent=4)
-    
+
     print(f"Removed {removed_count} parameters not in template")
     return True
 
 if __name__ == "__main__":
     if len(sys.argv) != 3:
         print("Usage: python clean_parameters.py <template_path> <parameters_path>")
         sys.exit(1)
-    
+
     template_path = Path(sys.argv[1])
     parameters_path = Path(sys.argv[2])
-    
+
     if not template_path.exists():
         print(f"Error: Template file {template_path} not found")
         sys.exit(1)
-    
+
     if not parameters_path.exists():
-        print(f"Error: Parameters file {parameters_path} not found") 
+        print(f"Error: Parameters file {parameters_path} not found")
         sys.exit(1)
-    
+
     success = filter_parameters_file(template_path, parameters_path)
     sys.exit(0 if success else 1)

src/emd/cfn/shared/openai_router/main.go

@@ -336,7 +336,7 @@ func isPartialJSON(s string) bool {
 	if strings.HasPrefix(s, "[") && !strings.HasSuffix(s, "]") {
 		return true
 	}
-	
+
 	// Count open vs close braces
 	openBraces := strings.Count(s, "{") + strings.Count(s, "[")
 	closeBraces := strings.Count(s, "}") + strings.Count(s, "]")
@@ -384,7 +384,7 @@ func requestHandler(c *gin.Context) {
 	var payload struct {
 		Model string `json:"model"`
 	}
-	
+
 	// Try to parse model from payload
 	if err := json.Unmarshal(inputBytes, &payload); err == nil && payload.Model != "" {
 		// Use model from payload if present
@@ -411,13 +411,13 @@ func requestHandler(c *gin.Context) {
 		c.JSON(400, gin.H{"error": "Invalid JSON payload"})
 		return
 	}
-	
+
 	// Extract modelID from model field (remove modelTag if present)
 	if modelVal, ok := payloadData["model"].(string); ok {
 		modelParts := strings.Split(modelVal, "/")
 		payloadData["model"] = modelParts[0] // Keep only modelID
 	}
-	
+
 	modifiedBytes, err := json.Marshal(payloadData)
 	if err != nil {
 		c.JSON(500, gin.H{"error": "Failed to modify payload"})
@@ -450,7 +450,7 @@ func requestHandler(c *gin.Context) {
 			// Start streaming in a goroutine
 			go func() {
 				defer closeOnce.Do(func() { close(stream) })
-				
+
 				ctx, cancel := context.WithTimeout(c.Request.Context(), 5*time.Minute)
 				defer cancel()
 
@@ -470,8 +470,8 @@ func requestHandler(c *gin.Context) {
 
 				eventStream := resp.GetStream()
 				defer eventStream.Close()
-				
-				
+
+
 				for event := range eventStream.Events() {
 					switch e := event.(type) {
 					case *sagemakerruntime.PayloadPart:
@@ -482,15 +482,15 @@ func requestHandler(c *gin.Context) {
 
 						chunk := string(e.Bytes)
 						log.Printf("[DEBUG] Received chunk: %s", chunk)
-						
+
 						// Check for finish_reason=stop to end stream
-						if strings.Contains(chunk, `"finish_reason":"stop"`) || 
+						if strings.Contains(chunk, `"finish_reason":"stop"`) ||
 						   strings.Contains(chunk, `"finish_reason": "stop"`) {
 							log.Printf("[DEBUG] Detected finish_reason=stop, ending stream")
 							// stream <- []byte("data: " + chunk + "\n\n")
 							break
 						}
-						
+
 						// Forward chunk as-is in SSE format
 						stream <- []byte(chunk + "\n\n")
 					case *sagemakerruntime.InternalStreamFailure: