diff --git a/internal/ai/client_test.go b/internal/ai/client_test.go new file mode 100644 index 0000000..ba190d7 --- /dev/null +++ b/internal/ai/client_test.go @@ -0,0 +1,25 @@ +package ai + +import ( + "testing" +) + +func TestGenerate(t *testing.T) { + client := NewClient(Config{ + Endpoint: `https://api.siliconflow.cn/v1`, + Model: `Pro/deepseek-ai/DeepSeek-V3.2`, + Timeout: 120, + MaxRetries: 3, + ApiKey: `sk-rrizhcztbcpqvzszpsosxbylpuaajjropmzrcdzlqynhopzm`, + }) + ans, err := client.Generate("你好") + if err != nil { + t.Fatalf("%v", err) + } + if ans == "" { + t.Fatalf("无输出") + } + + t.Log(ans) + +}