fix pass_headers

relay/common/override.go

@@ -24,6 +24,8 @@ const (
 	paramOverrideContextHeaderOverrideNormalized = "header_override_normalized"
 )
 
+var errSourceHeaderNotFound = errors.New("source header does not exist")
+
 type ConditionOperation struct {
 	Path           string      `json:"path"`             // JSON路径
 	Mode           string      `json:"mode"`             // full, prefix, suffix, contains, gt, gte, lt, lte
@@ -501,6 +503,10 @@ func applyOperations(jsonStr string, operations []ParamOperation, conditionConte
 			}
 			for _, headerName := range headerNames {
 				if err = copyHeaderInContext(context, headerName, headerName, op.KeepOrigin); err != nil {
+					if errors.Is(err, errSourceHeaderNotFound) {
+						err = nil
+						continue
+					}
 					break
 				}
 			}
@@ -654,7 +660,7 @@ func copyHeaderInContext(context map[string]interface{}, fromHeader, toHeader st
 	}
 	value, exists := getHeaderValueFromContext(context, fromHeader)
 	if !exists {
-		return fmt.Errorf("source header does not exist: %s", fromHeader)
+		return fmt.Errorf("%w: %s", errSourceHeaderNotFound, fromHeader)
 	}
 	return setHeaderOverrideInContext(context, toHeader, value, keepOrigin)
 }

relay/common/override_test.go

@@ -1060,6 +1060,43 @@ func TestApplyParamOverrideCopyHeaderFromRequestHeaders(t *testing.T) {
 	assertJSONEqual(t, `{"temperature":0.1}`, string(out))
 }
 
+func TestApplyParamOverridePassHeadersSkipsMissingHeaders(t *testing.T) {
+	input := []byte(`{"temperature":0.7}`)
+	override := map[string]interface{}{
+		"operations": []interface{}{
+			map[string]interface{}{
+				"mode":  "pass_headers",
+				"value": []interface{}{"X-Codex-Beta-Features", "Session_id"},
+			},
+		},
+	}
+	ctx := map[string]interface{}{
+		"request_headers_raw": map[string]interface{}{
+			"Session_id": "sess-123",
+		},
+		"request_headers": map[string]interface{}{
+			"session_id": "sess-123",
+		},
+	}
+
+	out, err := ApplyParamOverride(input, override, ctx)
+	if err != nil {
+		t.Fatalf("ApplyParamOverride returned error: %v", err)
+	}
+	assertJSONEqual(t, `{"temperature":0.7}`, string(out))
+
+	headers, ok := ctx["header_override"].(map[string]interface{})
+	if !ok {
+		t.Fatalf("expected header_override context map")
+	}
+	if headers["Session_id"] != "sess-123" {
+		t.Fatalf("expected Session_id to be passed, got: %v", headers["Session_id"])
+	}
+	if _, exists := headers["X-Codex-Beta-Features"]; exists {
+		t.Fatalf("expected missing header to be skipped")
+	}
+}
+
 func TestApplyParamOverrideSyncFieldsHeaderToJSON(t *testing.T) {
 	input := []byte(`{"model":"gpt-4"}`)
 	override := map[string]interface{}{