Skip to content

Commit

Permalink
collab: Bypass account age check for users with active LLM subscripti…
Browse files Browse the repository at this point in the history
…ons (#20837)

This PR makes it so users with an active LLM subscription can bypass the
account age check.

Release Notes:

- N/A
  • Loading branch information
maxdeviant authored Nov 18, 2024
1 parent 3789918 commit e2552b9
Showing 1 changed file with 10 additions and 7 deletions.
17 changes: 10 additions & 7 deletions crates/collab/src/rpc.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4030,12 +4030,15 @@ async fn get_llm_api_token(
Err(anyhow!("terms of service not accepted"))?
}

let mut account_created_at = user.created_at;
if let Some(github_created_at) = user.github_user_created_at {
account_created_at = account_created_at.min(github_created_at);
}
if Utc::now().naive_utc() - account_created_at < MIN_ACCOUNT_AGE_FOR_LLM_USE {
Err(anyhow!("account too young"))?
let has_llm_subscription = session.has_llm_subscription(&db).await?;
if !has_llm_subscription {
let mut account_created_at = user.created_at;
if let Some(github_created_at) = user.github_user_created_at {
account_created_at = account_created_at.min(github_created_at);
}
if Utc::now().naive_utc() - account_created_at < MIN_ACCOUNT_AGE_FOR_LLM_USE {
Err(anyhow!("account too young"))?
}
}

let billing_preferences = db.get_billing_preferences(user.id).await?;
Expand All @@ -4045,7 +4048,7 @@ async fn get_llm_api_token(
session.is_staff(),
billing_preferences,
has_llm_closed_beta_feature_flag,
session.has_llm_subscription(&db).await?,
has_llm_subscription,
session.current_plan(&db).await?,
&session.app_state.config,
)?;
Expand Down

0 comments on commit e2552b9

Please sign in to comment.