First Commit

intercept/review_context.go

@@ -0,0 +1,70 @@
+package intercept
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+	"strings"
+)
+
+const reviewTextLimit = 4000
+
+const BackgroundUserMessage = "user_message"
+
+// ReviewBackground is explicitly bound from the current human message. Generated
+// Worker summaries are not accepted. Background cannot override review policy.
+type ReviewBackground struct {
+	Source    string `json:"source"`
+	Text      string `json:"text"`
+	Truncated bool   `json:"truncated,omitempty"`
+}
+
+// ReviewInput contains only the current call and explicitly selected background.
+// Execution history and call correlation belong to the separate audit record.
+type ReviewInput struct {
+	Version    int               `json:"version"`
+	WorkingDir string            `json:"working_directory,omitempty"`
+	Background *ReviewBackground `json:"background,omitempty"`
+	Tool       string            `json:"tool_name"`
+	Arguments  json.RawMessage   `json:"arguments"`
+}
+
+type reviewContextKey struct{}
+type reviewEnvironment struct {
+	workingDir string
+	background ReviewBackground
+}
+
+// WithReviewContext explicitly binds the permitted background for one run. Never
+// fall back to the raw turn transcript: it may contain the full scheduler prompt.
+// This is application wiring, not a model-callable tool.
+func WithReviewContext(ctx context.Context, workingDir string, background ReviewBackground) context.Context {
+	return context.WithValue(ctx, reviewContextKey{}, reviewEnvironment{workingDir, background})
+}
+
+// WithReviewWorkingDirectory preserves only explicitly selected background.
+// Chat runs can be human-initiated or scheduled, so the Agent must not infer
+// message provenance from the text it receives.
+func WithReviewWorkingDirectory(ctx context.Context, workingDir string) context.Context {
+	env, _ := ctx.Value(reviewContextKey{}).(reviewEnvironment)
+	env.workingDir = workingDir
+	return context.WithValue(ctx, reviewContextKey{}, env)
+}
+
+func BuildReviewInput(ctx context.Context, tool string, arguments json.RawMessage) (ReviewInput, error) {
+	if !json.Valid(arguments) {
+		return ReviewInput{}, fmt.Errorf("工具参数不是有效 JSON")
+	}
+	in := ReviewInput{Version: 4, Tool: tool, Arguments: append(json.RawMessage(nil), arguments...)}
+	if env, ok := ctx.Value(reviewContextKey{}).(reviewEnvironment); ok {
+		in.WorkingDir = env.workingDir
+		background := env.background
+		if background.Source == BackgroundUserMessage && strings.TrimSpace(background.Text) != "" {
+			var cut bool
+			background.Text, cut = bounded(background.Text, reviewTextLimit)
+			background.Truncated = background.Truncated || cut
+			in.Background = &background
+		}
+	}
+	return in, nil
+}