From 2696618ca00782bb496eca84a0eebbfe9e50bd9a Mon Sep 17 00:00:00 2001
From: Klesti Selimaj <klesti.selimay@icloud.com>
Date: Wed, 29 Jul 2026 01:48:28 +0200
Subject: [PATCH] More flexible signal handling

---
 pulse-sdk/src/lib.rs    | 12 ++----------
 pulse-sdk/src/plugin.rs |  1 +
 2 files changed, 3 insertions(+), 10 deletions(-)

diff --git a/pulse-sdk/src/lib.rs b/pulse-sdk/src/lib.rs
index 189435a..5147e98 100644
--- a/pulse-sdk/src/lib.rs
+++ b/pulse-sdk/src/lib.rs
@@ -122,12 +122,7 @@ pub trait Risk {
             RiskEngineMessage::Initialize => self.initialize().await,
             RiskEngineMessage::Command { command, args } => self.command(command, args).await,
             RiskEngineMessage::WatchList(watchlist) => self.watchlist(watchlist).await,
-            RiskEngineMessage::Signal(signal) => {
-                self.send(&prelude::RiskMessage::Signal(
-                    self.signal_request(signal).await?,
-                ))
-                .await
-            }
+            RiskEngineMessage::Signal(signal) => self.signal_request(signal).await,
         }
     }
 
@@ -143,10 +138,7 @@ pub trait Risk {
         unimplemented!("Risk::watchlist")
     }
 
-    async fn signal_request(
-        &self,
-        _signal: prelude::StrategySignal,
-    ) -> tokio::io::Result<prelude::RiskSignal> {
+    async fn signal_request(&self, _signal: prelude::StrategySignal) -> tokio::io::Result<()> {
         unimplemented!("Risk::signal_request")
     }
 }
diff --git a/pulse-sdk/src/plugin.rs b/pulse-sdk/src/plugin.rs
index fc54bf7..e69c51c 100644
--- a/pulse-sdk/src/plugin.rs
+++ b/pulse-sdk/src/plugin.rs
@@ -97,6 +97,7 @@ pub struct StrategySignal {
 pub enum RiskSignal {
     Approve(Signal),
     Reject {
+        signal: StrategySignal,
         rejection_confidence: f32,
         reason: String,
     },