Adapt corrections through PromptKit

internal/framework/pipeline/debug.go

@@ -110,24 +110,7 @@ type debugSerializedOutput struct {
 	Content       debugBinaryEnvelope      `json:"content"`
 }
 
-type debugLLMInputMaterial struct {
-	Name      string `json:"name"`
-	MediaType string `json:"media_type,omitempty"`
-	Content   string `json:"content_base64,omitempty"`
-	Digest    string `json:"digest,omitempty"`
-	OriginURI string `json:"origin_uri,omitempty"`
-	SizeBytes int64  `json:"size_bytes,omitempty"`
-}
-
-type debugStructuredCompletionRequest struct {
-	StageName     string                           `json:"stage_name"`
-	PromptID      string                           `json:"prompt_id,omitempty"`
-	PromptVersion string                           `json:"prompt_version,omitempty"`
-	ProfileID     string                           `json:"profile_id,omitempty"`
-	SessionID     string                           `json:"session_id,omitempty"`
-	Inputs        map[string]debugLLMInputMaterial `json:"inputs,omitempty"`
-	Vars          map[string]any                   `json:"vars,omitempty"`
-}
+type debugStructuredCompletionRequest = contracts.DebugStructuredCompletionRequest
 
 type debugStructuredCompletionResponse struct {
 	Content          string `json:"content,omitempty"`
@@ -567,29 +550,7 @@ func debugOutputFiles(files []contracts.OutputFile) []debugOutputFile {
 }
 
 func debugCompletionRequest(req contracts.StructuredCompletionRequest) debugStructuredCompletionRequest {
-	inputs := make(map[string]debugLLMInputMaterial, len(req.Inputs))
-	for key, material := range req.Inputs {
-		inputs[key] = debugLLMInputMaterial{
-			Name:      material.Name,
-			MediaType: material.MediaType,
-			Content:   base64.StdEncoding.EncodeToString(redactSecretBytes(material.Content)),
-			Digest:    material.Digest,
-			OriginURI: material.OriginURI,
-			SizeBytes: material.SizeBytes,
-		}
-	}
-	if len(inputs) == 0 {
-		inputs = nil
-	}
-	return debugStructuredCompletionRequest{
-		StageName:     req.StageName,
-		PromptID:      req.PromptID,
-		PromptVersion: req.PromptVersion,
-		ProfileID:     req.ProfileID,
-		SessionID:     req.SessionID,
-		Inputs:        inputs,
-		Vars:          redactSensitiveMap(req.Vars),
-	}
+	return req.DebugSummary()
 }
 
 func debugCompletionResponse(response contracts.StructuredCompletionResponse) debugStructuredCompletionResponse {

internal/framework/pipeline/debug_test.go

@@ -47,6 +47,32 @@ func TestDebugLLMPathsKeepDotIdentitiesDistinct(t *testing.T) {
 	}
 }
 
+func TestDebugCompletionRequestOmitsCorrectionContent(t *testing.T) {
+	const assistantResponse = `{"secret":"assistant response"}`
+	const userGuidance = "secret user guidance"
+	correction, err := contracts.NewSemanticCorrection([]byte(assistantResponse), userGuidance)
+	if err != nil {
+		t.Fatalf("NewSemanticCorrection() error = %v", err)
+	}
+	summary := debugCompletionRequest(contracts.StructuredCompletionRequest{
+		Inputs:     contracts.LLMInputSet{"source": contracts.NewLLMInputMaterial("source", "application/json", []byte(`{"source":true}`), "", "")},
+		Vars:       map[string]any{"custom": "value"},
+		Correction: correction,
+	})
+	encoded, err := json.Marshal(summary)
+	if err != nil {
+		t.Fatalf("marshal completion summary: %v", err)
+	}
+	for _, secret := range []string{assistantResponse, userGuidance} {
+		if strings.Contains(string(encoded), secret) {
+			t.Fatalf("debug completion summary leaked %q: %s", secret, encoded)
+		}
+	}
+	if summary.InputCount != 1 || summary.VariableCount != 1 || summary.Correction == nil {
+		t.Fatalf("debug completion summary = %#v, want counts and correction metadata", summary)
+	}
+}
+
 func TestDebugSourceDocumentPreservesUnitReferences(t *testing.T) {
 	doc := validSourceDocument()
 	envelope := debugSourceDocumentEnvelope(doc)