Implement generic pool member scoring and probing

2026-09-02 17:30:23 +08:00 · 2026-05-12 01:46:22 +08:00
parent cb0ccb9cdb
commit b9e62d1667
76 changed files with 6276 additions and 68 deletions
--- a/apps/aether-gateway/src/data/state/core.rs
+++ b/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()
    }
--- a/apps/aether-gateway/src/data/state/mod.rs
+++ b/apps/aether-gateway/src/data/state/mod.rs
@@ -93,7 +93,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,
@@ -107,6 +108,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,
@@ -156,6 +164,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>>,
@@ -257,6 +267,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(),
@@ -287,6 +299,7 @@ mod catalog;
 mod core;
 mod integrations;
 mod models;
+mod pool_scores;
 mod runtime;
 #[cfg(test)]
 mod testing;
--- a/apps/aether-gateway/src/data/state/models.rs
+++ b/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,
--- a/apps/aether-gateway/src/data/state/pool_scores.rs
+++ b/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),
+        }
+    }
+}
--- a/apps/aether-gateway/src/data/state/testing/announcements.rs
+++ b/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,
--- a/apps/aether-gateway/src/data/state/testing/mod.rs
+++ b/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,
--- a/apps/aether-gateway/src/data/state/testing/video_tasks.rs
+++ b/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,