Improve NOFXi agent product handling

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
+}