i/Aether
1
0
Fork 0
mirror of https://github.com/fawney19/Aether.git synced 2026-10-12 20:19:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: carry session affinity through Gemini files

Browse Source
This commit is contained in:
RWDai
2026-05-05 12:25:07 +08:00
parent 628329e493
commit 657f9f0be1
6 changed files with 131 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/ai_serving/planner/specialized/files.rs
+30 -6
View File
@@ -103,7 +103,14 @@ pub(crate) async fn build_local_gemini_files_sync_attempt_source_for_kind<'a>(
return Ok(None);
};
let Some(input) = resolve_local_gemini_files_decision_input(state, trace_id, decision).await
let Some(input) = resolve_local_gemini_files_decision_input(
state,
parts,
Some(body_json),
trace_id,
decision,
)
.await
else {
return Ok(None);
};
@@ -140,7 +147,8 @@ pub(crate) async fn build_local_gemini_files_stream_attempt_source_for_kind<'a>(
return Ok(None);
};
let Some(input) = resolve_local_gemini_files_decision_input(state, trace_id, decision).await
let Some(input) =
resolve_local_gemini_files_decision_input(state, parts, None, trace_id, decision).await
else {
return Ok(None);
};
@@ -298,7 +306,14 @@ pub(crate) async fn maybe_build_sync_local_gemini_files_decision_payload(
return Ok(None);
};
let Some(input) = resolve_local_gemini_files_decision_input(state, trace_id, decision).await
let Some(input) = resolve_local_gemini_files_decision_input(
state,
parts,
Some(body_json),
trace_id,
decision,
)
.await
else {
return Ok(None);
};
@@ -338,7 +353,8 @@ pub(crate) async fn maybe_build_stream_local_gemini_files_decision_payload(
return Ok(None);
};
let Some(input) = resolve_local_gemini_files_decision_input(state, trace_id, decision).await
let Some(input) =
resolve_local_gemini_files_decision_input(state, parts, None, trace_id, decision).await
else {
return Ok(None);
};
@@ -379,7 +395,14 @@ async fn build_local_sync_plan_and_reports(
spec: LocalGeminiFilesSpec,
) -> Result<Vec<AiSyncAttempt>, GatewayError> {
let spec_metadata = local_gemini_files_spec_metadata(spec);
let Some(input) = resolve_local_gemini_files_decision_input(state, trace_id, decision).await
let Some(input) = resolve_local_gemini_files_decision_input(
state,
parts,
Some(body_json),
trace_id,
decision,
)
.await
else {
return Ok(Vec::new());
};
@@ -430,7 +453,8 @@ async fn build_local_stream_plan_and_reports(
spec: LocalGeminiFilesSpec,
) -> Result<Vec<AiStreamAttempt>, GatewayError> {
let spec_metadata = local_gemini_files_spec_metadata(spec);
let Some(input) = resolve_local_gemini_files_decision_input(state, trace_id, decision).await
let Some(input) =
resolve_local_gemini_files_decision_input(state, parts, None, trace_id, decision).await
else {
return Ok(Vec::new());
};