155 lines
6.6 KiB
Go
155 lines
6.6 KiB
Go
package gateway
|
||
|
||
import (
|
||
"context"
|
||
"errors"
|
||
"log/slog"
|
||
"net/http"
|
||
|
||
apperr "git.yeij.top/AskaEth/Cyrene_v2/internal/errors"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/history"
|
||
"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/loop"
|
||
)
|
||
|
||
// errPersonaBindingUnsupported 由 newSubSession 在调用方要求一个
|
||
// 绑定角色的会话、但注入的 history 存储无法绑定角色时返回。
|
||
// handler 把它映射为 501，而不是假装绑定已生效。
|
||
var errPersonaBindingUnsupported = errors.New("gateway: history store does not support session persona binding")
|
||
|
||
// 本文件拥有网关的回合环境解析（DevDocs/design/22
|
||
// §2/§4.5）：由哪个角色应答，表达为该回合运行的
|
||
// loop.SessionEnv 环境。
|
||
//
|
||
// 优先级（design/22 §2，自上而下）：
|
||
// 1. 请求级覆盖 —— 本批次不实现（批次 1 暴露的是
|
||
// 会话级绑定；按请求覆盖是加性且不持久化的）；
|
||
// 2. 会话级绑定 —— chat_sessions.persona_id，在此解析；
|
||
// 3. 渠道级映射 —— 批次 2（channels.json / channel_sessions）；
|
||
// 4. 全局默认 —— 表达为 PersonaID == ""，由 loop 解析到其
|
||
// 已配置的默认角色（Options.PersonaID）。这正是让每个
|
||
// 既有会话的行为与以往完全一致的原因。
|
||
//
|
||
// env 的其他字段不受影响：对这些面向 owner 的 Web/语音路径，
|
||
// ActorEnv 与 MemWritable 仍来自 loop.OwnerEnv()。这就是身份红线
|
||
// —— 谁可以做什么由 channel 的 actor 元数据决定（此处是一个 JWT
|
||
// owner），绝不是由所扮演的角色决定。人格选择只改变
|
||
// 提示词、记忆命名空间、情绪机与思考预算。
|
||
|
||
// sessionPersonaReader 是 HistoryStore 可选的更窄视图，用于读取
|
||
// 会话的绑定角色。*history.Store 实现了它；未实现的存储（测试
|
||
// 替身，或早于 design/22 的存储）只是得到 "" —— 全局默认 —— 因此
|
||
// 网关继续工作。
|
||
type sessionPersonaReader interface {
|
||
SessionPersona(ctx context.Context, sessionID string) (string, error)
|
||
}
|
||
|
||
// sessionPersonaWriter 是 HistoryStore 可选的更窄视图，用于写入
|
||
// 会话的角色绑定。会话角色在创建时决定（design/22 §8-1），
|
||
// 因此不再有任何请求路径重绑 —— 创建走 subSessionCreator ——
|
||
// 而网关也不再对该接口做类型断言。它留在读取器
|
||
// 与创建器旁边，使存储的角色绑定表面集中记录在一处：
|
||
// *history.Store 仍实现 SetSessionPersona（design/22 §4.1），而会用它的
|
||
// 管理/owner 写路径是独立端点，不是 PATCH。
|
||
type sessionPersonaWriter interface {
|
||
SetSessionPersona(ctx context.Context, sessionID, personaID string) error
|
||
}
|
||
|
||
// subSessionCreator 是 HistoryStore 可选的更窄视图，用于创建一个
|
||
// 已绑定角色的会话（POST /sessions {personaId}，design/22 §4.5）。
|
||
// *history.Store 实现了它；没有它的存储只能创建未绑定的
|
||
// 会话，因此显式要求角色的请求以 501 拒绝。
|
||
type subSessionCreator interface {
|
||
NewSubSessionFor(ctx context.Context, ownerID, personaID string) (string, error)
|
||
}
|
||
|
||
// turnEnvResolver 为一个回合构建 loop 会话环境。它持有
|
||
// 解析所需的两个只读依赖，因此 HTTP/WS handler
|
||
// 与 hub 都经同一条代码路径解析角色（design/22 §4.5：一个
|
||
// 解析器，而不是每个入口各硬编码一份 loop.OwnerEnv()）。
|
||
type turnEnvResolver struct {
|
||
history HistoryStore
|
||
personas PersonaStore
|
||
log *slog.Logger
|
||
}
|
||
|
||
// owner 返回 sessionID 一个面向 owner 回合的每回合环境：
|
||
// 全访问 owner env（身份不变）加上该会话绑定的角色。
|
||
// 空白角色意味着「继承全局默认」；loop 执行那最后一次
|
||
// 回退，因此网关永远不必知道默认角色的 id。
|
||
//
|
||
// 构造走 loop.OwnerEnvFor —— 与 IM 的主人私聊车道**同一个入口**
|
||
// （channelbridge/transcript.go 的 turnEnv）：两条车道的 runEnv 因此逐字段同源，
|
||
// SessionEnv 将来新增字段时会一起拿到，而不是各自漂移。
|
||
func (t turnEnvResolver) owner(ctx context.Context, sessionID string) loop.SessionEnv {
|
||
return loop.OwnerEnvFor(t.sessionPersona(ctx, sessionID))
|
||
}
|
||
|
||
// sessionPersona 返回绑定到 sessionID 的角色，会话继承全局
|
||
// 默认时返回 ""。未知会话、没有读取器的存储、或
|
||
// 读错误都退化为 ""（默认），而不是让该回合失败：
|
||
// 聊天绝不能被一次元数据查找阻塞，而该回退正是
|
||
// 多角色之前的行为。
|
||
func (t turnEnvResolver) sessionPersona(ctx context.Context, sessionID string) string {
|
||
if t.history == nil || sessionID == "" {
|
||
return ""
|
||
}
|
||
reader, ok := t.history.(sessionPersonaReader)
|
||
if !ok {
|
||
return ""
|
||
}
|
||
personaID, err := reader.SessionPersona(ctx, sessionID)
|
||
if err != nil {
|
||
if t.log != nil && !errors.Is(err, history.ErrSessionNotFound) {
|
||
t.log.Debug("session persona read", "err", err, "session", sessionID)
|
||
}
|
||
return ""
|
||
}
|
||
return personaID
|
||
}
|
||
|
||
// defaultPersonaID 返回库中默认角色的人格 id，未接线人格
|
||
// 存储或没有角色被标记为默认时返回 ""。它是
|
||
// owner.yaml:defaultPersona 的网关侧视图，用于在没有会话时
|
||
// 需要角色 id 的场合（管理记忆搜索回退，design/22 §4.5）。
|
||
func (s *Server) defaultPersonaID() string {
|
||
if s.personas == nil {
|
||
return ""
|
||
}
|
||
for _, c := range s.personas.List() {
|
||
if c.Default && c.PersonaID != "" {
|
||
return c.PersonaID
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// requireKnownPersona 拒绝注册表不认识的 personaId，返回
|
||
// 400 并返回 false。空 id 一律允许（它意味着「继承
|
||
// 全局默认」），而没有 persona 存储的网关无法校验，因此
|
||
// 那里接受该绑定，之后只是解析为未知角色。
|
||
func (s *Server) requireKnownPersona(w http.ResponseWriter, personaID string) bool {
|
||
if personaID == "" || s.personas == nil {
|
||
return true
|
||
}
|
||
if _, ok := s.personas.Persona(personaID); !ok {
|
||
writeError(w, apperr.Wrap(apperr.ErrBadRequest, "bad_request", "unknown personaId %q", personaID))
|
||
return false
|
||
}
|
||
return true
|
||
}
|
||
|
||
// newSubSession 创建一个绑定到 personaID 的子会话，存储提供
|
||
// 角色感知创建器时使用它。没有它的存储只能创建
|
||
// 未绑定会话，因此显式的角色请求被拒绝（501），而不是
|
||
// 静默丢弃该绑定。
|
||
func (s *Server) newSubSession(ctx context.Context, ownerID, personaID string) (string, error) {
|
||
creator, ok := s.history.(subSessionCreator)
|
||
if !ok {
|
||
if personaID != "" {
|
||
return "", errPersonaBindingUnsupported
|
||
}
|
||
return s.history.NewSubSession(ctx, ownerID)
|
||
}
|
||
return creator.NewSubSessionFor(ctx, ownerID, personaID)
|
||
}
|