diff --git a/agent/agent.go b/agent/agent.go index 7dced256..de75cadf 100644 --- a/agent/agent.go +++ b/agent/agent.go @@ -421,6 +421,9 @@ func (a *Agent) handleMessageForStoreUser(ctx context.Context, storeUserID strin if reply, handled := a.handleTradeConfirmation(ctx, userID, text, lang); handled { return reply, nil } + if reply, handled := a.handleModelWalletBalanceQuestion(storeUserID, lang, text); handled { + return reply, nil + } // Everything else goes through the planner and tool system. return a.thinkAndAct(ctx, storeUserID, userID, lang, text) @@ -468,6 +471,12 @@ func (a *Agent) handleMessageStreamForStoreUser(ctx context.Context, storeUserID } return reply, nil } + if reply, handled := a.handleModelWalletBalanceQuestion(storeUserID, lang, text); handled { + if onEvent != nil { + emitStreamText(onEvent, reply) + } + return reply, nil + } return a.thinkAndActStream(ctx, storeUserID, userID, lang, text, onEvent) } diff --git a/agent/model_wallet_fastpath.go b/agent/model_wallet_fastpath.go new file mode 100644 index 00000000..6670defe --- /dev/null +++ b/agent/model_wallet_fastpath.go @@ -0,0 +1,86 @@ +package agent + +import ( + "fmt" + "strconv" + "strings" +) + +func isModelWalletBalanceQuestion(text string) bool { + lower := strings.ToLower(strings.TrimSpace(text)) + if lower == "" || !strings.Contains(lower, "claw402") { + return false + } + return containsAny(lower, []string{"余额", "balance", "usdc"}) && + containsAny(lower, []string{"钱包", "wallet", "主钱包", "base"}) +} + +func (a *Agent) handleModelWalletBalanceQuestion(storeUserID, lang, text string) (string, bool) { + if !isModelWalletBalanceQuestion(text) || a == nil || a.store == nil { + return "", false + } + models, err := a.store.AIModel().List(storeUserID) + if err != nil { + if lang == "zh" { + return "我现在读取模型配置失败，暂时查不到 claw402 钱包余额。", true + } + return "I could not read model configs, so I cannot check the claw402 wallet balance right now.", true + } + + var matches []safeModelToolConfig + for _, model := range models { + if model == nil || strings.ToLower(strings.TrimSpace(model.Provider)) != "claw402" { + continue + } + matches = append(matches, safeModelForTool(model)) + } + if len(matches) == 0 { + if lang == "zh" { + return "当前没有找到 claw402 模型钱包配置。", true + } + return "No claw402 model wallet config was found.", true + } + + if lang == "zh" { + lines := []string{"当前 claw402 模型钱包余额："} + for _, model := range matches { + name := defaultIfEmpty(model.Name, model.ID) + lines = append(lines, fmt.Sprintf("- %s：%s USDC", name, defaultIfEmpty(model.BalanceUSDC, "暂时无法读取"))) + if strings.TrimSpace(model.WalletAddress) != "" { + lines = append(lines, fmt.Sprintf(" 钱包地址：%s", model.WalletAddress)) + } + if balanceIsZero(model.BalanceUSDC) { + if model.Enabled { + lines = append(lines, " 这个模型配置已启用，但钱包余额为 0 USDC；这不是“未启用”，而是需要先充值 Base USDC 后才能稳定调用。") + } else { + lines = append(lines, " 钱包余额为 0 USDC；启用并充值 Base USDC 后才能稳定调用。") + } + } + } + lines = append(lines, "注意：这是 claw402/Base 模型支付钱包余额，不是 OKX/Binance 等交易所账户余额。") + return strings.Join(lines, "\n"), true + } + + lines := []string{"Current claw402 model wallet balance:"} + for _, model := range matches { + name := defaultIfEmpty(model.Name, model.ID) + lines = append(lines, fmt.Sprintf("- %s: %s USDC", name, defaultIfEmpty(model.BalanceUSDC, "unavailable"))) + if strings.TrimSpace(model.WalletAddress) != "" { + lines = append(lines, fmt.Sprintf(" Wallet address: %s", model.WalletAddress)) + } + if balanceIsZero(model.BalanceUSDC) { + lines = append(lines, " This model config may be enabled, but the wallet balance is 0 USDC; recharge Base USDC before relying on it.") + } + } + lines = append(lines, "Note: this is the claw402/Base model payment wallet balance, not an exchange account balance.") + return strings.Join(lines, "\n"), true +} + +func balanceIsZero(value string) bool { + trimmed := strings.TrimSpace(value) + if trimmed == "" { + return false + } + parsed, err := strconv.ParseFloat(trimmed, 64) + return err == nil && parsed <= 0 +} diff --git a/agent/skill_dispatcher.go b/agent/skill_dispatcher.go index a28eb802..7147d6a6 100644 --- a/agent/skill_dispatcher.go +++ b/agent/skill_dispatcher.go @@ -244,6 +244,9 @@ func formatOptionList(prefix string, options []traderSkillOption) string { if label == "" { label = option.ID } + if hint := strings.TrimSpace(option.Hint); hint != "" { + label += "（" + hint + "）" + } if option.Enabled { label += "（已启用）" } else { @@ -267,6 +270,28 @@ func parseSkillError(raw string) string { return strings.TrimSpace(raw) } +func modelWalletBalanceHint(model *store.AIModel) string { + if model == nil || !agentProviderSupportsUSDCBalance(model.Provider) { + return "" + } + privateKey := strings.TrimSpace(string(model.APIKey)) + if privateKey == "" { + return "钱包未配置" + } + walletAddress, err := agentWalletAddressFromPrivateKey(privateKey) + if err != nil || strings.TrimSpace(walletAddress) == "" { + return "钱包私钥无效" + } + balance, err := agentQueryUSDCBalanceCached(walletAddress) + if err != nil { + return "钱包余额暂时无法读取" + } + if balance <= 0 { + return "钱包余额 0 USDC，需充值后才能稳定调用" + } + return fmt.Sprintf("钱包余额 %.4g USDC", balance) +} + func (a *Agent) loadEnabledModelOptions(storeUserID string) []traderSkillOption { if a.store == nil { return nil @@ -284,6 +309,7 @@ func (a *Agent) loadEnabledModelOptions(storeUserID string) []traderSkillOption hint := strings.Join(cleanStringList([]string{ strings.TrimSpace(model.CustomModelName), strings.TrimSpace(model.Provider), + modelWalletBalanceHint(model), }), " / ") out = append(out, traderSkillOption{ID: model.ID, Name: name, Hint: hint, Enabled: model.Enabled}) } @@ -1046,7 +1072,7 @@ func findOptionByIDOrName(options []traderSkillOption, query string) *traderSkil return nil } for i, opt := range options { - if opt.ID == query || strings.EqualFold(opt.Name, query) { + if opt.ID == query || strings.EqualFold(opt.Name, query) || strings.EqualFold(opt.Hint, query) { return &options[i] } } @@ -1060,7 +1086,12 @@ func findUniqueContainingOption(options []traderSkillOption, query string) *trad } matches := make([]traderSkillOption, 0, 1) for _, opt := range options { - if strings.Contains(strings.ToLower(opt.Name), query) || strings.Contains(query, strings.ToLower(opt.Name)) { + name := strings.ToLower(strings.TrimSpace(opt.Name)) + hint := strings.ToLower(strings.TrimSpace(opt.Hint)) + id := strings.ToLower(strings.TrimSpace(opt.ID)) + if (name != "" && (strings.Contains(name, query) || strings.Contains(query, name))) || + (hint != "" && (strings.Contains(hint, query) || strings.Contains(query, hint))) || + (id != "" && (strings.Contains(id, query) || strings.Contains(query, id))) { matches = append(matches, opt) } } diff --git a/agent/skill_management_handlers.go b/agent/skill_management_handlers.go index df738855..33fa02dd 100644 --- a/agent/skill_management_handlers.go +++ b/agent/skill_management_handlers.go @@ -1472,11 +1472,37 @@ func (a *Agent) describeModel(storeUserID, lang string, target *EntityReference) model = &payload.ModelConfigs[0] } if lang == "zh" { - return fmt.Sprintf("模型配置“%s”详情：\n- Provider：%s\n- 已启用：%t\n- API Key：%t\n- URL：%s\n- Model Name：%s", - defaultIfEmpty(model.Name, model.ID), model.Provider, model.Enabled, model.HasAPIKey, defaultIfEmpty(model.CustomAPIURL, "未设置"), defaultIfEmpty(model.CustomModelName, "未设置")), true + lines := []string{ + fmt.Sprintf("模型配置“%s”详情：", defaultIfEmpty(model.Name, model.ID)), + fmt.Sprintf("- Provider：%s", model.Provider), + fmt.Sprintf("- 已启用：%t", model.Enabled), + fmt.Sprintf("- API Key：%t", model.HasAPIKey), + fmt.Sprintf("- URL：%s", defaultIfEmpty(model.CustomAPIURL, "未设置")), + fmt.Sprintf("- Model Name：%s", defaultIfEmpty(model.CustomModelName, "未设置")), + } + if strings.TrimSpace(model.WalletAddress) != "" { + lines = append(lines, fmt.Sprintf("- 钱包地址：%s", model.WalletAddress)) + } + if strings.TrimSpace(model.BalanceUSDC) != "" { + lines = append(lines, fmt.Sprintf("- 钱包余额：%s USDC", model.BalanceUSDC)) + } + return strings.Join(lines, "\n"), true } - return fmt.Sprintf("Model config %q details:\n- Provider: %s\n- Enabled: %t\n- API key present: %t\n- URL: %s\n- Model name: %s", - defaultIfEmpty(model.Name, model.ID), model.Provider, model.Enabled, model.HasAPIKey, defaultIfEmpty(model.CustomAPIURL, "not set"), defaultIfEmpty(model.CustomModelName, "not set")), true + lines := []string{ + fmt.Sprintf("Model config %q details:", defaultIfEmpty(model.Name, model.ID)), + fmt.Sprintf("- Provider: %s", model.Provider), + fmt.Sprintf("- Enabled: %t", model.Enabled), + fmt.Sprintf("- API key present: %t", model.HasAPIKey), + fmt.Sprintf("- URL: %s", defaultIfEmpty(model.CustomAPIURL, "not set")), + fmt.Sprintf("- Model name: %s", defaultIfEmpty(model.CustomModelName, "not set")), + } + if strings.TrimSpace(model.WalletAddress) != "" { + lines = append(lines, fmt.Sprintf("- Wallet address: %s", model.WalletAddress)) + } + if strings.TrimSpace(model.BalanceUSDC) != "" { + lines = append(lines, fmt.Sprintf("- Wallet balance: %s USDC", model.BalanceUSDC)) + } + return strings.Join(lines, "\n"), true } func findTraderByReference(items []safeTraderToolConfig, target *EntityReference) *safeTraderToolConfig { diff --git a/agent/skill_registry.go b/agent/skill_registry.go index bc7e576d..8b231c09 100644 --- a/agent/skill_registry.go +++ b/agent/skill_registry.go @@ -18,6 +18,8 @@ type SkillDefinition struct { Domain string `json:"domain"` Description string `json:"description"` Intents []string `json:"intents,omitempty"` + Capabilities []string `json:"capabilities,omitempty"` + DynamicRules []string `json:"dynamic_rules,omitempty"` Actions map[string]SkillActionDefinition `json:"actions,omitempty"` ToolMapping map[string]string `json:"tool_mapping,omitempty"` FieldConstraints map[string]SkillFieldConstraint `json:"field_constraints,omitempty"` @@ -96,6 +98,8 @@ func normalizeSkillDefinition(def SkillDefinition) SkillDefinition { def.Domain = strings.TrimSpace(def.Domain) def.Description = strings.TrimSpace(def.Description) def.Intents = cleanStringList(def.Intents) + def.Capabilities = cleanStringList(def.Capabilities) + def.DynamicRules = cleanStringList(def.DynamicRules) if len(def.Actions) > 0 { normalized := make(map[string]SkillActionDefinition, len(def.Actions)) @@ -194,6 +198,9 @@ func buildSkillRoutingSummary(lang string, skillNames []string) string { continue } parts := []string{strings.TrimSpace(def.Description)} + if len(def.DynamicRules) > 0 { + parts = append(parts, strings.Join(def.DynamicRules, " ")) + } switch name { case "trader_management": if lang == "zh" { @@ -221,6 +228,20 @@ func buildSkillDefinitionSummary(lang string, skillNames []string) string { continue } parts := []string{strings.TrimSpace(def.Description)} + if len(def.Capabilities) > 0 { + if lang == "zh" { + parts = append(parts, "能力: "+strings.Join(def.Capabilities, "；")) + } else { + parts = append(parts, "capabilities: "+strings.Join(def.Capabilities, "; ")) + } + } + if len(def.DynamicRules) > 0 { + if lang == "zh" { + parts = append(parts, "规则: "+strings.Join(def.DynamicRules, "；")) + } else { + parts = append(parts, "rules: "+strings.Join(def.DynamicRules, "; ")) + } + } if action, ok := def.Actions["create"]; ok && len(action.RequiredSlots) > 0 { if lang == "zh" { parts = append(parts, "创建必填: "+formatRequiredSlotList(lang, action.RequiredSlots)) diff --git a/agent/skills/exchange_diagnosis.json b/agent/skills/exchange_diagnosis.json index c8d9b0ba..cebe3363 100644 --- a/agent/skills/exchange_diagnosis.json +++ b/agent/skills/exchange_diagnosis.json @@ -2,5 +2,17 @@ "name": "exchange_diagnosis", "kind": "diagnosis", "domain": "exchange", - "description": "当用户反馈交易所 API 连接失败、签名错误、timestamp 异常、权限不足、IP 白名单限制、账户不可用等问题时调用。适用于用户在手动配置或运行交易员时遇到的交易所接入故障。不用于创建、修改、删除或查询交易所配置这类管理操作。" + "description": "当用户反馈交易所 API 连接失败、签名错误、timestamp 异常、权限不足、IP 白名单限制、账户不可用、余额读取失败、下单失败或仓位模式错误等问题时调用。适用于用户在手动配置或运行交易员时遇到的交易所接入与执行故障。不用于创建、修改、删除或查询交易所配置这类管理操作。", + "capabilities": [ + "区分凭证缺失、签名错误、时间戳偏差、IP 白名单、权限不足、余额不足、仓位模式和 symbol 不可交易等原因", + "解释不同交易所的必填字段差异，尤其是 OKX/Bitget/KuCoin passphrase、Hyperliquid 钱包地址、Aster signer/private key、Lighter API key private key", + "把交易所原始错误翻译成新手可执行的修复步骤" + ], + "dynamic_rules": [ + "交易所连接失败优先按顺序排查：配置是否启用 -> 必填凭证是否齐全 -> API Key/Secret/Passphrase 是否填反或过期 -> 系统时间/timestamp -> IP 白名单 -> 合约/交易权限 -> 测试网/主网是否选错。", + "OKX、Bitget、KuCoin 的 passphrase/API 口令不是可选项；如果缺失，必须明确提示补齐。", + "下单失败时继续排查：账户余额/可用保证金 -> 杠杆限制 -> 仓位模式（单向/双向） -> symbol 是否支持合约交易 -> 最小下单金额/数量。", + "Hyperliquid、Aster、Lighter 这类钱包/DEX 配置错误时，不要用 CEX 的 API Key/Secret 逻辑套用；按各自 required fields 解释。", + "诊断回复不得展示完整 API Key、Secret、Passphrase 或私钥。" + ] } diff --git a/agent/skills/exchange_management.json b/agent/skills/exchange_management.json index bcb29a06..b2c12cdf 100644 --- a/agent/skills/exchange_management.json +++ b/agent/skills/exchange_management.json @@ -28,8 +28,8 @@ }, "passphrase": { "type": "credential", - "required_for": ["okx"], - "description": "OKX 专用 Passphrase，OKX 账户启用前必须填写，其他交易所不需要。" + "required_for": ["okx", "bitget", "kucoin"], + "description": "OKX、Bitget、KuCoin 专用 Passphrase/API 口令，对这些交易所启用前必须填写；Binance、Bybit、Gate、Indodax 通常不需要。" }, "testnet": { "type": "bool", @@ -94,11 +94,10 @@ "api_key 格式：至少 8 位字母数字，不符合时提示用户重新输入完整 Key。", "secret_key 格式：至少 8 位字母数字，或十六进制格式，不符合时提示用户重新输入。", "OKX 账户启用前必须填写 passphrase，否则拒绝启用并提示补填。", - "Bitget/KuCoin 页面流程里也需要 passphrase，缺失时应明确提示补填。", + "Bitget 和 KuCoin 页面流程里也需要 passphrase/API 口令，不能回答“没有就留空”；缺失时应明确提示补填。", "Hyperliquid 创建/更新时应与手动页面保持一致：至少收集 api_key + hyperliquid_wallet_addr。", "Hyperliquid 账户启用前必须填写 hyperliquid_wallet_addr。", "若用户使用 Hyperliquid unified account 模式，应明确记录 hyperliquid_unified_account 开关状态。", - "Aster 账户启用前必须填写 aster_user、aster_signer、aster_private_key 三个字段。", "Aster 账户启用前必须填写 aster_user、aster_signer、aster_private_key 三个字段，任一缺失都不能启用。", "Lighter 账户启用前必须填写 lighter_wallet_addr + lighter_api_key_private_key；若当前账户模式还依赖 lighter_private_key，也要先补齐后再启用。", "lighter_api_key_index 超出 0～255 时自动收敛到边界值并告知用户。", @@ -125,6 +124,7 @@ "dynamic_rules": [ "确认 exchange_type 后，根据 per_exchange_required_fields 决定需要追问哪些凭证字段。", "Binance/Bybit/Gate/Indodax 需要 API Key + Secret；OKX/Bitget/KuCoin 还必须追问 passphrase；Hyperliquid 必须追问 api_key + 钱包地址，并允许记录 unified account 开关；Aster 必须追问 user/signer/private_key；Lighter 必须追问钱包地址和 api_key_private_key。", + "如果用户选择 OKX、Bitget 或 KuCoin，不能把 passphrase 说成可选项；没有 passphrase 时应停在补字段，不要创建半成品。", "凭证字段格式不符时，用人话告知用户正确格式，不要静默丢弃。", "若当前父任务只是缺一个可用交易所，本动作完成后应允许父任务恢复并消费新的 exchange_id。", "若请求只是在启用已有交易所，不应误走 create，应改走 update_status。" diff --git a/agent/skills/model_diagnosis.json b/agent/skills/model_diagnosis.json index d47e0d77..f09b5a90 100644 --- a/agent/skills/model_diagnosis.json +++ b/agent/skills/model_diagnosis.json @@ -2,5 +2,16 @@ "name": "model_diagnosis", "kind": "diagnosis", "domain": "model", - "description": "当用户反馈模型配置失败、API Key 无效、Base URL 非法、模型名不匹配、调用返回错误、模型不可用等问题时调用。适用于用户在接入或测试大模型时遇到的配置与兼容性故障。不用于创建、修改、删除或查询模型配置这类管理操作。" + "description": "当用户反馈模型配置失败、API Key 无效、Base URL 非法、模型名不匹配、调用返回错误、模型不可用、claw402 钱包余额不足或支付失败等问题时调用。适用于用户在接入或测试大模型时遇到的配置、兼容性、支付和调用故障。不用于创建、修改、删除或查询模型配置这类管理操作。", + "capabilities": [ + "区分模型未启用、凭证缺失、endpoint/model name 配置错误、钱包余额不足、上游限流或网关异常", + "对 claw402 / blockrun-base 这类钱包付费模型，解释钱包地址、USDC 余额和支付状态", + "给出不泄露敏感凭证的下一步修复建议" + ], + "dynamic_rules": [ + "诊断模型不可用时，按顺序检查：是否存在该模型配置 -> enabled 是否为 true -> provider 是否支持 -> 凭证/API Key 或钱包私钥是否存在 -> custom_api_url 是否合法 HTTPS 或可留空 -> custom_model_name 是否有默认值或已填写 -> 钱包余额/支付状态 -> 上游限流、超时或网关错误。", + "claw402 是模型 provider，使用 Base USDC 钱包按次付费；余额为 0 USDC 时应明确说需要充值，不要说成“未配置模型”。", + "429/rate_limit_error、空响应、超时不应默认归因为余额不足；只有工具结果或错误文本指向余额/支付失败时才这么判断。", + "任何诊断回复都不得展示 API Key、钱包私钥或完整敏感凭证。" + ] } diff --git a/agent/skills/model_management.json b/agent/skills/model_management.json index 12677dc2..5e06b36e 100644 --- a/agent/skills/model_management.json +++ b/agent/skills/model_management.json @@ -40,6 +40,10 @@ "custom_api_url 若填写，必须是合法 HTTPS 地址，系统拒绝 HTTP 地址，提示用户改用 HTTPS。", "启用（enabled=true）前必须填写 provider 对应的凭证；如果 custom_model_name 留空，则系统应先尝试使用 provider 默认模型。", "启用（enabled=true）前，custom_api_url 若填写必须是合法 HTTPS 地址；不允许用 HTTP 地址硬启用。", + "claw402 是 AI 模型 provider，不是交易所、策略或交易员名称；用户说“用 claw402”时应解释为选择/绑定 claw402 模型配置。", + "claw402 使用 Base 链 EVM 钱包 + USDC 按次付费；enabled=true 只代表模型配置已启用，不代表钱包一定有余额。", + "claw402 或 blockrun-base 钱包余额为 0 USDC 时，应明确提示“钱包余额不足/需要充值”，不要说成“模型未启用”或静默改用其他模型。", + "用户明确指定某个 provider 或模型时，如果当前不可用，必须先说明不可用原因，再让用户选择修复该模型或改用其他已可用模型；不得静默替换。", "删除操作不可逆，必须先向用户确认再执行。" ], "actions": { @@ -52,6 +56,7 @@ "确认 provider 后，先说明该 provider 的默认模型和凭证类型，再按 provider 特性补充追问。", "普通 provider（openai、deepseek、claude 等）通常需要 api_key；custom_api_url 和 custom_model_name 可留空走默认值。", "claw402 需要钱包私钥，不需要 custom_api_url；custom_model_name 留空时默认 deepseek。", + "创建 claw402 后若钱包余额为 0 USDC，应提示用户充值 Base USDC 后再用于稳定调用；不要把余额不足误报为配置未启用。", "blockrun-base 和 blockrun-sol 需要钱包私钥，不需要 custom_api_url；custom_model_name 留空时默认 auto。", "若用户提供了 custom_api_url，校验是否为合法 HTTPS 地址，不合法则提示修正。", "OpenAI 的 api_key 不以 sk- 开头时，提示用户检查 Key 格式。", @@ -68,6 +73,7 @@ "goal": "更新一个已有模型配置的指定字段，而不覆盖未提及字段。", "dynamic_rules": [ "只更新用户明确提到的字段，不要覆盖未提及的字段。", + "如果用户只是想给 trader 改用 claw402，不要在模型配置里误改显示名称；应把 claw402 作为 provider/model 选择处理。", "更新 custom_api_url 时校验 HTTPS 格式。", "更新 api_key 时对 OpenAI 校验 sk- 前缀。" ], @@ -79,7 +85,8 @@ "required_slots": ["target_ref", "enabled"], "goal": "切换模型配置的启用状态。", "dynamic_rules": [ - "启用前必须确保 api_key 和 custom_model_name 已经齐全；若 provider 有特殊规则，也要在提示中体现。" + "启用前必须确保 provider 对应凭证已经齐全；若 provider 有默认模型，custom_model_name 可按默认值处理。", + "启用 claw402 只校验钱包私钥等配置完整性；若钱包 0 USDC，应提示充值，但不要把它等同于 enabled=false。" ], "success_output": "返回 model_id，并明确告知该模型已启用或已禁用。", "failure_output": "明确指出目标模型不存在、缺少启用前必填项，或当前状态切换失败。" @@ -119,7 +126,8 @@ "description": "查询所有模型配置列表，包含 provider、名称、启用状态。", "goal": "列出当前用户可见的模型配置，便于后续选择或绑定。", "dynamic_rules": [ - "优先返回 provider、名称、启用状态，不返回 API Key 明文。" + "优先返回 provider、名称、启用状态，不返回 API Key 明文。", + "对于 claw402 / blockrun-base，若工具结果包含钱包地址或 USDC 余额，应用它解释支付状态；余额不足时要说“需要充值”，不要说“没配置”。" ], "success_output": "返回模型配置列表摘要。", "failure_output": "若列表为空，应明确告知当前没有模型配置。" @@ -129,7 +137,8 @@ "required_slots": ["target_ref"], "goal": "读取一个模型配置的详细信息。", "dynamic_rules": [ - "详情返回中只能暴露 API Key 是否存在，不得返回明文凭证。" + "详情返回中只能暴露 API Key/钱包私钥是否存在，不得返回明文凭证。", + "对于 claw402，应区分三种状态：配置未启用、钱包凭证缺失、钱包余额不足。" ], "success_output": "返回目标模型配置的详细摘要。", "failure_output": "明确指出目标模型不存在，或当前引用已经失效。" diff --git a/agent/skills/strategy_diagnosis.json b/agent/skills/strategy_diagnosis.json index 827185c6..7f69b98e 100644 --- a/agent/skills/strategy_diagnosis.json +++ b/agent/skills/strategy_diagnosis.json @@ -2,5 +2,17 @@ "name": "strategy_diagnosis", "kind": "diagnosis", "domain": "strategy", - "description": "当用户反馈策略未生效、策略输出异常、提示词或配置结果与预期不一致、策略执行表现异常时调用。适用于策略内容和执行效果相关的排障与解释。不用于创建、修改、删除、激活、复制或查询策略模板这类管理操作。" + "description": "当用户反馈策略未生效、候选币为空、策略输出异常、提示词或配置结果与预期不一致、AI 一直 hold/wait、策略执行表现异常时调用。适用于策略内容、候选币、风控边界和执行效果相关的排障与解释。不用于创建、修改、删除、激活、复制或查询策略模板这类管理操作。", + "capabilities": [ + "区分策略模板配置问题、交易员绑定问题、市场数据/候选币问题、AI 决策为 hold/wait、风控拦截和交易所下单失败", + "解释 AI 策略与网格策略的字段边界、页面范围和 System enforced 字段", + "指出策略模板不能直接运行，必须由交易员绑定后执行" + ], + "dynamic_rules": [ + "策略没生效时，先区分：只是策略模板未被交易员绑定，还是交易员已绑定但运行结果不符合预期。", + "若候选币为空，检查 source_type/static_coins/AI500/OI 榜单/排除币/量化数据开关，不要直接归因为模型问题。", + "若 AI 一直 hold/wait，先检查 min_confidence、min_risk_reward_ratio、提示词是否过于保守、行情是否满足入场条件，再判断是否需要放宽策略。", + "若交易员绑定了策略但没有下单，应与 trader_diagnosis 协作区分策略无信号、风控拦截和交易所下单失败。", + "策略模板本身不保存交易所、模型、扫描间隔或初始余额；这些问题应引导到 trader/model/exchange 相关诊断。" + ] } diff --git a/agent/skills/strategy_management.json b/agent/skills/strategy_management.json index 0497e248..b37322ba 100644 --- a/agent/skills/strategy_management.json +++ b/agent/skills/strategy_management.json @@ -383,7 +383,9 @@ "策略模板不能直接启动运行，只有绑定了该策略的交易员才能启动。", "策略模板创建成功后应出现在策略列表/策略页。", "创建策略模板时不要要求用户先绑定或添加交易所账户，也不要要求绑定 AI 模型；交易所和模型只属于 trader 创建、部署或启动流程。", + "策略是模板/规则，不保存交易所 API、模型 provider、钱包余额或扫描间隔；这些属于交易员、模型或交易所配置。", "如果用户只是创建或配置策略模板，不要把它升级成 trader 创建流程。", + "如果用户说“用某个策略创建/启动交易员”，不要新建同名策略；应先查找并绑定已有策略，只有用户明确要求新策略时才创建。", "删除操作不可逆，必须先向用户确认再执行。", "激活（activate）操作将该策略设为默认模板，不是启动运行。", "scan_interval_minutes、initial_balance、lighter_api_key_index 这类交易员/交易所边界值不属于策略本身，若用户在改策略时提到，应引导去对应 trader 或 exchange 配置。", @@ -401,6 +403,7 @@ "若用户只是要给 trader 绑定现有策略，应优先在父任务里补 strategy 槽位，而不是误开新的 create。", "若用户明确要求新建策略，至少先收齐名称；其他配置可继续追问或按默认值协助补齐。", "创建策略模板本身不需要交易所账户或 AI 模型；不要在 create 策略时询问用户是否已有交易所账户。", + "如果用户同时提到模型 provider、交易所账户或扫描间隔，应把这些信息留给 trader 创建流程，不要写入策略配置。", "只有当用户明确要求运行、部署、实盘、创建交易员或绑定到交易员时，才进入 trader 流程并收集交易所/模型。", "策略模板创建成功后应出现在策略列表/策略页。", "只有用户要运行或部署时，才继续 trader 绑定。", @@ -416,6 +419,7 @@ "goal": "更新一个已有策略模板的指定配置，而不覆盖未提及字段。", "dynamic_rules": [ "只更新用户明确提到的字段，不要覆盖未提及的字段。", + "如果用户想修改绑定的模型、交易所、扫描间隔或交易员运行状态，应转去 trader_management，不要写进策略。", "杠杆超出 1～20 范围时，自动收敛并告知用户。", "grid_trading 类型时，lower_price 必须小于 upper_price。" ], @@ -453,6 +457,7 @@ "dynamic_rules": [ "配置值超出手动面板边界时，应先自动收敛并明确告知用户。", "若用户一次提到多个配置 patch，可在同一轮内整体应用，但要明确说明最终修改了哪些字段。", + "拒绝把 exchange_id、ai_model_id、scan_interval_minutes、initial_balance、wallet/private key 等非策略字段作为策略 config 写入。", "当配置更新涉及 custom_prompt、role_definition、trading_frequency、entry_standards、decision_process、description、name 等文本槽位，且用户表达了“交给你”“你帮我写”“你自己设计”等委托生成意图时，严禁再次向用户索要正文。", "此时你必须直接生成一版可用文本，写入对应 extracted 字段，并用确认式问题向用户展示：“我先为你拟了一版……，要直接按这版更新吗？”" ], diff --git a/agent/skills/trader_diagnosis.json b/agent/skills/trader_diagnosis.json index 7adc673d..db3e5daa 100644 --- a/agent/skills/trader_diagnosis.json +++ b/agent/skills/trader_diagnosis.json @@ -11,6 +11,10 @@ "dynamic_rules": [ "当用户问“为什么报错”“为什么不交易”“为什么停了”这类问题时，优先走诊断而不是管理类 skill。", "如果已经能唯一确定目标交易员，应优先结合 get_backend_logs、持仓、状态和决策记录一起分析，而不是只看配置。", + "交易员不下单的排查顺序固定为：是否运行中 -> 是否已到扫描间隔 -> 策略候选币/行情数据是否为空 -> 最近 AI 决策是否为 hold/wait -> 风控是否拦截 -> 交易所下单是否报错 -> 余额、杠杆、仓位模式或权限是否限制。", + "如果模型是 claw402 或 blockrun-base，应单独检查钱包 USDC 余额；余额不足时应说“支付余额不足/需要充值”，不要泛化成“模型没启用”。", + "如果日志显示 AI 返回 hold/wait，应解释为模型判断当前没有足够交易信号，不应误判为系统没有运行。", + "如果日志显示下单失败，应优先归因到交易所权限、API 凭证、仓位模式、余额、杠杆或 symbol 可交易性，而不是策略没有生效。", "当用户表达“启动不了”“启动失败”“无法启动”“一启动就报错”“为什么启动不起来”这类启动故障时，只要目标交易员能唯一确定，就优先自动读取 get_backend_logs。", "当日志中已经出现明确错误原因时，直接用人话解释原因和下一步，不要只复述原始日志。" ], diff --git a/agent/skills/trader_management.json b/agent/skills/trader_management.json index 5efc6272..cdff2e11 100644 --- a/agent/skills/trader_management.json +++ b/agent/skills/trader_management.json @@ -62,6 +62,9 @@ "strategy_id 对应的策略模板必须存在，否则无法创建交易员。", "scan_interval_minutes 超出 3～60 范围时，系统自动收敛到边界值，并通过 LLM 告知用户已调整，询问是否接受。", "交易员初始余额由系统在创建时自动读取绑定交易所账户净值，不接受用户手动设置、充值或修改。", + "交易员名称不能从模型 provider 自动推断；用户说“用 claw402”表示模型选择，不表示交易员名称叫 claw402。", + "用户明确指定模型、交易所或策略时，若该资源不存在、被禁用、配置不完整或钱包余额不足，必须说明具体原因并让用户确认修复或替换；不得静默换成另一个资源。", + "若用户指定 claw402 作为模型，但 claw402 钱包余额为 0 USDC，应提示先充值或确认临时改用其他可用模型；不得说成 claw402 未启用，除非 enabled 确实为 false。", "启动交易员前，绑定的模型必须已启用且完整，绑定的交易所也必须已启用且通过对应交易所的完整性校验，否则拒绝启动并明确指出缺哪一项。", "若绑定的是 OKX 交易所，启用前必须已有 passphrase；若绑定的是 Hyperliquid，启用前必须已有 wallet_addr；若绑定的是 Aster，启用前必须已有 user、signer、private_key；若绑定的是 Lighter，启用前必须已有 wallet_addr 和 api_key_private_key。", "启动（start）和停止（stop）操作属于高风险操作，必须先向用户确认再执行。", @@ -75,6 +78,8 @@ "goal": "创建并初始化一个交易员。", "dynamic_rules": [ "若用户提到的交易所、模型或策略已经存在且可用，应优先直接补入对应槽位，不要重新创建。", + "如果用户明确指定某个模型 provider（如 claw402），应先尝试匹配该 provider 对应的模型配置；只有在说明原因并得到用户确认后，才可改用其他模型。", + "若用户没有提供交易员名称，应生成一个来自交易所/策略/方向的清晰名称，或向用户追问；不要把模型 provider、交易所类型或策略字段误用为交易员名称。", "若依赖资源不存在、被禁用，或用户明确要求新建或启用，禁止直接报缺字段；应切去对应 management:create 或 management:update_status 子任务。", "子任务成功后，系统会恢复当前交易员草稿并继续补齐剩余槽位。", "scan_interval_minutes 超出 3～60 时，自动收敛并告知用户。", @@ -91,7 +96,8 @@ "goal": "更新一个已有交易员的手动面板字段，但不改动策略、模型、交易所内部配置。", "dynamic_rules": [ "只更新用户明确提到的字段，不要覆盖未提及的字段。", - "换绑交易所/模型/策略时，新的资源必须已存在且已启用，否则提示用户先启用或新建。", + "换绑交易所/模型/策略时，新的资源必须已存在且已启用；若是钱包付费模型，还要解释余额不足等支付状态。", + "用户明确要求换成某个模型/交易所/策略时，不能自动选择另一个看起来可用的资源，除非用户确认。", "如果用户要求改名，应明确告知交易员改名不在这里处理。", "如果用户实际上是想修改策略参数、模型配置或交易所凭证，不要继续留在 trader update；应切到对应 management skill。" ], @@ -105,6 +111,7 @@ "goal": "调整交易员手动面板可编辑的字段，而不改动无关配置。", "dynamic_rules": [ "新绑定的资源必须已存在且已启用，否则提示用户先启用或新建。", + "当指定模型是 claw402 或 blockrun-base 且钱包余额不足时，应提示充值或让用户确认临时切换模型。", "扫描间隔超出 3～60 时，自动收敛并告知用户。" ], "success_output": "返回 trader_id，并明确展示新的模型/交易所/策略绑定结果。", @@ -135,7 +142,8 @@ "required_slots": ["target_ref", "ai_model_id"], "goal": "为指定交易员换绑一个 AI 模型配置。", "dynamic_rules": [ - "新的模型配置必须已启用且可调用，否则提示用户先启用或补齐模型配置。" + "新的模型配置必须已启用且可调用，否则提示用户先启用或补齐模型配置。", + "若用户指定的是 claw402，应优先绑定 claw402；只有在钱包余额不足、凭证缺失或配置不可用且用户确认后，才允许改绑其他模型。" ], "success_output": "返回 trader_id，并明确告知当前生效的 ai_model_id/模型名称。", "failure_output": "明确指出目标交易员或模型不存在，或模型当前不可用。" @@ -147,6 +155,7 @@ "goal": "让一个已配置好的交易员进入运行状态。", "dynamic_rules": [ "启动前系统会自动校验绑定的交易所、模型、策略是否均可用。", + "若绑定模型为 claw402 或 blockrun-base 且钱包余额不足，应提示充值或换模型；不要把它泛化成“模型不可用”。", "若校验失败，用人话告知用户具体哪个依赖不可用，并引导修复。" ], "success_output": "返回 trader_id，并明确告知交易员已开始运行。", diff --git a/agent/tools.go b/agent/tools.go index 26e77395..f80b1efc 100644 --- a/agent/tools.go +++ b/agent/tools.go @@ -938,6 +938,8 @@ type safeModelToolConfig struct { HasAPIKey bool `json:"has_api_key"` CustomAPIURL string `json:"custom_api_url,omitempty"` CustomModelName string `json:"custom_model_name,omitempty"` + WalletAddress string `json:"wallet_address,omitempty"` + BalanceUSDC string `json:"balance_usdc,omitempty"` } type safeTraderToolConfig struct { @@ -1115,7 +1117,7 @@ func extractTraderInitialBalance(balanceInfo map[string]interface{}) (float64, b } func safeModelForTool(model *store.AIModel) safeModelToolConfig { - return safeModelToolConfig{ + safeModel := safeModelToolConfig{ ID: model.ID, Name: model.Name, Provider: model.Provider, @@ -1124,6 +1126,18 @@ func safeModelForTool(model *store.AIModel) safeModelToolConfig { CustomAPIURL: model.CustomAPIURL, CustomModelName: model.CustomModelName, } + if agentProviderSupportsUSDCBalance(model.Provider) { + privateKey := strings.TrimSpace(string(model.APIKey)) + if privateKey != "" { + if walletAddress, err := agentWalletAddressFromPrivateKey(privateKey); err == nil && strings.TrimSpace(walletAddress) != "" { + safeModel.WalletAddress = walletAddress + if balance, balanceErr := agentQueryUSDCBalanceCached(walletAddress); balanceErr == nil { + safeModel.BalanceUSDC = fmt.Sprintf("%.6f", balance) + } + } + } + } + return safeModel } func modelConfigUsable(provider, modelID, apiKey, customAPIURL, customModelName string) bool { diff --git a/agent/web.go b/agent/web.go index ce290f65..e1571226 100644 --- a/agent/web.go +++ b/agent/web.go @@ -3,6 +3,7 @@ package agent import ( "context" "encoding/json" + "errors" "fmt" "io" "log/slog" @@ -165,13 +166,23 @@ func (w *WebHandler) HandleChatStream(rw http.ResponseWriter, r *http.Request) { defer cancel() resp, err := w.agent.HandleMessageStreamForStoreUser(ctx, storeUserIDFromContext(r.Context()), req.UserID, msg, func(event, data string) { + if ctx.Err() != nil { + return + } writeSSE(rw, flusher, event, data) }) if err != nil { + if errors.Is(err, context.Canceled) || errors.Is(err, context.DeadlineExceeded) || ctx.Err() != nil { + w.logger.Info("agent stream cancelled", "user_id", req.UserID, "error", err) + return + } w.logger.Error("agent HandleMessageStream failed", "error", err, "user_id", req.UserID) writeSSE(rw, flusher, "error", "I ran into a problem while handling that message. Please try again.") return } + if ctx.Err() != nil { + return + } // Send final done event with complete response writeSSE(rw, flusher, "done", resp) } diff --git a/web/src/components/agent/ChatInput.tsx b/web/src/components/agent/ChatInput.tsx index b2012cb5..6004796c 100644 --- a/web/src/components/agent/ChatInput.tsx +++ b/web/src/components/agent/ChatInput.tsx @@ -1,5 +1,12 @@ -import { useRef, useState, useCallback, useEffect, useImperativeHandle, forwardRef } from 'react' -import { ArrowUp } from 'lucide-react' +import { + useRef, + useState, + useCallback, + useEffect, + useImperativeHandle, + forwardRef, +} from 'react' +import { ArrowUp, Square } from 'lucide-react' export interface ChatInputHandle { focus: () => void @@ -10,43 +17,60 @@ export interface ChatInputHandle { interface ChatInputProps { language: string loading: boolean + value: string + onChange: (value: string) => void onSend: (text: string) => void + onStop: () => void } export const ChatInput = forwardRef( - function ChatInput({ language, loading, onSend }, ref) { - const [input, setInput] = useState('') + function ChatInput( + { language, loading, value, onChange, onSend, onStop }, + ref + ) { const [composing, setComposing] = useState(false) const inputRef = useRef(null) - useImperativeHandle(ref, () => ({ - focus: () => inputRef.current?.focus(), - clear: () => { - setInput('') - if (inputRef.current) inputRef.current.style.height = 'auto' - }, - getValue: () => input, - })) + useImperativeHandle( + ref, + () => ({ + focus: () => inputRef.current?.focus(), + clear: () => { + onChange('') + if (inputRef.current) inputRef.current.style.height = 'auto' + }, + getValue: () => value, + }), + [onChange, value] + ) + + const resizeInput = useCallback(() => { + const el = inputRef.current + if (!el) return + el.style.height = 'auto' + el.style.height = Math.min(el.scrollHeight, 150) + 'px' + }, []) const handleInputChange = useCallback( (e: React.ChangeEvent) => { - setInput(e.target.value) - const el = e.target - el.style.height = 'auto' - el.style.height = Math.min(el.scrollHeight, 150) + 'px' + onChange(e.target.value) }, - [] + [onChange] ) const handleSend = () => { - const msg = input.trim() + const msg = value.trim() if (!msg || loading) return - setInput('') + onChange('') if (inputRef.current) inputRef.current.style.height = 'auto' onSend(msg) inputRef.current?.focus() } + useEffect(() => { + resizeInput() + }, [resizeInput, value]) + // Keyboard shortcut: Cmd+K to focus useEffect(() => { const handleKeyDown = (e: KeyboardEvent) => { @@ -84,7 +108,7 @@ export const ChatInput = forwardRef( >