Merge remote-tracking branch 'origin/aether-rust-pioneer' into payment-billing-plans

# Conflicts:
#	crates/aether-data/src/lifecycle/bootstrap/postgres.rs
#	crates/aether-data/src/lifecycle/migrate/tests.rs

apps/aether-gateway/src/data/state/core.rs

@@ -45,6 +45,8 @@ impl GatewayDataState {
                 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,
@@ -86,6 +88,8 @@ impl GatewayDataState {
         let gemini_file_mapping_writer = backends.write().gemini_file_mappings();
         let provider_catalog_reader = backends.read().provider_catalog();
         let provider_catalog_writer = backends.write().provider_catalog();
+        let pool_score_reader = backends.read().pool_scores();
+        let pool_score_writer = backends.write().pool_scores();
         let provider_quota_reader = backends.read().provider_quotas();
         let provider_quota_writer = backends.write().provider_quotas();
         let usage_reader = backends.read().usage();
@@ -125,6 +129,8 @@ impl GatewayDataState {
             request_candidate_writer,
             provider_catalog_reader,
             provider_catalog_writer,
+            pool_score_reader,
+            pool_score_writer,
             provider_quota_reader,
             provider_quota_writer,
             usage_reader,
@@ -263,6 +269,14 @@ impl GatewayDataState {
         self.provider_catalog_writer.is_some()
     }
 
+    pub(crate) fn has_pool_score_reader(&self) -> bool {
+        self.pool_score_reader.is_some()
+    }
+
+    pub(crate) fn has_pool_score_writer(&self) -> bool {
+        self.pool_score_writer.is_some()
+    }
+
     pub(crate) fn has_proxy_node_reader(&self) -> bool {
         self.proxy_node_reader.is_some()
     }

apps/aether-gateway/src/data/state/mod.rs

@@ -95,7 +95,8 @@ use aether_data_contracts::repository::billing::{
 };
 use aether_data_contracts::repository::candidate_selection::{
     MinimalCandidateSelectionReadRepository, StoredMinimalCandidateSelectionRow,
-    StoredPoolKeyCandidateRowsQuery, StoredRequestedModelCandidateRowsQuery,
+    StoredPoolKeyCandidateRowsByKeyIdsQuery, StoredPoolKeyCandidateRowsQuery,
+    StoredRequestedModelCandidateRowsQuery,
 };
 use aether_data_contracts::repository::candidates::{
     PublicHealthStatusCount, PublicHealthTimelineBucket, RequestCandidateReadRepository,
@@ -109,6 +110,13 @@ use aether_data_contracts::repository::global_models::{
     StoredProviderModelStats, StoredPublicCatalogModel, StoredPublicGlobalModel,
     StoredPublicGlobalModelPage, UpdateAdminGlobalModelRecord, UpsertAdminProviderModelRecord,
 };
+use aether_data_contracts::repository::pool_scores::{
+    GetPoolMemberScoresByIdsQuery, ListPoolMemberProbeCandidatesQuery, ListPoolMemberScoresQuery,
+    ListRankedPoolMembersQuery, PoolMemberHardState, PoolMemberIdentity, PoolMemberProbeAttempt,
+    PoolMemberProbeResult, PoolMemberProbeStatus, PoolMemberScheduleFeedback,
+    PoolMemberScoreWriteRepository, PoolScoreReadRepository, PoolScoreScope, StoredPoolMemberScore,
+    UpsertPoolMemberScore,
+};
 use aether_data_contracts::repository::provider_catalog::{
     ProviderCatalogKeyListQuery, ProviderCatalogReadRepository, ProviderCatalogWriteRepository,
     StoredProviderCatalogEndpoint, StoredProviderCatalogKey, StoredProviderCatalogKeyPage,
@@ -158,6 +166,8 @@ pub(crate) struct GatewayDataState {
     request_candidate_writer: Option<Arc<dyn RequestCandidateWriteRepository>>,
     provider_catalog_reader: Option<Arc<dyn ProviderCatalogReadRepository>>,
     provider_catalog_writer: Option<Arc<dyn ProviderCatalogWriteRepository>>,
+    pool_score_reader: Option<Arc<dyn PoolScoreReadRepository>>,
+    pool_score_writer: Option<Arc<dyn PoolMemberScoreWriteRepository>>,
     provider_quota_reader: Option<Arc<dyn ProviderQuotaReadRepository>>,
     provider_quota_writer: Option<Arc<dyn ProviderQuotaWriteRepository>>,
     usage_reader: Option<Arc<dyn UsageReadRepository>>,
@@ -259,6 +269,8 @@ impl fmt::Debug for GatewayDataState {
                 "has_provider_catalog_writer",
                 &self.provider_catalog_writer.is_some(),
             )
+            .field("has_pool_score_reader", &self.pool_score_reader.is_some())
+            .field("has_pool_score_writer", &self.pool_score_writer.is_some())
             .field(
                 "has_provider_quota_reader",
                 &self.provider_quota_reader.is_some(),
@@ -289,6 +301,7 @@ mod catalog;
 mod core;
 mod integrations;
 mod models;
+mod pool_scores;
 mod runtime;
 #[cfg(test)]
 mod testing;

apps/aether-gateway/src/data/state/models.rs

@@ -2,7 +2,8 @@ use super::{
     AdminGlobalModelListQuery, AdminProviderModelListQuery, CreateAdminGlobalModelRecord,
     DataLayerError, GatewayDataState, PublicCatalogModelListQuery, PublicCatalogModelSearchQuery,
     PublicGlobalModelQuery, StoredAdminGlobalModel, StoredAdminGlobalModelPage,
-    StoredAdminProviderModel, StoredMinimalCandidateSelectionRow, StoredPoolKeyCandidateRowsQuery,
+    StoredAdminProviderModel, StoredMinimalCandidateSelectionRow,
+    StoredPoolKeyCandidateRowsByKeyIdsQuery, StoredPoolKeyCandidateRowsQuery,
     StoredProviderActiveGlobalModel, StoredProviderModelStats, StoredPublicCatalogModel,
     StoredPublicGlobalModel, StoredPublicGlobalModelPage, StoredRequestedModelCandidateRowsQuery,
     UpdateAdminGlobalModelRecord, UpsertAdminProviderModelRecord,
@@ -73,6 +74,16 @@ impl GatewayDataState {
         }
     }
 
+    pub(crate) async fn list_pool_key_candidate_rows_for_group_key_ids(
+        &self,
+        query: &StoredPoolKeyCandidateRowsByKeyIdsQuery,
+    ) -> Result<Vec<StoredMinimalCandidateSelectionRow>, DataLayerError> {
+        match &self.minimal_candidate_selection_reader {
+            Some(repository) => repository.list_pool_key_rows_for_group_key_ids(query).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
     pub(crate) async fn list_public_global_models(
         &self,
         query: &PublicGlobalModelQuery,

apps/aether-gateway/src/data/state/pool_scores.rs

@@ -0,0 +1,123 @@
+use super::{
+    DataLayerError, GatewayDataState, GetPoolMemberScoresByIdsQuery,
+    ListPoolMemberProbeCandidatesQuery, ListPoolMemberScoresQuery, ListRankedPoolMembersQuery,
+    PoolMemberHardState, PoolMemberIdentity, PoolMemberProbeAttempt, PoolMemberProbeResult,
+    PoolMemberScheduleFeedback, PoolScoreScope, StoredPoolMemberScore, UpsertPoolMemberScore,
+};
+
+impl GatewayDataState {
+    pub(crate) async fn list_ranked_pool_members(
+        &self,
+        query: &ListRankedPoolMembersQuery,
+    ) -> Result<Vec<StoredPoolMemberScore>, DataLayerError> {
+        match &self.pool_score_reader {
+            Some(repository) => repository.list_ranked_pool_members(query).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn list_pool_member_probe_candidates(
+        &self,
+        query: &ListPoolMemberProbeCandidatesQuery,
+    ) -> Result<Vec<StoredPoolMemberScore>, DataLayerError> {
+        match &self.pool_score_reader {
+            Some(repository) => repository.list_pool_member_probe_candidates(query).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn list_pool_member_scores(
+        &self,
+        query: &ListPoolMemberScoresQuery,
+    ) -> Result<Vec<StoredPoolMemberScore>, DataLayerError> {
+        match &self.pool_score_reader {
+            Some(repository) => repository.list_pool_member_scores(query).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn get_pool_member_scores_by_ids(
+        &self,
+        query: &GetPoolMemberScoresByIdsQuery,
+    ) -> Result<Vec<StoredPoolMemberScore>, DataLayerError> {
+        match &self.pool_score_reader {
+            Some(repository) => repository.get_pool_member_scores_by_ids(query).await,
+            None => Ok(Vec::new()),
+        }
+    }
+
+    pub(crate) async fn upsert_pool_member_score(
+        &self,
+        score: UpsertPoolMemberScore,
+    ) -> Result<Option<StoredPoolMemberScore>, DataLayerError> {
+        match &self.pool_score_writer {
+            Some(repository) => repository.upsert_pool_member_score(score).await.map(Some),
+            None => Ok(None),
+        }
+    }
+
+    pub(crate) async fn record_pool_member_probe_result(
+        &self,
+        result: PoolMemberProbeResult,
+    ) -> Result<usize, DataLayerError> {
+        match &self.pool_score_writer {
+            Some(repository) => repository.record_pool_member_probe_result(result).await,
+            None => Ok(0),
+        }
+    }
+
+    pub(crate) async fn mark_pool_member_probe_in_progress(
+        &self,
+        attempt: PoolMemberProbeAttempt,
+    ) -> Result<usize, DataLayerError> {
+        match &self.pool_score_writer {
+            Some(repository) => repository.mark_pool_member_probe_in_progress(attempt).await,
+            None => Ok(0),
+        }
+    }
+
+    pub(crate) async fn record_pool_member_schedule_feedback(
+        &self,
+        feedback: PoolMemberScheduleFeedback,
+    ) -> Result<usize, DataLayerError> {
+        match &self.pool_score_writer {
+            Some(repository) => {
+                repository
+                    .record_pool_member_schedule_feedback(feedback)
+                    .await
+            }
+            None => Ok(0),
+        }
+    }
+
+    pub(crate) async fn mark_pool_member_hard_state(
+        &self,
+        identity: &PoolMemberIdentity,
+        scope: Option<&PoolScoreScope>,
+        hard_state: PoolMemberHardState,
+        updated_at: u64,
+    ) -> Result<usize, DataLayerError> {
+        match &self.pool_score_writer {
+            Some(repository) => {
+                repository
+                    .mark_pool_member_hard_state(identity, scope, hard_state, updated_at)
+                    .await
+            }
+            None => Ok(0),
+        }
+    }
+
+    pub(crate) async fn delete_pool_member_scores_for_member(
+        &self,
+        identity: &PoolMemberIdentity,
+    ) -> Result<usize, DataLayerError> {
+        match &self.pool_score_writer {
+            Some(repository) => {
+                repository
+                    .delete_pool_member_scores_for_member(identity)
+                    .await
+            }
+            None => Ok(0),
+        }
+    }
+}

apps/aether-gateway/src/data/state/testing/announcements.rs

@@ -34,6 +34,8 @@ impl GatewayDataState {
             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,
@@ -86,6 +88,8 @@ impl GatewayDataState {
             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,

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

@@ -2,6 +2,7 @@ use std::collections::BTreeMap;
 use std::sync::{Arc, RwLock};
 
 use aether_data_contracts::repository::candidates::RequestCandidateRepository;
+use aether_data_contracts::repository::pool_scores::PoolMemberScoreRepository;
 use aether_data_contracts::repository::quota::ProviderQuotaRepository;
 use aether_data_contracts::repository::usage::UsageRepository;
 
@@ -12,12 +13,13 @@ use super::{
     GeminiFileMappingWriteRepository, GlobalModelReadRepository, GlobalModelWriteRepository,
     ManagementTokenReadRepository, ManagementTokenWriteRepository,
     MinimalCandidateSelectionReadRepository, OAuthProviderReadRepository,
-    OAuthProviderWriteRepository, ProviderCatalogReadRepository, ProviderCatalogWriteRepository,
-    ProviderQuotaReadRepository, ProviderQuotaWriteRepository, ProxyNodeReadRepository,
-    ProxyNodeWriteRepository, RequestCandidateReadRepository, RequestCandidateWriteRepository,
-    SettlementWriteRepository, StoredSystemConfigEntry, StoredUserPreferenceRecord,
-    UsageReadRepository, UsageWriteRepository, UserReadRepository, VideoTaskReadRepository,
-    VideoTaskWriteRepository, WalletReadRepository, WalletWriteRepository,
+    OAuthProviderWriteRepository, PoolMemberScoreWriteRepository, PoolScoreReadRepository,
+    ProviderCatalogReadRepository, ProviderCatalogWriteRepository, ProviderQuotaReadRepository,
+    ProviderQuotaWriteRepository, ProxyNodeReadRepository, ProxyNodeWriteRepository,
+    RequestCandidateReadRepository, RequestCandidateWriteRepository, SettlementWriteRepository,
+    StoredSystemConfigEntry, StoredUserPreferenceRecord, UsageReadRepository, UsageWriteRepository,
+    UserReadRepository, VideoTaskReadRepository, VideoTaskWriteRepository, WalletReadRepository,
+    WalletWriteRepository,
 };
 
 mod announcements;
@@ -67,6 +69,8 @@ impl GatewayDataState {
             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,
@@ -118,6 +122,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             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,
@@ -165,6 +171,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -300,6 +308,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -333,6 +343,18 @@ impl GatewayDataState {
         self
     }
 
+    #[cfg(test)]
+    pub(crate) fn with_pool_score_repository_for_tests<T>(mut self, repository: Arc<T>) -> Self
+    where
+        T: PoolMemberScoreRepository + 'static,
+    {
+        let pool_score_reader: Arc<dyn PoolScoreReadRepository> = repository.clone();
+        let pool_score_writer: Arc<dyn PoolMemberScoreWriteRepository> = repository;
+        self.pool_score_reader = Some(pool_score_reader);
+        self.pool_score_writer = Some(pool_score_writer);
+        self
+    }
+
     #[cfg(test)]
     pub(crate) fn with_provider_catalog_and_request_candidate_reader_for_tests(
         provider_catalog_repository: Arc<dyn ProviderCatalogReadRepository>,
@@ -363,6 +385,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -419,6 +443,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,
@@ -484,6 +510,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,
@@ -531,6 +559,8 @@ impl GatewayDataState {
             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,
@@ -579,6 +609,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -638,6 +670,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             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: Some(usage_reader),
@@ -699,6 +733,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             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,
@@ -744,6 +780,8 @@ impl GatewayDataState {
             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: Some(repository),
@@ -804,6 +842,8 @@ impl GatewayDataState {
             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,
@@ -857,6 +897,8 @@ impl GatewayDataState {
             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,
@@ -915,6 +957,8 @@ impl GatewayDataState {
             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: Some(usage_reader),
@@ -974,6 +1018,8 @@ impl GatewayDataState {
             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,
@@ -1032,6 +1078,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: Some(usage_reader),
@@ -1079,6 +1127,8 @@ impl GatewayDataState {
             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,
@@ -1126,6 +1176,8 @@ impl GatewayDataState {
             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,
@@ -1185,6 +1237,8 @@ impl GatewayDataState {
             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,
@@ -1249,6 +1303,8 @@ impl GatewayDataState {
             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,
@@ -1296,6 +1352,8 @@ impl GatewayDataState {
             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,
@@ -1348,6 +1406,8 @@ impl GatewayDataState {
             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,
@@ -1417,6 +1477,8 @@ impl GatewayDataState {
             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,
@@ -1481,6 +1543,8 @@ impl GatewayDataState {
             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,
@@ -1529,6 +1593,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -1577,6 +1643,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -1627,6 +1695,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: Some(usage_repository),
@@ -1675,6 +1745,8 @@ impl GatewayDataState {
             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,
@@ -1723,6 +1795,8 @@ impl GatewayDataState {
             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,
@@ -1779,6 +1853,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: None,
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,
@@ -1836,6 +1912,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: None,
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,
@@ -1896,6 +1974,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,
@@ -1962,6 +2042,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -2029,6 +2111,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -2100,6 +2184,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: Some(usage_reader),
@@ -2178,6 +2264,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: Some(provider_catalog_writer),
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: Some(usage_reader),
@@ -2238,6 +2326,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,
@@ -2289,6 +2379,8 @@ impl GatewayDataState {
             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: Some(usage_reader),
@@ -2336,6 +2428,8 @@ impl GatewayDataState {
             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,
@@ -2389,6 +2483,8 @@ impl GatewayDataState {
             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,
@@ -2446,6 +2542,8 @@ impl GatewayDataState {
             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: Some(usage_reader),
@@ -2504,6 +2602,8 @@ impl GatewayDataState {
             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: Some(usage_reader),
@@ -2555,6 +2655,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: None,
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: Some(provider_quota_reader),
             provider_quota_writer: Some(provider_quota_writer),
             usage_reader: None,

apps/aether-gateway/src/data/state/testing/video_tasks.rs

@@ -38,6 +38,8 @@ impl GatewayDataState {
             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,
@@ -92,6 +94,8 @@ impl GatewayDataState {
             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,
@@ -143,6 +147,8 @@ impl GatewayDataState {
             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,
@@ -198,6 +204,8 @@ impl GatewayDataState {
             request_candidate_writer: None,
             provider_catalog_reader: Some(provider_catalog_repository),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,
@@ -257,6 +265,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             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,
@@ -325,6 +335,8 @@ impl GatewayDataState {
             request_candidate_writer: Some(request_candidate_writer),
             provider_catalog_reader: Some(provider_catalog_reader),
             provider_catalog_writer: None,
+            pool_score_reader: None,
+            pool_score_writer: None,
             provider_quota_reader: None,
             provider_quota_writer: None,
             usage_reader: None,