weichen
30ad92730d
Details (original message): Integrate the agent-intent gate: plugin intent overlays rebuild the registry (fail-closed, keeps previous snapshot), HITL free-input pivots on matched foreign intents; unit and e2e tests; pin logging-system and error-code-system submodules.
96 lines
2.4 KiB
Go
96 lines
2.4 KiB
Go
// Package intentgate 接入 agent-intent（生态仓）：插件意图 overlay → 注册表 →
|
||
// HITL 自由输入路由的确定性识别（设计 4.5 三分支之「命中其他意图→转轨」）。
|
||
// 未配置/未装载意图 = 行为回退（T0：直接按澄清回填），识别失败不阻断。
|
||
package intentgate
|
||
|
||
import (
|
||
"context"
|
||
"encoding/json"
|
||
"fmt"
|
||
"sync"
|
||
|
||
agentintent "agent-intent/intent"
|
||
)
|
||
|
||
type Gate struct {
|
||
mu sync.RWMutex
|
||
reg *agentintent.Registry
|
||
rec *agentintent.FunnelRecognizer
|
||
specs int
|
||
}
|
||
|
||
func New() *Gate {
|
||
return &Gate{rec: agentintent.NewRecognizer(agentintent.Options{}, nil, nil)}
|
||
}
|
||
|
||
// Rebuild 用插件意图 overlay 重建注册表；失败保留上一快照（fail-closed）并返回错误。
|
||
func (g *Gate) Rebuild(ctx context.Context, overlay []json.RawMessage) error {
|
||
if len(overlay) == 0 {
|
||
g.mu.Lock()
|
||
g.reg, g.specs = nil, 0
|
||
g.mu.Unlock()
|
||
return nil
|
||
}
|
||
specs := make([]agentintent.IntentSpec, 0, len(overlay))
|
||
for i, raw := range overlay {
|
||
var s agentintent.IntentSpec
|
||
if err := json.Unmarshal(raw, &s); err != nil {
|
||
return fmt.Errorf("intentgate: spec[%d]: %w", i, err)
|
||
}
|
||
specs = append(specs, s)
|
||
}
|
||
b := agentintent.NewRegistryBuilder(agentintent.Options{TopN: 3})
|
||
if err := b.Register(specs...); err != nil {
|
||
return fmt.Errorf("intentgate: register: %w", err)
|
||
}
|
||
reg, err := b.Build()
|
||
if err != nil {
|
||
return fmt.Errorf("intentgate: build: %w", err)
|
||
}
|
||
g.mu.Lock()
|
||
g.reg, g.specs = reg, len(specs)
|
||
g.mu.Unlock()
|
||
return nil
|
||
}
|
||
|
||
// Match 识别自由输入；返回 (branch, topID)。
|
||
// branch：matched | ambiguous | fallback | ""（未配置/系统级错误 → 回退通道）。
|
||
func (g *Gate) Match(ctx context.Context, input string) (string, string) {
|
||
g.mu.RLock()
|
||
reg, rec := g.reg, g.rec
|
||
g.mu.RUnlock()
|
||
if reg == nil {
|
||
return "", ""
|
||
}
|
||
res, err := rec.Recognize(ctx, input, reg.Snapshot())
|
||
if err != nil {
|
||
return "", ""
|
||
}
|
||
topID := ""
|
||
if res.Top != nil {
|
||
topID = res.Top.ID
|
||
}
|
||
switch res.Branch {
|
||
case agentintent.BranchMatched:
|
||
return "matched", topID
|
||
case agentintent.BranchAmbiguous:
|
||
return "ambiguous", topID
|
||
default:
|
||
return "fallback", topID
|
||
}
|
||
}
|
||
|
||
// Enabled 报告是否装载了意图（诊断/设置展示用）。
|
||
func (g *Gate) Enabled() bool {
|
||
g.mu.RLock()
|
||
defer g.mu.RUnlock()
|
||
return g.reg != nil && g.specs > 0
|
||
}
|
||
|
||
// SpecCount 返回已装载意图数。
|
||
func (g *Gate) SpecCount() int {
|
||
g.mu.RLock()
|
||
defer g.mu.RUnlock()
|
||
return g.specs
|
||
}
|