Version: 0.4.4.dev.260307
feat: 🚀 增强会话管理与缓存机制 * 会话 ID 空值兜底,若 `conversation_id` 为空时自动生成 UUID * 在响应头写入 `X-Conversation-ID`,供前端使用,保持同一会话状态 perf: ⚡ 会话状态缓存优化 * 当缓存未命中但 DB 已确认/创建会话后,调用 `SetConversationStatus` 回写 Redis * 缓存写回失败时记录日志,不中断聊天主流程,确保业务流畅性 fix: 🐛 修复历史消息顺序问题与编译错误 * 修复历史消息顺序问题,保证返回的 N 条历史消息按时间正序喂给模型 * 通过反转 `created_at desc` 查询结果的切片,确保模型输入顺序正确 * 修复 `fmt.Errorf` 参数不匹配问题,修正编译错误 * 整理 `agent-cache.go` 为标准 UTF-8 编码,避免 Go 编译报错 `invalid UTF-8 encoding` feat: 🛠️ 独立构建 MCP 服务器 * 使用 `Codex` 构建独立于后端的 MCP 服务器,简化与 Codex 的协作 * 通过该服务器方便 Codex 直接测试和查看 Redis 与 MySQL 中的数据
This commit is contained in:
56
infra/smartflow-mcp-server/internal/mcp/server_test.go
Normal file
56
infra/smartflow-mcp-server/internal/mcp/server_test.go
Normal file
@@ -0,0 +1,56 @@
|
||||
package mcp
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"testing"
|
||||
"time"
|
||||
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/ratelimit"
|
||||
"github.com/LoveLosita/smartflow/infra/smartflow-mcp-server/internal/tools"
|
||||
)
|
||||
|
||||
type dummyTool struct{}
|
||||
|
||||
func (d dummyTool) Name() string { return "dummy" }
|
||||
func (d dummyTool) Description() string { return "dummy" }
|
||||
func (d dummyTool) InputSchema() map[string]any {
|
||||
return map[string]any{"type": "object"}
|
||||
}
|
||||
func (d dummyTool) Execute(ctx context.Context, args map[string]any) (map[string]any, error) {
|
||||
return map[string]any{"ok": true}, nil
|
||||
}
|
||||
|
||||
func TestHandleInitialize(t *testing.T) {
|
||||
registry, err := tools.NewRegistry(dummyTool{})
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
out := bytes.NewBuffer(nil)
|
||||
s := NewServer(bytes.NewBuffer(nil), out, registry, nil, ratelimit.New(10, 10), "name", "ver", "2024-11-05", "unknown", time.Second)
|
||||
|
||||
req := request{JSONRPC: jsonRPCVersion, ID: json.RawMessage("1"), Method: "initialize", Params: json.RawMessage(`{"protocolVersion":"2024-11-05"}`)}
|
||||
if err := s.handleRequest(context.Background(), req); err != nil {
|
||||
t.Fatalf("handle initialize error: %v", err)
|
||||
}
|
||||
if !bytes.Contains(out.Bytes(), []byte("protocolVersion")) {
|
||||
t.Fatalf("response missing protocolVersion: %s", out.String())
|
||||
}
|
||||
}
|
||||
|
||||
func TestReadMessage(t *testing.T) {
|
||||
body := []byte(`{"jsonrpc":"2.0","id":1,"method":"ping"}`)
|
||||
frame := fmt.Sprintf("Content-Length: %d\r\n\r\n%s", len(body), string(body))
|
||||
input := bufio.NewReader(bytes.NewBufferString(frame))
|
||||
|
||||
msg, err := readMessage(input)
|
||||
if err != nil {
|
||||
t.Fatalf("read message failed: %v", err)
|
||||
}
|
||||
if !bytes.Equal(msg, body) {
|
||||
t.Fatalf("unexpected body: %s", string(msg))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user