move memory caching module

This commit is contained in:
Brandon Thomas committed 2024-07-05 22:02:15 -04:00
1 parent bbaed857d0
commit 1cd3175e13
9 files changed
+5 -7

No files matched your search

@@ -37,7 +37,6 @@ use crate::state::server_state::{DurableInMemoryCaches, EnvConfig, EphemeralInMe
pub mod billing;
pub mod configs;
pub mod http_server;
pub mod memory_cache;
pub mod model;
pub mod state;
pub mod threads;
@@ -34,7 +34,6 @@ use crate::state::server_state::ServerState;
pub mod billing;
pub mod configs;
pub mod http_server;
pub mod memory_cache;
pub mod model;
pub mod state;
pub mod threads;
@@ -35,7 +35,7 @@ use tts_common::priority::FAKEYOU_INVESTOR_PRIORITY_LEVEL;
use crate::configs::plans::get_correct_plan_for_session::get_correct_plan_for_session;
use crate::http_server::endpoints::investor_demo::demo_cookie::request_has_demo_cookie;
use crate::http_server::web_utils::response_error_helpers::to_simple_json_error;
use crate::memory_cache::model_token_to_info_cache::{ModelInfoForInferenceJob, ModelTokenToInfoCache};
use crate::state::memory_cache::model_token_to_info_cache::{ModelInfoForInferenceJob, ModelTokenToInfoCache};
use crate::state::server_state::ServerState;
/// Debug requests can get routed to special "debug-only" workers, which can
@@ -78,7 +78,7 @@ use crate::configs::static_api_tokens::StaticApiTokenSet;
use crate::http_server::middleware::pushback_filter_middleware::PushbackFilter;
use crate::http_server::routes::add_routes::add_routes;
use crate::http_server::web_utils::handle_multipart_error::handle_multipart_error;
use crate::memory_cache::model_token_to_info_cache::ModelTokenToInfoCache;
use crate::state::memory_cache::model_token_to_info_cache::ModelTokenToInfoCache;
use crate::state::server_state::{DurableInMemoryCaches, EnvConfig, EphemeralInMemoryCaches, InMemoryCaches, ServerInfo, ServerState, StaticFeatureFlags, StripeSettings, TrollBans};
use crate::threads::db_health_checker_thread::db_health_check_status::HealthCheckStatus;
use crate::threads::db_health_checker_thread::db_health_checker_thread::db_health_checker_thread;
@@ -94,7 +94,6 @@ pub const RESERVED_SUBSTRINGS : &str = include_str!("../../../../../includes/bin
pub mod billing;
pub mod configs;
pub mod http_server;
pub mod memory_cache;
pub mod model;
pub mod state;
pub mod threads;
@@ -1 +1,2 @@
pub mod memory_cache;
pub mod server_state;
@@ -28,7 +28,7 @@ use crate::http_server::endpoints::stats::result_transformer::CacheableQueueStat
use crate::http_server::endpoints::tts::list_tts_models::TtsModelRecordForResponse;
use crate::http_server::endpoints::voice_conversion::models::list_voice_conversion_models::VoiceConversionModel;
use crate::http_server::web_utils::redis_rate_limiter::RedisRateLimiter;
use crate::memory_cache::model_token_to_info_cache::ModelTokenToInfoCache;
use crate::state::memory_cache::model_token_to_info_cache::ModelTokenToInfoCache;
use crate::StaticApiTokenSet;
use crate::threads::db_health_checker_thread::db_health_check_status::HealthCheckStatus;
use crate::util::encrypted_sort_id::SortKeyCrypto;
@@ -11,7 +11,7 @@ use errors::AnyhowResult;
use mysql_queries::queries::model_weights::model_weight_info_lite::list_model_weight_info_lite::list_model_weight_info_lite;
use mysql_queries::queries::voice_conversion::model_info_lite::list_voice_conversion_model_info_lite::list_voice_conversion_model_info_lite;
use crate::memory_cache::model_token_to_info_cache::{ModelInfoForInferenceJob, ModelTokenToInfoCache};
use crate::state::memory_cache::model_token_to_info_cache::{ModelInfoForInferenceJob, ModelTokenToInfoCache};
pub async fn poll_model_token_info_thread(
model_token_info_cache: ModelTokenToInfoCache,