i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-09-01 17:00:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(scheduler): candidate report 携带 header_rules 与 body_rules

Browse Source 
各 planner 在构造 report context 时附带端点的 header/body 改写规则，scheduler 解析后写入 candidate extra_data，便于追踪请求实际应用的透传规则
This commit is contained in:
fawney19
2026-04-25 11:43:53 +08:00
parent 65e915fd1d
commit 61e12e2c17
9 changed files with 68 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/aether-gateway/src/ai_pipeline/planner/passthrough/provider/family/payload.rs
View File

@@ -99,6 +99,8 @@ pub(crate) async fn maybe_build_local_same_format_provider_decision_payload_for_
mapped_model: Some(&resolved.mapped_model),
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: Some(&resolved.upstream_url),
header_rules: resolved.transport.endpoint.header_rules.as_ref(),
body_rules: resolved.transport.endpoint.body_rules.as_ref(),
provider_request_method: Some(serde_json::Value::Null),
provider_request_headers: Some(&resolved.provider_request_headers),
original_headers: &parts.headers,

8
apps/aether-gateway/src/ai_pipeline/planner/report_context.rs
View File

@@ -24,6 +24,8 @@ pub(crate) struct LocalExecutionReportContextParts<'a> {
pub(crate) mapped_model: Option<&'a str>,
pub(crate) candidate_group_id: Option<&'a str>,
pub(crate) upstream_url: Option<&'a str>,
pub(crate) header_rules: Option<&'a Value>,
pub(crate) body_rules: Option<&'a Value>,
pub(crate) provider_request_method: Option<Value>,
pub(crate) provider_request_headers: Option<&'a BTreeMap<String, String>>,
pub(crate) original_headers: &'a http::HeaderMap,
@@ -172,6 +174,12 @@ pub(crate) fn build_local_execution_report_context(
Value::String(upstream_url.to_string()),
);
}
if let Some(header_rules) = parts.header_rules {
object.insert("header_rules".to_string(), header_rules.clone());
}
if let Some(body_rules) = parts.body_rules {
object.insert("body_rules".to_string(), body_rules.clone());
}
if let Some(provider_request_method) = parts.provider_request_method {
object.insert(
"provider_request_method".to_string(),

2
apps/aether-gateway/src/ai_pipeline/planner/specialized/files/decision.rs
View File

@@ -84,6 +84,8 @@ pub(super) async fn maybe_build_local_gemini_files_decision_payload_for_candidat
mapped_model: None,
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: None,
header_rules: transport.endpoint.header_rules.as_ref(),
body_rules: transport.endpoint.body_rules.as_ref(),
provider_request_method: None,
provider_request_headers: None,
original_headers: &parts.headers,

2
apps/aether-gateway/src/ai_pipeline/planner/specialized/image/decision.rs
View File

@@ -81,6 +81,8 @@ pub(super) async fn maybe_build_local_openai_image_decision_payload_for_candidat
mapped_model: Some(&resolved.mapped_model),
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: Some(&resolved.upstream_url),
header_rules: transport.endpoint.header_rules.as_ref(),
body_rules: transport.endpoint.body_rules.as_ref(),
provider_request_method: Some(serde_json::Value::String(parts.method.to_string())),
provider_request_headers: Some(&resolved.provider_request_headers),
original_headers: &parts.headers,

2
apps/aether-gateway/src/ai_pipeline/planner/specialized/video/decision.rs
View File

@@ -67,6 +67,8 @@ pub(super) async fn maybe_build_local_video_create_decision_payload_for_candidat
mapped_model: Some(&resolved.mapped_model),
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: None,
header_rules: transport.endpoint.header_rules.as_ref(),
body_rules: transport.endpoint.body_rules.as_ref(),
provider_request_method: None,
provider_request_headers: None,
original_headers: &parts.headers,

2
apps/aether-gateway/src/ai_pipeline/planner/standard/family/payload.rs
View File

@@ -79,6 +79,8 @@ pub(super) async fn maybe_build_local_standard_decision_payload_for_candidate(
mapped_model: Some(&resolved.mapped_model),
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: Some(&resolved.upstream_url),
header_rules: resolved.transport.endpoint.header_rules.as_ref(),
body_rules: resolved.transport.endpoint.body_rules.as_ref(),
provider_request_method: Some(serde_json::Value::Null),
provider_request_headers: Some(&resolved.provider_request_headers),
original_headers: &parts.headers,

2
apps/aether-gateway/src/ai_pipeline/planner/standard/openai/chat/decision/payload.rs
View File

@@ -101,6 +101,8 @@ pub(crate) async fn maybe_build_local_openai_chat_decision_payload_for_candidate
mapped_model: Some(&resolved.mapped_model),
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: Some(&resolved.upstream_url),
header_rules: resolved.transport.endpoint.header_rules.as_ref(),
body_rules: resolved.transport.endpoint.body_rules.as_ref(),
provider_request_method: Some(serde_json::Value::Null),
provider_request_headers: Some(&resolved.provider_request_headers),
original_headers: &parts.headers,

2
apps/aether-gateway/src/ai_pipeline/planner/standard/openai/cli/decision/payload.rs
View File

@@ -99,6 +99,8 @@ pub(crate) async fn maybe_build_local_openai_cli_decision_payload_for_candidate(
mapped_model: Some(&resolved.mapped_model),
candidate_group_id: eligible.orchestration.candidate_group_id.as_deref(),
upstream_url: Some(&resolved.upstream_url),
header_rules: resolved.transport.endpoint.header_rules.as_ref(),
body_rules: resolved.transport.endpoint.body_rules.as_ref(),
provider_request_method: Some(serde_json::Value::Null),
provider_request_headers: Some(&resolved.provider_request_headers),
original_headers: &parts.headers,