yougandar commited on
Commit
3ef3ad4
Β·
verified Β·
1 Parent(s): 3a6a277

Update rsi_agent.py

Browse files
Files changed (1) hide show
  1. rsi_agent.py +18 -2
rsi_agent.py CHANGED
@@ -1,7 +1,23 @@
1
  from ta.momentum import RSIIndicator
2
 
3
  class RSIAlertAgent:
 
 
 
 
 
4
  def run(self, df):
5
  rsi = RSIIndicator(close=df['Close'], window=14).rsi().iloc[-1]
6
- signal = "Overbought" if rsi > 70 else "Oversold" if rsi < 30 else "Neutral"
7
- return {"signal": signal, "rsi": round(rsi, 2)}
 
 
 
 
 
 
 
 
 
 
 
 
1
  from ta.momentum import RSIIndicator
2
 
3
  class RSIAlertAgent:
4
+ """
5
+ Detects RSI-based signals for overbought/oversold conditions.
6
+ Returns structured signal info for use in dashboards and decision logic.
7
+ """
8
+
9
  def run(self, df):
10
  rsi = RSIIndicator(close=df['Close'], window=14).rsi().iloc[-1]
11
+
12
+ if rsi > 70:
13
+ signal = "Overbought"
14
+ elif rsi < 30:
15
+ signal = "Oversold"
16
+ else:
17
+ signal = "Neutral"
18
+
19
+ return {
20
+ "signal": signal,
21
+ "rsi": round(rsi, 2),
22
+ "status": "βœ…" if signal != "Neutral" else "⚠️"
23
+ }