mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-07 13:00:59 +08:00
0ddcba9df9
Core behavior change: - Agent NO LONGER auto-prompts setup on first message - Setup only triggers on explicit: '开始配置' / 'setup' / '绑定交易所' - Normal questions answered normally, even without trader configured - AI analysis works without AI model (shows raw data + guidance) - Stock analysis shows 'coming soon' instead of error Intent routing improved: - '你能分析拓维信息吗' → correctly routes to analyze - Chinese suffixes cleaned (吗/呢/这只股票) - More natural language patterns matched Philosophy: Agent should be helpful from minute one. Configuration is optional, not a gate.
109 lines
3.2 KiB
Go
109 lines
3.2 KiB
Go
package agent
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
type IntentType int
|
|
|
|
const (
|
|
IntentChat IntentType = iota
|
|
IntentTrade
|
|
IntentQuery
|
|
IntentAnalyze
|
|
IntentSettings
|
|
IntentHelp
|
|
IntentStatus
|
|
IntentWatch
|
|
IntentStrategy
|
|
)
|
|
|
|
type Intent struct {
|
|
Type IntentType
|
|
Params map[string]string
|
|
Raw string
|
|
}
|
|
|
|
type Router struct {
|
|
tradePatterns []*regexp.Regexp
|
|
queryPatterns []*regexp.Regexp
|
|
analyzePatterns []*regexp.Regexp
|
|
}
|
|
|
|
func NewRouter() *Router {
|
|
return &Router{
|
|
tradePatterns: []*regexp.Regexp{
|
|
regexp.MustCompile(`(?i)(buy|sell|long|short|open|close)\s+(.+)`),
|
|
regexp.MustCompile(`(?i)(做多|做空|买入|卖出|开仓|平仓)\s*(.+)`),
|
|
},
|
|
queryPatterns: []*regexp.Regexp{
|
|
regexp.MustCompile(`(?i)(position|balance|pnl|profit|loss|持仓|余额|盈亏|trader|交易员|账户|仓位|资金)`),
|
|
regexp.MustCompile(`(?i)(多少钱|赚了|亏了|收益|回报)`),
|
|
},
|
|
analyzePatterns: []*regexp.Regexp{
|
|
regexp.MustCompile(`(?i)(analyze|analysis|分析|看看|研究)\s*(.+)`),
|
|
regexp.MustCompile(`(?i)(what.*think|怎么看|你觉得)\s*(.+)?`),
|
|
regexp.MustCompile(`(?i)(走势|趋势|行情|前景|预测)\s*(.+)?`),
|
|
regexp.MustCompile(`(?i)(.+)(走势|趋势|行情|前景|怎么样|如何)`),
|
|
regexp.MustCompile(`(?i)(该不该|适合|能不能|要不要).*(买|卖|做多|做空|入场|开仓).*`),
|
|
regexp.MustCompile(`(?i)(should\s+i|is\s+it\s+good).*(buy|sell|long|short).*`),
|
|
regexp.MustCompile(`(?i)你.*(?:分析|看看|研究|了解)\s*(.+?)(?:吗|呢|嘛|啊|这只|这个)?$`),
|
|
},
|
|
}
|
|
}
|
|
|
|
func (r *Router) Route(text string) Intent {
|
|
text = strings.TrimSpace(text)
|
|
|
|
if strings.HasPrefix(text, "/") {
|
|
return r.routeCommand(text)
|
|
}
|
|
|
|
for _, p := range r.tradePatterns {
|
|
if m := p.FindStringSubmatch(text); m != nil {
|
|
return Intent{Type: IntentTrade, Params: map[string]string{"action": strings.ToLower(m[1]), "detail": m[2]}, Raw: text}
|
|
}
|
|
}
|
|
for _, p := range r.queryPatterns {
|
|
if p.MatchString(text) {
|
|
return Intent{Type: IntentQuery, Raw: text}
|
|
}
|
|
}
|
|
for _, p := range r.analyzePatterns {
|
|
if m := p.FindStringSubmatch(text); m != nil {
|
|
d := ""
|
|
if len(m) > 2 { d = m[2] }
|
|
return Intent{Type: IntentAnalyze, Params: map[string]string{"detail": d}, Raw: text}
|
|
}
|
|
}
|
|
|
|
return Intent{Type: IntentChat, Raw: text}
|
|
}
|
|
|
|
func (r *Router) routeCommand(text string) Intent {
|
|
cmd := strings.ToLower(strings.Fields(text)[0])
|
|
parts := strings.SplitN(text, " ", 2)
|
|
detail := ""
|
|
if len(parts) > 1 { detail = parts[1] }
|
|
|
|
switch cmd {
|
|
case "/start", "/help":
|
|
return Intent{Type: IntentHelp, Raw: text}
|
|
case "/status":
|
|
return Intent{Type: IntentStatus, Raw: text}
|
|
case "/buy", "/sell", "/long", "/short", "/open", "/close":
|
|
return Intent{Type: IntentTrade, Params: map[string]string{"action": strings.TrimPrefix(cmd, "/"), "detail": detail}, Raw: text}
|
|
case "/positions", "/balance", "/pnl", "/traders":
|
|
return Intent{Type: IntentQuery, Raw: text}
|
|
case "/analyze":
|
|
return Intent{Type: IntentAnalyze, Params: map[string]string{"detail": detail}, Raw: text}
|
|
case "/watch", "/unwatch":
|
|
return Intent{Type: IntentWatch, Raw: text}
|
|
case "/strategy":
|
|
return Intent{Type: IntentStrategy, Raw: text}
|
|
default:
|
|
return Intent{Type: IntentChat, Raw: text}
|
|
}
|
|
}
|