Fix formatting

Author: APuzik-devpulse

crates/agentgateway/src/app.rs

@@ -109,7 +109,10 @@ pub async fn run(config: Arc<Config>) -> anyhow::Result<Bound> {
 	.await
 	.context("admin server starts")?;
 	#[cfg(feature = "ui")]
-	admin_server.set_admin_handler(Arc::new(crate::ui::UiHandler::new(config.clone(), stores.guard_registry.clone())));
+	admin_server.set_admin_handler(Arc::new(crate::ui::UiHandler::new(
+		config.clone(),
+		stores.guard_registry.clone(),
+	)));
 	#[cfg(feature = "ui")]
 	info!("serving UI at http://{}/ui", config.admin_addr);
 

crates/agentgateway/src/mcp/handler.rs

@@ -177,7 +177,10 @@ impl Relay {
 				identity: None,
 				metadata: serde_json::Value::Null,
 			};
-			match self.security_guards.evaluate_connection(&server_name, None, &context) {
+			match self
+				.security_guards
+				.evaluate_connection(&server_name, None, &context)
+			{
 				Ok(crate::mcp::security::GuardDecision::Allow) => {
 					tracing::info!(server = %server_name, "Connection guard: allowed");
 				},

crates/agentgateway/src/mcp/security/wasm.rs

@@ -625,20 +625,17 @@ impl NativeGuard for WasmGuard {
 				})?;
 
 			// Try to get evaluate-tool-invoke — if not exported, allow (backward compat)
-			let func_export_idx = match instance.get_export(
-				&mut store,
-				Some(&guard_export_idx),
-				"evaluate-tool-invoke",
-			) {
-				Some(idx) => idx,
-				None => {
-					tracing::debug!(
-						guard_id = %self.guard_id,
-						"WASM guard does not export evaluate-tool-invoke, allowing"
-					);
-					return Ok(GuardDecision::Allow);
-				},
-			};
+			let func_export_idx =
+				match instance.get_export(&mut store, Some(&guard_export_idx), "evaluate-tool-invoke") {
+					Some(idx) => idx,
+					None => {
+						tracing::debug!(
+							guard_id = %self.guard_id,
+							"WASM guard does not export evaluate-tool-invoke, allowing"
+						);
+						return Ok(GuardDecision::Allow);
+					},
+				};
 
 			let func = instance
 				.get_func(&mut store, &func_export_idx)
@@ -715,20 +712,17 @@ impl NativeGuard for WasmGuard {
 				})?;
 
 			// Try to get evaluate-response — if not exported, allow (backward compat)
-			let func_export_idx = match instance.get_export(
-				&mut store,
-				Some(&guard_export_idx),
-				"evaluate-response",
-			) {
-				Some(idx) => idx,
-				None => {
-					tracing::debug!(
-						guard_id = %self.guard_id,
-						"WASM guard does not export evaluate-response, allowing"
-					);
-					return Ok(GuardDecision::Allow);
-				},
-			};
+			let func_export_idx =
+				match instance.get_export(&mut store, Some(&guard_export_idx), "evaluate-response") {
+					Some(idx) => idx,
+					None => {
+						tracing::debug!(
+							guard_id = %self.guard_id,
+							"WASM guard does not export evaluate-response, allowing"
+						);
+						return Ok(GuardDecision::Allow);
+					},
+				};
 
 			let func = instance
 				.get_func(&mut store, &func_export_idx)

crates/agentgateway/src/ui.rs

@@ -209,7 +209,10 @@ fn collect_wasm_schemas_from_config(config: &Value, schemas: &mut serde_json::Ma
 
 /// Extract schemas from a list of guard config values.
 #[allow(unused_variables)]
-fn collect_wasm_schemas_from_guards(guards: &[Value], schemas: &mut serde_json::Map<String, Value>) {
+fn collect_wasm_schemas_from_guards(
+	guards: &[Value],
+	schemas: &mut serde_json::Map<String, Value>,
+) {
 	for guard_val in guards {
 		let Some(guard_type) = guard_val.get("type").and_then(|v| v.as_str()) else {
 			continue;