collab: Remove unused billing preferences queries (#34830)
This PR removes some billing preferences queries that are no longer in use. Release Notes: - N/A
This commit is contained in:
parent
e14c9479e4
commit
bf8aba566c
2 changed files with 0 additions and 77 deletions
|
@ -42,9 +42,6 @@ pub use tests::TestDb;
|
||||||
|
|
||||||
pub use ids::*;
|
pub use ids::*;
|
||||||
pub use queries::billing_customers::{CreateBillingCustomerParams, UpdateBillingCustomerParams};
|
pub use queries::billing_customers::{CreateBillingCustomerParams, UpdateBillingCustomerParams};
|
||||||
pub use queries::billing_preferences::{
|
|
||||||
CreateBillingPreferencesParams, UpdateBillingPreferencesParams,
|
|
||||||
};
|
|
||||||
pub use queries::billing_subscriptions::{
|
pub use queries::billing_subscriptions::{
|
||||||
CreateBillingSubscriptionParams, UpdateBillingSubscriptionParams,
|
CreateBillingSubscriptionParams, UpdateBillingSubscriptionParams,
|
||||||
};
|
};
|
||||||
|
|
|
@ -1,21 +1,5 @@
|
||||||
use anyhow::Context as _;
|
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
|
|
||||||
#[derive(Debug)]
|
|
||||||
pub struct CreateBillingPreferencesParams {
|
|
||||||
pub max_monthly_llm_usage_spending_in_cents: i32,
|
|
||||||
pub model_request_overages_enabled: bool,
|
|
||||||
pub model_request_overages_spend_limit_in_cents: i32,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[derive(Debug, Default)]
|
|
||||||
pub struct UpdateBillingPreferencesParams {
|
|
||||||
pub max_monthly_llm_usage_spending_in_cents: ActiveValue<i32>,
|
|
||||||
pub model_request_overages_enabled: ActiveValue<bool>,
|
|
||||||
pub model_request_overages_spend_limit_in_cents: ActiveValue<i32>,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Database {
|
impl Database {
|
||||||
/// Returns the billing preferences for the given user, if they exist.
|
/// Returns the billing preferences for the given user, if they exist.
|
||||||
pub async fn get_billing_preferences(
|
pub async fn get_billing_preferences(
|
||||||
|
@ -30,62 +14,4 @@ impl Database {
|
||||||
})
|
})
|
||||||
.await
|
.await
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Creates new billing preferences for the given user.
|
|
||||||
pub async fn create_billing_preferences(
|
|
||||||
&self,
|
|
||||||
user_id: UserId,
|
|
||||||
params: &CreateBillingPreferencesParams,
|
|
||||||
) -> Result<billing_preference::Model> {
|
|
||||||
self.transaction(|tx| async move {
|
|
||||||
let preferences = billing_preference::Entity::insert(billing_preference::ActiveModel {
|
|
||||||
user_id: ActiveValue::set(user_id),
|
|
||||||
max_monthly_llm_usage_spending_in_cents: ActiveValue::set(
|
|
||||||
params.max_monthly_llm_usage_spending_in_cents,
|
|
||||||
),
|
|
||||||
model_request_overages_enabled: ActiveValue::set(
|
|
||||||
params.model_request_overages_enabled,
|
|
||||||
),
|
|
||||||
model_request_overages_spend_limit_in_cents: ActiveValue::set(
|
|
||||||
params.model_request_overages_spend_limit_in_cents,
|
|
||||||
),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.exec_with_returning(&*tx)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(preferences)
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Updates the billing preferences for the given user.
|
|
||||||
pub async fn update_billing_preferences(
|
|
||||||
&self,
|
|
||||||
user_id: UserId,
|
|
||||||
params: &UpdateBillingPreferencesParams,
|
|
||||||
) -> Result<billing_preference::Model> {
|
|
||||||
self.transaction(|tx| async move {
|
|
||||||
let preferences = billing_preference::Entity::update_many()
|
|
||||||
.set(billing_preference::ActiveModel {
|
|
||||||
max_monthly_llm_usage_spending_in_cents: params
|
|
||||||
.max_monthly_llm_usage_spending_in_cents
|
|
||||||
.clone(),
|
|
||||||
model_request_overages_enabled: params.model_request_overages_enabled.clone(),
|
|
||||||
model_request_overages_spend_limit_in_cents: params
|
|
||||||
.model_request_overages_spend_limit_in_cents
|
|
||||||
.clone(),
|
|
||||||
..Default::default()
|
|
||||||
})
|
|
||||||
.filter(billing_preference::Column::UserId.eq(user_id))
|
|
||||||
.exec_with_returning(&*tx)
|
|
||||||
.await?;
|
|
||||||
|
|
||||||
Ok(preferences
|
|
||||||
.into_iter()
|
|
||||||
.next()
|
|
||||||
.context("billing preferences not found")?)
|
|
||||||
})
|
|
||||||
.await
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue