mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-06-06 05:51:19 +08:00
309 lines
12 KiB
Go
309 lines
12 KiB
Go
package agent
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"strings"
|
||
)
|
||
|
||
func (a *Agent) executeTraderManagementAction(storeUserID string, userID int64, lang, text string, session skillSession) string {
|
||
switch session.Action {
|
||
case "query":
|
||
return formatReadFastPathResponse(lang, "list_traders", a.toolListTraders(storeUserID))
|
||
case "start", "stop", "delete":
|
||
if msg, waiting := beginConfirmationIfNeeded(userID, lang, &session, defaultIfEmpty(session.TargetRef.Name, session.TargetRef.ID)); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
if msg, waiting := awaitingConfirmationButNotApproved(lang, session, text); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
var resp string
|
||
switch session.Action {
|
||
case "start":
|
||
resp = a.toolStartTrader(storeUserID, session.TargetRef.ID)
|
||
case "stop":
|
||
resp = a.toolStopTrader(storeUserID, session.TargetRef.ID)
|
||
case "delete":
|
||
resp = a.toolDeleteTrader(storeUserID, session.TargetRef.ID)
|
||
}
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "执行失败：" + errMsg
|
||
}
|
||
return "Action failed: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return fmt.Sprintf("已完成交易员操作：%s。", session.Action)
|
||
}
|
||
return fmt.Sprintf("Completed trader action: %s.", session.Action)
|
||
case "update":
|
||
newName := extractTraderName(text)
|
||
if newName == "" {
|
||
newName = extractPostKeywordName(text, []string{"改成", "改为", "rename to"})
|
||
}
|
||
if newName == "" {
|
||
a.saveSkillSession(userID, session)
|
||
if lang == "zh" {
|
||
return "目前更新交易员这条 skill 先支持改名。请直接告诉我新的名字。"
|
||
}
|
||
return "This trader update skill currently supports renaming first. Tell me the new name."
|
||
}
|
||
args := manageTraderArgs{Action: "update", TraderID: session.TargetRef.ID, Name: newName}
|
||
resp := a.toolUpdateTrader(storeUserID, args)
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "更新交易员失败：" + errMsg
|
||
}
|
||
return "Failed to update trader: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return fmt.Sprintf("已将交易员改名为“%s”。", newName)
|
||
}
|
||
return fmt.Sprintf("Renamed trader to %q.", newName)
|
||
default:
|
||
return ""
|
||
}
|
||
}
|
||
|
||
func (a *Agent) executeExchangeManagementAction(storeUserID string, userID int64, lang, text string, session skillSession) string {
|
||
switch session.Action {
|
||
case "delete":
|
||
if msg, waiting := beginConfirmationIfNeeded(userID, lang, &session, defaultIfEmpty(session.TargetRef.Name, session.TargetRef.ID)); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
if msg, waiting := awaitingConfirmationButNotApproved(lang, session, text); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
args, _ := json.Marshal(map[string]any{"action": "delete", "exchange_id": session.TargetRef.ID})
|
||
resp := a.toolManageExchangeConfig(storeUserID, string(args))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "删除交易所配置失败：" + errMsg
|
||
}
|
||
return "Failed to delete exchange config: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return "已删除交易所配置。"
|
||
}
|
||
return "Deleted exchange config."
|
||
case "update":
|
||
accountName := extractTraderName(text)
|
||
if accountName == "" {
|
||
accountName = extractPostKeywordName(text, []string{"改成", "改为", "账户名改成", "rename to"})
|
||
}
|
||
payload := map[string]any{"action": "update", "exchange_id": session.TargetRef.ID}
|
||
if accountName != "" {
|
||
payload["account_name"] = accountName
|
||
}
|
||
if containsAny(strings.ToLower(text), []string{"启用", "enable"}) {
|
||
payload["enabled"] = true
|
||
}
|
||
if containsAny(strings.ToLower(text), []string{"禁用", "disable"}) {
|
||
payload["enabled"] = false
|
||
}
|
||
if len(payload) == 2 {
|
||
a.saveSkillSession(userID, session)
|
||
if lang == "zh" {
|
||
return "目前更新交易所 skill 先支持改账户名和启用/禁用。请告诉我你要改什么。"
|
||
}
|
||
return "This exchange update skill currently supports renaming and enable/disable. Tell me what to change."
|
||
}
|
||
raw, _ := json.Marshal(payload)
|
||
resp := a.toolManageExchangeConfig(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "更新交易所配置失败：" + errMsg
|
||
}
|
||
return "Failed to update exchange config: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return "已更新交易所配置。"
|
||
}
|
||
return "Updated exchange config."
|
||
default:
|
||
return ""
|
||
}
|
||
}
|
||
|
||
func (a *Agent) executeModelManagementAction(storeUserID string, userID int64, lang, text string, session skillSession) string {
|
||
switch session.Action {
|
||
case "delete":
|
||
if msg, waiting := beginConfirmationIfNeeded(userID, lang, &session, defaultIfEmpty(session.TargetRef.Name, session.TargetRef.ID)); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
if msg, waiting := awaitingConfirmationButNotApproved(lang, session, text); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
raw, _ := json.Marshal(map[string]any{"action": "delete", "model_id": session.TargetRef.ID})
|
||
resp := a.toolManageModelConfig(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "删除模型配置失败：" + errMsg
|
||
}
|
||
return "Failed to delete model config: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return "已删除模型配置。"
|
||
}
|
||
return "Deleted model config."
|
||
case "update":
|
||
payload := map[string]any{"action": "update", "model_id": session.TargetRef.ID}
|
||
if url := extractURL(text); url != "" {
|
||
payload["custom_api_url"] = url
|
||
}
|
||
if containsAny(strings.ToLower(text), []string{"启用", "enable"}) {
|
||
payload["enabled"] = true
|
||
}
|
||
if containsAny(strings.ToLower(text), []string{"禁用", "disable"}) {
|
||
payload["enabled"] = false
|
||
}
|
||
if modelName := extractPostKeywordName(text, []string{"model name", "模型名", "模型名称", "改成"}); modelName != "" {
|
||
payload["custom_model_name"] = modelName
|
||
}
|
||
if len(payload) == 2 {
|
||
a.saveSkillSession(userID, session)
|
||
if lang == "zh" {
|
||
return "目前更新模型 skill 先支持改 URL、模型名和启用状态。请告诉我你要改什么。"
|
||
}
|
||
return "This model update skill currently supports URL, model name, and enabled state."
|
||
}
|
||
raw, _ := json.Marshal(payload)
|
||
resp := a.toolManageModelConfig(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "更新模型配置失败：" + errMsg
|
||
}
|
||
return "Failed to update model config: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return "已更新模型配置。"
|
||
}
|
||
return "Updated model config."
|
||
default:
|
||
return ""
|
||
}
|
||
}
|
||
|
||
func (a *Agent) executeStrategyManagementAction(storeUserID string, userID int64, lang, text string, session skillSession) string {
|
||
switch session.Action {
|
||
case "activate":
|
||
raw, _ := json.Marshal(map[string]any{"action": "activate", "strategy_id": session.TargetRef.ID})
|
||
resp := a.toolManageStrategy(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "激活策略失败：" + errMsg
|
||
}
|
||
return "Failed to activate strategy: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return "已激活策略。"
|
||
}
|
||
return "Activated strategy."
|
||
case "duplicate":
|
||
newName := extractTraderName(text)
|
||
if newName == "" {
|
||
newName = extractPostKeywordName(text, []string{"叫", "名为", "改成", "rename to"})
|
||
}
|
||
if newName == "" {
|
||
a.saveSkillSession(userID, session)
|
||
if lang == "zh" {
|
||
return "复制策略时，我还需要一个新名称。"
|
||
}
|
||
return "I still need a new name for the duplicated strategy."
|
||
}
|
||
raw, _ := json.Marshal(map[string]any{"action": "duplicate", "strategy_id": session.TargetRef.ID, "name": newName})
|
||
resp := a.toolManageStrategy(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "复制策略失败：" + errMsg
|
||
}
|
||
return "Failed to duplicate strategy: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return fmt.Sprintf("已复制策略，新名称为“%s”。", newName)
|
||
}
|
||
return fmt.Sprintf("Duplicated strategy as %q.", newName)
|
||
case "delete":
|
||
if msg, waiting := beginConfirmationIfNeeded(userID, lang, &session, defaultIfEmpty(session.TargetRef.Name, session.TargetRef.ID)); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
if msg, waiting := awaitingConfirmationButNotApproved(lang, session, text); waiting {
|
||
a.saveSkillSession(userID, session)
|
||
return msg
|
||
}
|
||
raw, _ := json.Marshal(map[string]any{"action": "delete", "strategy_id": session.TargetRef.ID})
|
||
resp := a.toolManageStrategy(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "删除策略失败：" + errMsg
|
||
}
|
||
return "Failed to delete strategy: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return "已删除策略。"
|
||
}
|
||
return "Deleted strategy."
|
||
case "update":
|
||
newName := extractTraderName(text)
|
||
if newName == "" {
|
||
newName = extractPostKeywordName(text, []string{"改成", "改为", "rename to"})
|
||
}
|
||
if newName == "" {
|
||
a.saveSkillSession(userID, session)
|
||
if lang == "zh" {
|
||
return "目前更新策略 skill 先支持改名。请告诉我新的策略名称。"
|
||
}
|
||
return "This strategy update skill currently supports renaming first."
|
||
}
|
||
raw, _ := json.Marshal(map[string]any{"action": "update", "strategy_id": session.TargetRef.ID, "name": newName})
|
||
resp := a.toolManageStrategy(storeUserID, string(raw))
|
||
a.clearSkillSession(userID)
|
||
if errMsg := parseSkillError(resp); strings.Contains(resp, `"error"`) {
|
||
if lang == "zh" {
|
||
return "更新策略失败：" + errMsg
|
||
}
|
||
return "Failed to update strategy: " + errMsg
|
||
}
|
||
if lang == "zh" {
|
||
return fmt.Sprintf("已将策略改名为“%s”。", newName)
|
||
}
|
||
return fmt.Sprintf("Renamed strategy to %q.", newName)
|
||
default:
|
||
return ""
|
||
}
|
||
}
|
||
|
||
func (a *Agent) handleTraderDiagnosisSkill(storeUserID, lang, text string) string {
|
||
raw := a.toolListTraders(storeUserID)
|
||
list := formatReadFastPathResponse(lang, "list_traders", raw)
|
||
if lang == "zh" {
|
||
return "现象：这是交易员运行诊断问题。\n优先排查：\n1. 交易员是否已创建并处于运行状态。\n2. 绑定的模型、交易所、策略是否齐全。\n3. 是“没有启动”、还是“启动了但 AI 没有下单”、还是“下单失败”。\n当前交易员概览：\n" + list
|
||
}
|
||
return "This looks like a trader diagnosis issue.\nCheck whether the trader exists, is running, and has model/exchange/strategy bindings.\nCurrent trader overview:\n" + list
|
||
}
|
||
|
||
func (a *Agent) handleStrategyDiagnosisSkill(storeUserID, lang, text string) string {
|
||
raw := a.toolGetStrategies(storeUserID)
|
||
list := formatReadFastPathResponse(lang, "get_strategies", raw)
|
||
if lang == "zh" {
|
||
return "现象：这是策略或提示词生效问题。\n优先排查：\n1. 你改的是策略模板，还是 trader 上的 custom prompt。\n2. 策略是否真的保存成功。\n3. 运行结果不符合预期，是配置问题还是市场条件问题。\n当前策略概览：\n" + list
|
||
}
|
||
return "This looks like a strategy or prompt diagnosis issue.\nCheck whether you changed the strategy template or a trader-specific prompt override.\nCurrent strategy overview:\n" + list
|
||
}
|