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

fix(usage): align fast-tier pricing and model metadata

Browse Source
This commit is contained in:
elky
2026-07-18 16:57:04 +08:00
parent 6c33b8d8fb
commit e3d97b573b
11 changed files with 430 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files
apps/aether-gateway/src/data/state/models.rs
+30 -6
View File
@@ -255,20 +255,28 @@ impl GatewayDataState {
&self,
record: &UpsertAdminProviderModelRecord,
) -> Result<Option<StoredAdminProviderModel>, DataLayerError> {
match &self.global_model_writer {
let result = match &self.global_model_writer {
Some(repository) => repository.create_admin_provider_model(record).await,
None => Ok(None),
};
if result.as_ref().is_ok_and(Option::is_some) {
self.clear_billing_model_context_cache();
}
result
}
pub(crate) async fn update_admin_provider_model(
&self,
record: &UpsertAdminProviderModelRecord,
) -> Result<Option<StoredAdminProviderModel>, DataLayerError> {
match &self.global_model_writer {
let result = match &self.global_model_writer {
Some(repository) => repository.update_admin_provider_model(record).await,
None => Ok(None),
};
if result.as_ref().is_ok_and(Option::is_some) {
self.clear_billing_model_context_cache();
}
result
}
pub(crate) async fn delete_admin_provider_model(
@@ -276,44 +284,60 @@ impl GatewayDataState {
provider_id: &str,
model_id: &str,
) -> Result<bool, DataLayerError> {
match &self.global_model_writer {
let result = match &self.global_model_writer {
Some(repository) => {
repository
.delete_admin_provider_model(provider_id, model_id)
.await
}
None => Ok(false),
};
if result.as_ref().is_ok_and(|changed| *changed) {
self.clear_billing_model_context_cache();
}
result
}
pub(crate) async fn create_admin_global_model(
&self,
record: &CreateAdminGlobalModelRecord,
) -> Result<Option<StoredAdminGlobalModel>, DataLayerError> {
match &self.global_model_writer {
let result = match &self.global_model_writer {
Some(repository) => repository.create_admin_global_model(record).await,
None => Ok(None),
};
if result.as_ref().is_ok_and(Option::is_some) {
self.clear_billing_model_context_cache();
}
result
}
pub(crate) async fn update_admin_global_model(
&self,
record: &UpdateAdminGlobalModelRecord,
) -> Result<Option<StoredAdminGlobalModel>, DataLayerError> {
match &self.global_model_writer {
let result = match &self.global_model_writer {
Some(repository) => repository.update_admin_global_model(record).await,
None => Ok(None),
};
if result.as_ref().is_ok_and(Option::is_some) {
self.clear_billing_model_context_cache();
}
result
}
pub(crate) async fn delete_admin_global_model(
&self,
global_model_id: &str,
) -> Result<bool, DataLayerError> {
match &self.global_model_writer {
let result = match &self.global_model_writer {
Some(repository) => repository.delete_admin_global_model(global_model_id).await,
None => Ok(false),
};
if result.as_ref().is_ok_and(|changed| *changed) {
self.clear_billing_model_context_cache();
}
result
}
pub(crate) async fn list_provider_model_stats(