mirror of
https://github.com/NoFxAiOS/nofx.git
synced 2026-07-14 00:07:01 +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:
@@ -1,7 +1,6 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"log"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
@@ -15,36 +14,67 @@ type DeepSeekClient struct {
|
||||
*Client
|
||||
}
|
||||
|
||||
// NewDeepSeekClient 创建 DeepSeek 客户端(向前兼容)
|
||||
//
|
||||
// Deprecated: 推荐使用 NewDeepSeekClientWithOptions 以获得更好的灵活性
|
||||
func NewDeepSeekClient() AIClient {
|
||||
client := New().(*Client)
|
||||
client.Provider = ProviderDeepSeek
|
||||
client.Model = DefaultDeepSeekModel
|
||||
client.BaseURL = DefaultDeepSeekBaseURL
|
||||
return &DeepSeekClient{
|
||||
Client: client,
|
||||
return NewDeepSeekClientWithOptions()
|
||||
}
|
||||
|
||||
// NewDeepSeekClientWithOptions 创建 DeepSeek 客户端(支持选项模式)
|
||||
//
|
||||
// 使用示例:
|
||||
// // 基础用法
|
||||
// client := mcp.NewDeepSeekClientWithOptions()
|
||||
//
|
||||
// // 自定义配置
|
||||
// client := mcp.NewDeepSeekClientWithOptions(
|
||||
// mcp.WithAPIKey("sk-xxx"),
|
||||
// mcp.WithLogger(customLogger),
|
||||
// mcp.WithTimeout(60*time.Second),
|
||||
// )
|
||||
func NewDeepSeekClientWithOptions(opts ...ClientOption) AIClient {
|
||||
// 1. 创建 DeepSeek 预设选项
|
||||
deepseekOpts := []ClientOption{
|
||||
WithProvider(ProviderDeepSeek),
|
||||
WithModel(DefaultDeepSeekModel),
|
||||
WithBaseURL(DefaultDeepSeekBaseURL),
|
||||
}
|
||||
|
||||
// 2. 合并用户选项(用户选项优先级更高)
|
||||
allOpts := append(deepseekOpts, opts...)
|
||||
|
||||
// 3. 创建基础客户端
|
||||
baseClient := NewClient(allOpts...).(*Client)
|
||||
|
||||
// 4. 创建 DeepSeek 客户端
|
||||
dsClient := &DeepSeekClient{
|
||||
Client: baseClient,
|
||||
}
|
||||
|
||||
// 5. 设置 hooks 指向 DeepSeekClient(实现动态分派)
|
||||
baseClient.hooks = dsClient
|
||||
|
||||
return dsClient
|
||||
}
|
||||
|
||||
func (dsClient *DeepSeekClient) SetAPIKey(apiKey string, customURL string, customModel string) {
|
||||
if dsClient.Client == nil {
|
||||
dsClient.Client = New().(*Client)
|
||||
}
|
||||
dsClient.Client.APIKey = apiKey
|
||||
dsClient.APIKey = apiKey
|
||||
|
||||
if len(apiKey) > 8 {
|
||||
log.Printf("🔧 [MCP] DeepSeek API Key: %s...%s", apiKey[:4], apiKey[len(apiKey)-4:])
|
||||
dsClient.logger.Infof("🔧 [MCP] DeepSeek API Key: %s...%s", apiKey[:4], apiKey[len(apiKey)-4:])
|
||||
}
|
||||
if customURL != "" {
|
||||
dsClient.Client.BaseURL = customURL
|
||||
log.Printf("🔧 [MCP] DeepSeek 使用自定义 BaseURL: %s", customURL)
|
||||
dsClient.BaseURL = customURL
|
||||
dsClient.logger.Infof("🔧 [MCP] DeepSeek 使用自定义 BaseURL: %s", customURL)
|
||||
} else {
|
||||
log.Printf("🔧 [MCP] DeepSeek 使用默认 BaseURL: %s", dsClient.Client.BaseURL)
|
||||
dsClient.logger.Infof("🔧 [MCP] DeepSeek 使用默认 BaseURL: %s", dsClient.BaseURL)
|
||||
}
|
||||
if customModel != "" {
|
||||
dsClient.Client.Model = customModel
|
||||
log.Printf("🔧 [MCP] DeepSeek 使用自定义 Model: %s", customModel)
|
||||
dsClient.Model = customModel
|
||||
dsClient.logger.Infof("🔧 [MCP] DeepSeek 使用自定义 Model: %s", customModel)
|
||||
} else {
|
||||
log.Printf("🔧 [MCP] DeepSeek 使用默认 Model: %s", dsClient.Client.Model)
|
||||
dsClient.logger.Infof("🔧 [MCP] DeepSeek 使用默认 Model: %s", dsClient.Model)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user