wehub-resource-sync
59 lines
2.0 KiB
Go
59 lines
2.0 KiB
Go
package agent
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
)
|
|
|
|
type memoryCompilerSourceInputContextKey struct{}
|
|
|
|
// WithMemoryCompilerSourceInput carries the user's unexpanded turn text for
|
|
// Memory v5 planning. Controller-level @reference resolution may prepend large
|
|
// file/resource blocks to the model input; those blocks should stay out of the
|
|
// compiler goal and source_event so strategy matching keys off the task itself.
|
|
func WithMemoryCompilerSourceInput(ctx context.Context, input string) context.Context {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
return context.WithValue(ctx, memoryCompilerSourceInputContextKey{}, input)
|
|
}
|
|
|
|
// MemoryCompilerSourceInputFromContext returns the unexpanded Memory v5 source
|
|
// turn when a controller supplied one.
|
|
func MemoryCompilerSourceInputFromContext(ctx context.Context) (string, bool) {
|
|
if ctx == nil {
|
|
return "", false
|
|
}
|
|
v, ok := ctx.Value(memoryCompilerSourceInputContextKey{}).(string)
|
|
if !ok {
|
|
return "", false
|
|
}
|
|
return strings.TrimSpace(v), strings.TrimSpace(v) != ""
|
|
}
|
|
|
|
type memoryCompilerSkipContextKey struct{}
|
|
|
|
// WithMemoryCompilerSkip marks a turn whose input is a synthetic,
|
|
// controller-injected message (goal-loop continuation, plan-approved execution,
|
|
// stream-recovery retries, …) rather than a genuine user request. Such turns
|
|
// must not be compiled into a memory_v5_execution_contract: the contract is
|
|
// echoed back by the model and, for the goal loop, re-injected every turn,
|
|
// which spins the loop indefinitely (#5342, #5329). Real user turns are
|
|
// unaffected and still compile normally.
|
|
func WithMemoryCompilerSkip(ctx context.Context) context.Context {
|
|
if ctx == nil {
|
|
ctx = context.Background()
|
|
}
|
|
return context.WithValue(ctx, memoryCompilerSkipContextKey{}, true)
|
|
}
|
|
|
|
// MemoryCompilerSkipFromContext reports whether the current turn was marked as a
|
|
// synthetic message that should bypass Memory v5 compilation.
|
|
func MemoryCompilerSkipFromContext(ctx context.Context) bool {
|
|
if ctx == nil {
|
|
return false
|
|
}
|
|
skip, _ := ctx.Value(memoryCompilerSkipContextKey{}).(bool)
|
|
return skip
|
|
}
|