feat: pass session affinity through planner inputs

This commit is contained in:
RWDai
2026-05-05 11:23:54 +08:00
parent 37da4e245a
commit bcd542f7ee
3 changed files with 19 additions and 2 deletions
@@ -1,4 +1,5 @@
use aether_ai_serving::{run_ai_authenticated_decision_input, AiAuthenticatedDecisionInputPort};
use aether_scheduler_core::ClientSessionAffinity;
use async_trait::async_trait;
use crate::ai_serving::{ExecutionRuntimeAuthContext, GatewayAuthApiKeySnapshot, PlannerAppState};
@@ -19,6 +20,7 @@ pub(crate) struct LocalRequestedModelDecisionInput {
pub(crate) auth_snapshot: GatewayAuthApiKeySnapshot,
pub(crate) required_capabilities: Option<serde_json::Value>,
pub(crate) request_auth_channel: Option<String>,
pub(crate) client_session_affinity: Option<ClientSessionAffinity>,
}
#[derive(Debug, Clone)]
@@ -95,6 +97,7 @@ pub(crate) fn build_local_requested_model_decision_input(
auth_snapshot: resolved_input.auth_snapshot,
required_capabilities: resolved_input.required_capabilities,
request_auth_channel: None,
client_session_affinity: None,
}
}