Blizzard
4928ffc0f7
会话历史(Redis,易失,与长期画像分开)经 MCP 工具进出 Eino 图：
recall 召回历史填 MessagesPlaceholder，写回把本轮 user/assistant 落历史。
- mcp-go: internal/history(go-redis, sundynix:history:<session>, LPUSH+LTRIM 保留近20条,
24h TTL) + 工具 history_get(返回JSON turns)/history_append; main 开 Redis(降级)
- dispatcher Eino: 模板加 MessagesPlaceholder('history'); recall 调 history_get→转 schema.Message;
Handle 累积 answer; memorize 异步 history_append(user+assistant)
- shared: contract.MetaSessionID; gateway: SubmitTask 注入 Meta[session_id](X-Session-ID 头,缺省 default)
- demo.sh: 同会话两轮提交，验证第2轮召回第1轮历史
- 验证: 4 模块 build✓ + 3 e2e PASS; live 跑通——轮1=0轮历史→落库, 轮2 history_get 命中→注入
Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
74 lines
3.3 KiB
Go
74 lines
3.3 KiB
Go
// Package contract 是 Gateway / Dispatcher / MCP 之间的共享契约：
|
|
// Task 数据结构与 NATS subject 命名约定。
|
|
package contract
|
|
|
|
import "encoding/json"
|
|
|
|
// NATS subject / stream 约定（与 README、各服务 config 保持一致）。
|
|
const (
|
|
StreamTasks = "SUNDYNIX_TASKS" // JetStream stream 名
|
|
SubjectTasks = "sundynix.tasks" // 任务发布主题前缀；实际为 sundynix.tasks.<id>
|
|
SubjectTasksAll = "sundynix.tasks.>" // stream 捕获的通配
|
|
SubjectStream = "sundynix.streams" // Token 回流前缀；实际 sundynix.streams.<id>
|
|
ConsumerDurable = "dispatchers" // Dispatcher 持久消费者（队列组负载均衡）
|
|
|
|
// HeaderStreamEnd 是 Token 流的结束信号（core NATS 消息头）。
|
|
// 置为 "1" 的消息体为空，表示该 task 的 Token 流结束。
|
|
HeaderStreamEnd = "X-Stream-End"
|
|
|
|
// MCP 工具调用约定（第 4 层 Dispatcher → 第 5 层 MCP Tools）。
|
|
// 用 core NATS request-reply：同步拿结果，队列组内负载均衡。
|
|
SubjectToolsGo = "sundynix.tools.go" // Go I/O 型工具前缀；实际 sundynix.tools.go.<tool>
|
|
SubjectToolsGoAll = "sundynix.tools.go.>" // mcp-go 通配订阅
|
|
SubjectToolsPy = "sundynix.tools.py" // Python 算法型工具前缀；实际 sundynix.tools.py.<tool>
|
|
SubjectToolsPyAll = "sundynix.tools.py.>" // mcp-py 通配订阅
|
|
QueueToolsGo = "mcp-go-workers" // mcp-go 队列组（多副本负载均衡）
|
|
QueueToolsPy = "mcp-py-workers" // mcp-py 队列组
|
|
|
|
// MetaUserID 是 Task.Meta 中承载已登录用户标识的键（用于偏好记忆召回）。
|
|
MetaUserID = "user_id"
|
|
// MetaSessionID 是 Task.Meta 中承载会话标识的键（用于短期多轮历史）。
|
|
MetaSessionID = "session_id"
|
|
)
|
|
|
|
// Task 是 DSL 解析组装后的可调度任务，在 NATS 上以 JSON 传输。
|
|
type Task struct {
|
|
ID string `json:"id"`
|
|
Graph json.RawMessage `json:"graph"` // React Flow 导出的 Agent 编排图
|
|
Meta map[string]any `json:"meta,omitempty"`
|
|
}
|
|
|
|
// TaskSubject 返回某任务的发布主题。
|
|
func TaskSubject(id string) string { return SubjectTasks + "." + id }
|
|
|
|
// StreamSubject 返回某任务的 Token 回流主题。
|
|
func StreamSubject(id string) string { return SubjectStream + "." + id }
|
|
|
|
// ToolSubjectGo / ToolSubjectPy 返回某工具的调用主题。
|
|
func ToolSubjectGo(tool string) string { return SubjectToolsGo + "." + tool }
|
|
func ToolSubjectPy(tool string) string { return SubjectToolsPy + "." + tool }
|
|
|
|
// ToolCall 是 Dispatcher 对一个 MCP 工具的调用请求（NATS request 体）。
|
|
type ToolCall struct {
|
|
Tool string `json:"tool"` // 工具名，如 wiki_search
|
|
Args map[string]any `json:"args,omitempty"` // 工具参数
|
|
TaskID string `json:"task_id,omitempty"` // 触发该调用的任务（便于追踪）
|
|
}
|
|
|
|
// ToolResult 是 MCP 工具的应答（NATS reply 体）。
|
|
type ToolResult struct {
|
|
OK bool `json:"ok"`
|
|
Content string `json:"content,omitempty"` // 工具产出（如检索结果文本）
|
|
Error string `json:"error,omitempty"` // 非空表示工具内部出错
|
|
}
|
|
|
|
// Marshal / Unmarshal 便捷方法。
|
|
func (t *Task) Marshal() ([]byte, error) { return json.Marshal(t) }
|
|
func Unmarshal(b []byte) (*Task, error) {
|
|
var t Task
|
|
if err := json.Unmarshal(b, &t); err != nil {
|
|
return nil, err
|
|
}
|
|
return &t, nil
|
|
}
|