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

fix(gateway): 修复额度预检误判余额不足

Browse Source
This commit is contained in:
Entropy.Xu
2026-05-16 20:45:17 +08:00
parent 3a5922d4ee
commit bddae47454
2 changed files with 251 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
apps/aether-gateway/src/data/state/testing/mod.rs
View File

@@ -1765,6 +1765,56 @@ impl GatewayDataState {
}
}
#[cfg(test)]
pub(crate) fn with_minimal_candidate_selection_and_billing_for_tests(
candidate_selection_repository: Arc<dyn MinimalCandidateSelectionReadRepository>,
billing_repository: Arc<dyn BillingReadRepository>,
) -> Self {
Self {
config: GatewayDataConfig::disabled(),
backends: None,
auth_api_key_reader: None,
auth_api_key_writer: None,
auth_module_reader: None,
auth_module_writer: None,
announcement_reader: None,
announcement_writer: None,
management_token_reader: None,
management_token_writer: None,
oauth_provider_reader: None,
oauth_provider_writer: None,
proxy_node_reader: None,
proxy_node_writer: None,
billing_reader: Some(billing_repository),
gemini_file_mapping_reader: None,
gemini_file_mapping_writer: None,
global_model_reader: None,
global_model_writer: None,
minimal_candidate_selection_reader: Some(candidate_selection_repository),
request_candidate_reader: None,
request_candidate_writer: None,
provider_catalog_reader: None,
provider_catalog_writer: None,
pool_score_reader: None,
pool_score_writer: None,
provider_quota_reader: None,
provider_quota_writer: None,
usage_reader: None,
usage_writer: None,
user_reader: None,
user_preferences: None,
usage_worker_queue: None,
video_task_reader: None,
video_task_writer: None,
background_task_reader: None,
background_task_writer: None,
wallet_reader: None,
wallet_writer: None,
settlement_writer: None,
system_config_values: None,
}
}
#[cfg(test)]
pub(crate) fn with_minimal_candidate_selection_and_auth_for_tests(
candidate_selection_repository: Arc<dyn MinimalCandidateSelectionReadRepository>,