mirror of
https://github.com/laoxong/nofx.git
synced 2026-06-04 09:58:22 +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% 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
GitHub
@@ -0,0 +1,310 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// ============================================================
|
||||
// Mock Logger
|
||||
// ============================================================
|
||||
|
||||
// MockLogger Mock 日志器（用于测试）
|
||||
type MockLogger struct {
|
||||
mu sync.Mutex
|
||||
Logs []LogEntry
|
||||
Enabled bool // 是否启用日志记录
|
||||
}
|
||||
|
||||
// LogEntry 日志条目
|
||||
type LogEntry struct {
|
||||
Level string
|
||||
Format string
|
||||
Args []any
|
||||
Message string // 格式化后的消息
|
||||
}
|
||||
|
||||
func NewMockLogger() *MockLogger {
|
||||
return &MockLogger{
|
||||
Logs: make([]LogEntry, 0),
|
||||
Enabled: true,
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MockLogger) Debugf(format string, args ...any) {
|
||||
m.log("DEBUG", format, args...)
|
||||
}
|
||||
|
||||
func (m *MockLogger) Infof(format string, args ...any) {
|
||||
m.log("INFO", format, args...)
|
||||
}
|
||||
|
||||
func (m *MockLogger) Warnf(format string, args ...any) {
|
||||
m.log("WARN", format, args...)
|
||||
}
|
||||
|
||||
func (m *MockLogger) Errorf(format string, args ...any) {
|
||||
m.log("ERROR", format, args...)
|
||||
}
|
||||
|
||||
func (m *MockLogger) log(level, format string, args ...any) {
|
||||
if !m.Enabled {
|
||||
return
|
||||
}
|
||||
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
message := fmt.Sprintf(format, args...)
|
||||
m.Logs = append(m.Logs, LogEntry{
|
||||
Level: level,
|
||||
Format: format,
|
||||
Args: args,
|
||||
Message: message,
|
||||
})
|
||||
}
|
||||
|
||||
// GetLogs 获取所有日志
|
||||
func (m *MockLogger) GetLogs() []LogEntry {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
return append([]LogEntry{}, m.Logs...)
|
||||
}
|
||||
|
||||
// GetLogsByLevel 获取指定级别的日志
|
||||
func (m *MockLogger) GetLogsByLevel(level string) []LogEntry {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
var result []LogEntry
|
||||
for _, log := range m.Logs {
|
||||
if log.Level == level {
|
||||
result = append(result, log)
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
// Clear 清空日志
|
||||
func (m *MockLogger) Clear() {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
m.Logs = make([]LogEntry, 0)
|
||||
}
|
||||
|
||||
// HasLog 检查是否包含指定消息
|
||||
func (m *MockLogger) HasLog(level, message string) bool {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
for _, log := range m.Logs {
|
||||
if log.Level == level && log.Message == message {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Mock HTTP Client (实现 http.RoundTripper)
|
||||
// ============================================================
|
||||
|
||||
// MockHTTPClient Mock HTTP 客户端（实现 http.RoundTripper）
|
||||
type MockHTTPClient struct {
|
||||
mu sync.Mutex
|
||||
|
||||
// 配置
|
||||
Response string
|
||||
StatusCode int
|
||||
Error error
|
||||
ResponseFunc func(req *http.Request) (*http.Response, error) // 自定义响应函数
|
||||
|
||||
// 记录
|
||||
Requests []*http.Request
|
||||
}
|
||||
|
||||
func NewMockHTTPClient() *MockHTTPClient {
|
||||
return &MockHTTPClient{
|
||||
StatusCode: http.StatusOK,
|
||||
Requests: make([]*http.Request, 0),
|
||||
}
|
||||
}
|
||||
|
||||
// ToHTTPClient 转换为 http.Client
|
||||
func (m *MockHTTPClient) ToHTTPClient() *http.Client {
|
||||
return &http.Client{
|
||||
Transport: m,
|
||||
}
|
||||
}
|
||||
|
||||
// RoundTrip 实现 http.RoundTripper 接口
|
||||
func (m *MockHTTPClient) RoundTrip(req *http.Request) (*http.Response, error) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
// 记录请求
|
||||
m.Requests = append(m.Requests, req)
|
||||
|
||||
// 如果有自定义响应函数，使用它
|
||||
if m.ResponseFunc != nil {
|
||||
return m.ResponseFunc(req)
|
||||
}
|
||||
|
||||
// 如果设置了错误，返回错误
|
||||
if m.Error != nil {
|
||||
return nil, m.Error
|
||||
}
|
||||
|
||||
// 返回模拟响应
|
||||
resp := &http.Response{
|
||||
StatusCode: m.StatusCode,
|
||||
Body: io.NopCloser(bytes.NewBufferString(m.Response)),
|
||||
Header: make(http.Header),
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// GetRequests 获取所有请求
|
||||
func (m *MockHTTPClient) GetRequests() []*http.Request {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
return append([]*http.Request{}, m.Requests...)
|
||||
}
|
||||
|
||||
// GetLastRequest 获取最后一次请求
|
||||
func (m *MockHTTPClient) GetLastRequest() *http.Request {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
if len(m.Requests) == 0 {
|
||||
return nil
|
||||
}
|
||||
return m.Requests[len(m.Requests)-1]
|
||||
}
|
||||
|
||||
// Reset 重置状态
|
||||
func (m *MockHTTPClient) Reset() {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
m.Requests = make([]*http.Request, 0)
|
||||
}
|
||||
|
||||
// SetSuccessResponse 设置成功响应
|
||||
func (m *MockHTTPClient) SetSuccessResponse(content string) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
m.StatusCode = http.StatusOK
|
||||
m.Response = `{"choices":[{"message":{"content":"` + content + `"}}]}`
|
||||
m.Error = nil
|
||||
}
|
||||
|
||||
// SetErrorResponse 设置错误响应
|
||||
func (m *MockHTTPClient) SetErrorResponse(statusCode int, message string) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
m.StatusCode = statusCode
|
||||
m.Response = message
|
||||
m.Error = nil
|
||||
}
|
||||
|
||||
// SetNetworkError 设置网络错误
|
||||
func (m *MockHTTPClient) SetNetworkError(err error) {
|
||||
m.mu.Lock()
|
||||
defer m.mu.Unlock()
|
||||
|
||||
m.Error = err
|
||||
}
|
||||
|
||||
// ============================================================
|
||||
// Mock Client Hooks (用于测试钩子机制)
|
||||
// ============================================================
|
||||
|
||||
// MockClientHooks Mock 客户端钩子
|
||||
type MockClientHooks struct {
|
||||
BuildRequestBodyCalled int
|
||||
BuildUrlCalled int
|
||||
SetAuthHeaderCalled int
|
||||
MarshalRequestCalled int
|
||||
ParseResponseCalled int
|
||||
IsRetryableErrorCalled int
|
||||
|
||||
// 自定义返回值
|
||||
BuildUrlFunc func() string
|
||||
ParseResponseFunc func([]byte) (string, error)
|
||||
IsRetryableErrorFunc func(error) bool
|
||||
BuildRequestBodyFunc func(string, string) map[string]any
|
||||
MarshalRequestBodyFunc func(map[string]any) ([]byte, error)
|
||||
}
|
||||
|
||||
func NewMockClientHooks() *MockClientHooks {
|
||||
return &MockClientHooks{}
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) buildMCPRequestBody(systemPrompt, userPrompt string) map[string]any {
|
||||
m.BuildRequestBodyCalled++
|
||||
if m.BuildRequestBodyFunc != nil {
|
||||
return m.BuildRequestBodyFunc(systemPrompt, userPrompt)
|
||||
}
|
||||
return map[string]any{
|
||||
"model": "test-model",
|
||||
"messages": []map[string]string{
|
||||
{"role": "system", "content": systemPrompt},
|
||||
{"role": "user", "content": userPrompt},
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) buildUrl() string {
|
||||
m.BuildUrlCalled++
|
||||
if m.BuildUrlFunc != nil {
|
||||
return m.BuildUrlFunc()
|
||||
}
|
||||
return "https://api.test.com/chat/completions"
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) setAuthHeader(headers http.Header) {
|
||||
m.SetAuthHeaderCalled++
|
||||
headers.Set("Authorization", "Bearer test-key")
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) marshalRequestBody(body map[string]any) ([]byte, error) {
|
||||
m.MarshalRequestCalled++
|
||||
if m.MarshalRequestBodyFunc != nil {
|
||||
return m.MarshalRequestBodyFunc(body)
|
||||
}
|
||||
return json.Marshal(body)
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) parseMCPResponse(body []byte) (string, error) {
|
||||
m.ParseResponseCalled++
|
||||
if m.ParseResponseFunc != nil {
|
||||
return m.ParseResponseFunc(body)
|
||||
}
|
||||
return "mocked response", nil
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) isRetryableError(err error) bool {
|
||||
m.IsRetryableErrorCalled++
|
||||
if m.IsRetryableErrorFunc != nil {
|
||||
return m.IsRetryableErrorFunc(err)
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) buildRequest(url string, jsonData []byte) (*http.Request, error) {
|
||||
req, _ := http.NewRequest("POST", url, bytes.NewBuffer(jsonData))
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
m.setAuthHeader(req.Header)
|
||||
return req, nil
|
||||
}
|
||||
|
||||
func (m *MockClientHooks) call(systemPrompt, userPrompt string) (string, error) {
|
||||
return "mocked call result", nil
|
||||
}
|
||||
Reference in New Issue
Block a user