mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-04 03:21:04 +08:00
87 lines
3.1 KiB
Go
87 lines
3.1 KiB
Go
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
|
||
}
|