mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-06-06 14:01:29 +08:00
76 lines
2.2 KiB
Go
76 lines
2.2 KiB
Go
package agent
|
||
|
||
import (
|
||
"log/slog"
|
||
"path/filepath"
|
||
"strings"
|
||
"testing"
|
||
|
||
"nofx/store"
|
||
)
|
||
|
||
func TestHandleModelCreateSkillAsksProviderFirstWithClaw402Recommendation(t *testing.T) {
|
||
dbPath := filepath.Join(t.TempDir(), "agent-model-create.db")
|
||
st, err := store.New(dbPath)
|
||
if err != nil {
|
||
t.Fatalf("create store: %v", err)
|
||
}
|
||
|
||
a := New(nil, st, DefaultConfig(), slog.Default())
|
||
reply := a.handleModelCreateSkill("default", 42, "zh", "请帮我创建一个模型", skillSession{})
|
||
|
||
for _, want := range []string{
|
||
"还缺这些字段：模型提供商",
|
||
"可选模型 provider",
|
||
"推荐 `claw402`",
|
||
"并列可选",
|
||
"按次付费",
|
||
"Base USDC 钱包支付",
|
||
"直接创建 Base 钱包",
|
||
"直接扫码充值/支付",
|
||
} {
|
||
if !strings.Contains(reply, want) {
|
||
t.Fatalf("expected reply to contain %q, got: %s", want, reply)
|
||
}
|
||
}
|
||
for _, unexpected := range []string{
|
||
"还缺这些字段：模型提供商、API Key",
|
||
"还缺这些字段：模型提供商、钱包私钥",
|
||
"还缺这些字段：模型提供商、wallet private key",
|
||
} {
|
||
if strings.Contains(reply, unexpected) {
|
||
t.Fatalf("provider-first reply should not ask for credentials yet: %s", reply)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestHandleModelCreateSkillUsesCollectedClaw402PrivateKey(t *testing.T) {
|
||
dbPath := filepath.Join(t.TempDir(), "agent-model-create-claw402.db")
|
||
st, err := store.New(dbPath)
|
||
if err != nil {
|
||
t.Fatalf("create store: %v", err)
|
||
}
|
||
|
||
a := New(nil, st, DefaultConfig(), slog.Default())
|
||
session := skillSession{
|
||
Name: "model_management",
|
||
Action: "create",
|
||
Phase: "collecting",
|
||
Fields: map[string]string{
|
||
"provider": "claw402",
|
||
"name": "Claw402 (Base USDC)",
|
||
"api_key": "0x205d759b80bae1afa31a36c4afaeec0b10378c1c55e3363bcde5a1db75c747ca",
|
||
"custom_model_name": "deepseek",
|
||
},
|
||
}
|
||
|
||
reply := a.handleModelCreateSkill("default", 42, "zh", "继续", session)
|
||
|
||
if strings.Contains(reply, "还缺这些字段：钱包私钥") {
|
||
t.Fatalf("expected bare private key to be accepted, got: %s", reply)
|
||
}
|
||
if !strings.Contains(reply, "我先整理了一份模型配置草稿") {
|
||
t.Fatalf("expected draft summary after accepting private key, got: %s", reply)
|
||
}
|
||
}
|