mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-10 22:36:58 +08:00
refactor(mcp) (#1042)
* improve(interface): replace with interface * feat(mcp): 添加构建器模式支持 新增功能： - RequestBuilder 构建器，支持流式 API - 多轮对话支持（AddAssistantMessage） - Function Calling / Tools 支持 - 精细参数控制（temperature, top_p, penalties 等） - 3个预设场景（Chat, CodeGen, CreativeWriting） - 完整的测试套件（19个新测试） 修复问题： - Config 字段未使用（MaxRetries、Temperature 等） - DeepSeek/Qwen SetAPIKey 的冗余 nil 检查 向后兼容： - 保留 CallWithMessages API - 新增 CallWithRequest API 测试： - 81 个测试全部通过 - 覆盖率 80.6% 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: tinkle-community <tinklefund@gmail.com> --------- Co-authored-by: zbhan <zbhan@freewheel.tv> Co-authored-by: tinkle-community <tinklefund@gmail.com>
This commit is contained in:
Shui
296
mcp/examples_test.go
Normal file
296
mcp/examples_test.go
Normal file
@@ -0,0 +1,296 @@
|
||||
package mcp_test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"net/http"
|
||||
"time"
|
||||
|
||||
"nofx/mcp"
|
||||
)
|
||||
|
||||
// ============================================================
|
||||
// 示例 1: 基础用法（向前兼容）
|
||||
// ============================================================
|
||||
|
||||
func Example_backward_compatible() {
|
||||
// ✅ 旧代码继续工作，无需修改
|
||||
client := mcp.New()
|
||||
client.SetAPIKey("sk-xxx", "https://api.custom.com", "gpt-4")
|
||||
|
||||
// 使用
|
||||
result, _ := client.CallWithMessages("system prompt", "user prompt")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
func Example_deepseek_backward_compatible() {
|
||||
// ✅ DeepSeek 旧代码继续工作
|
||||
client := mcp.NewDeepSeekClient()
|
||||
client.SetAPIKey("sk-xxx", "", "")
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 2: 新的推荐用法（选项模式）
|
||||
// ============================================================
|
||||
|
||||
func Example_new_client_basic() {
|
||||
// 使用默认配置
|
||||
client := mcp.NewClient()
|
||||
|
||||
// 使用 DeepSeek
|
||||
client = mcp.NewClient(
|
||||
mcp.WithDeepSeekConfig("sk-xxx"),
|
||||
)
|
||||
|
||||
// 使用 Qwen
|
||||
client = mcp.NewClient(
|
||||
mcp.WithQwenConfig("sk-xxx"),
|
||||
)
|
||||
|
||||
_ = client
|
||||
}
|
||||
|
||||
func Example_new_client_with_options() {
|
||||
// 组合多个选项
|
||||
client := mcp.NewClient(
|
||||
mcp.WithDeepSeekConfig("sk-xxx"),
|
||||
mcp.WithTimeout(60*time.Second),
|
||||
mcp.WithMaxRetries(5),
|
||||
mcp.WithMaxTokens(4000),
|
||||
mcp.WithTemperature(0.7),
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 3: 自定义日志器
|
||||
// ============================================================
|
||||
|
||||
// CustomLogger 自定义日志器示例
|
||||
type CustomLogger struct{}
|
||||
|
||||
func (l *CustomLogger) Debugf(format string, args ...any) {
|
||||
fmt.Printf("[DEBUG] "+format+"\n", args...)
|
||||
}
|
||||
|
||||
func (l *CustomLogger) Infof(format string, args ...any) {
|
||||
fmt.Printf("[INFO] "+format+"\n", args...)
|
||||
}
|
||||
|
||||
func (l *CustomLogger) Warnf(format string, args ...any) {
|
||||
fmt.Printf("[WARN] "+format+"\n", args...)
|
||||
}
|
||||
|
||||
func (l *CustomLogger) Errorf(format string, args ...any) {
|
||||
fmt.Printf("[ERROR] "+format+"\n", args...)
|
||||
}
|
||||
|
||||
func Example_custom_logger() {
|
||||
// 使用自定义日志器
|
||||
customLogger := &CustomLogger{}
|
||||
|
||||
client := mcp.NewClient(
|
||||
mcp.WithDeepSeekConfig("sk-xxx"),
|
||||
mcp.WithLogger(customLogger),
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
func Example_no_logger_for_testing() {
|
||||
// 测试时禁用日志
|
||||
client := mcp.NewClient(
|
||||
mcp.WithLogger(mcp.NewNoopLogger()),
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 4: 自定义 HTTP 客户端
|
||||
// ============================================================
|
||||
|
||||
func Example_custom_http_client() {
|
||||
// 自定义 HTTP 客户端（添加代理、TLS等）
|
||||
customHTTP := &http.Client{
|
||||
Timeout: 30 * time.Second,
|
||||
Transport: &http.Transport{
|
||||
Proxy: http.ProxyFromEnvironment,
|
||||
// 自定义 TLS、连接池等
|
||||
},
|
||||
}
|
||||
|
||||
client := mcp.NewClient(
|
||||
mcp.WithDeepSeekConfig("sk-xxx"),
|
||||
mcp.WithHTTPClient(customHTTP),
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 5: DeepSeek 客户端（新 API）
|
||||
// ============================================================
|
||||
|
||||
func Example_deepseek_new_api() {
|
||||
// 基础用法
|
||||
client := mcp.NewDeepSeekClientWithOptions(
|
||||
mcp.WithAPIKey("sk-xxx"),
|
||||
)
|
||||
|
||||
// 高级用法
|
||||
client = mcp.NewDeepSeekClientWithOptions(
|
||||
mcp.WithAPIKey("sk-xxx"),
|
||||
mcp.WithLogger(&CustomLogger{}),
|
||||
mcp.WithTimeout(90*time.Second),
|
||||
mcp.WithMaxTokens(8000),
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 6: Qwen 客户端（新 API）
|
||||
// ============================================================
|
||||
|
||||
func Example_qwen_new_api() {
|
||||
// 基础用法
|
||||
client := mcp.NewQwenClientWithOptions(
|
||||
mcp.WithAPIKey("sk-xxx"),
|
||||
)
|
||||
|
||||
// 高级用法
|
||||
client = mcp.NewQwenClientWithOptions(
|
||||
mcp.WithAPIKey("sk-xxx"),
|
||||
mcp.WithLogger(&CustomLogger{}),
|
||||
mcp.WithTimeout(90*time.Second),
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 7: 在 trader/auto_trader.go 中的迁移示例
|
||||
// ============================================================
|
||||
|
||||
func Example_trader_migration() {
|
||||
// === 旧代码（继续工作）===
|
||||
oldStyleClient := func(apiKey, customURL, customModel string) mcp.AIClient {
|
||||
client := mcp.NewDeepSeekClient()
|
||||
client.SetAPIKey(apiKey, customURL, customModel)
|
||||
return client
|
||||
}
|
||||
|
||||
// === 新代码（推荐）===
|
||||
newStyleClient := func(apiKey, customURL, customModel string) mcp.AIClient {
|
||||
opts := []mcp.ClientOption{
|
||||
mcp.WithAPIKey(apiKey),
|
||||
}
|
||||
|
||||
if customURL != "" {
|
||||
opts = append(opts, mcp.WithBaseURL(customURL))
|
||||
}
|
||||
|
||||
if customModel != "" {
|
||||
opts = append(opts, mcp.WithModel(customModel))
|
||||
}
|
||||
|
||||
return mcp.NewDeepSeekClientWithOptions(opts...)
|
||||
}
|
||||
|
||||
// 两种方式都能工作
|
||||
_ = oldStyleClient("sk-xxx", "", "")
|
||||
_ = newStyleClient("sk-xxx", "", "")
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 8: 测试场景
|
||||
// ============================================================
|
||||
|
||||
// MockHTTPClient Mock HTTP 客户端
|
||||
type MockHTTPClient struct {
|
||||
Response string
|
||||
}
|
||||
|
||||
func (m *MockHTTPClient) Do(req *http.Request) (*http.Response, error) {
|
||||
// 返回预设的响应
|
||||
return &http.Response{
|
||||
StatusCode: 200,
|
||||
Body: nil, // 实际测试中需要实现
|
||||
}, nil
|
||||
}
|
||||
|
||||
func Example_testing_with_mock() {
|
||||
// 测试时使用 Mock
|
||||
// mockHTTP := &MockHTTPClient{
|
||||
// Response: `{"choices":[{"message":{"content":"test response"}}]}`,
|
||||
// }
|
||||
|
||||
client := mcp.NewClient(
|
||||
// mcp.WithHTTPClient(mockHTTP), // 实际测试中使用 mockHTTP
|
||||
mcp.WithLogger(mcp.NewNoopLogger()), // 禁用日志
|
||||
)
|
||||
|
||||
result, _ := client.CallWithMessages("system", "user")
|
||||
fmt.Println(result)
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 9: 环境特定配置
|
||||
// ============================================================
|
||||
|
||||
func Example_environment_specific() {
|
||||
// 开发环境：详细日志
|
||||
devClient := mcp.NewClient(
|
||||
mcp.WithDeepSeekConfig("sk-xxx"),
|
||||
mcp.WithLogger(&CustomLogger{}), // 详细日志
|
||||
)
|
||||
|
||||
// 生产环境：结构化日志 + 超时保护
|
||||
prodClient := mcp.NewClient(
|
||||
mcp.WithDeepSeekConfig("sk-xxx"),
|
||||
// mcp.WithLogger(&ZapLogger{}), // 生产级日志
|
||||
mcp.WithTimeout(30*time.Second),
|
||||
mcp.WithMaxRetries(3),
|
||||
)
|
||||
|
||||
_, _ = devClient.CallWithMessages("system", "user")
|
||||
_, _ = prodClient.CallWithMessages("system", "user")
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// 示例 10: 完整实战示例
|
||||
// ============================================================
|
||||
|
||||
func Example_real_world_usage() {
|
||||
// 创建带有完整配置的客户端
|
||||
client := mcp.NewDeepSeekClientWithOptions(
|
||||
mcp.WithAPIKey("sk-xxxxxxxxxx"),
|
||||
mcp.WithTimeout(60*time.Second),
|
||||
mcp.WithMaxRetries(5),
|
||||
mcp.WithMaxTokens(4000),
|
||||
mcp.WithTemperature(0.5),
|
||||
mcp.WithLogger(&CustomLogger{}),
|
||||
)
|
||||
|
||||
// 使用客户端
|
||||
systemPrompt := "你是一个专业的量化交易顾问"
|
||||
userPrompt := "分析 BTC 当前走势"
|
||||
|
||||
result, err := client.CallWithMessages(systemPrompt, userPrompt)
|
||||
if err != nil {
|
||||
fmt.Printf("Error: %v\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("AI 响应: %s\n", result)
|
||||
}
|
||||
Reference in New Issue
Block a user