Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

collab: Bypass account age check for users with active LLM subscriptions (#20837)

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

Release Notes:

- N/A
This commit is contained in:
Marshall Bowers 2024-11-18 16:37:28 -05:00 • committed by GitHub
parent 37899187c6
commit e2552b9add
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
crates/collab/src/rpc.rs
View file

@ -4030,12 +4030,15 @@ async fn get_llm_api_token(
Err(anyhow!("terms of service not accepted"))? Err(anyhow!("terms of service not accepted"))?
} }
let mut account_created_at = user.created_at; let has_llm_subscription = session.has_llm_subscription(&db).await?;
if let Some(github_created_at) = user.github_user_created_at { if !has_llm_subscription {
account_created_at = account_created_at.min(github_created_at); let mut account_created_at = user.created_at;
} if let Some(github_created_at) = user.github_user_created_at {
if Utc::now().naive_utc() - account_created_at < MIN_ACCOUNT_AGE_FOR_LLM_USE { account_created_at = account_created_at.min(github_created_at);
Err(anyhow!("account too young"))? }
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?; let billing_preferences = db.get_billing_preferences(user.id).await?;
@ -4045,7 +4048,7 @@ async fn get_llm_api_token(
session.is_staff(), session.is_staff(),
billing_preferences, billing_preferences,
has_llm_closed_beta_feature_flag, has_llm_closed_beta_feature_flag,
session.has_llm_subscription(&db).await?, has_llm_subscription,
session.current_plan(&db).await?, session.current_plan(&db).await?,
&session.app_state.config, &session.app_state.config,
)?; )?;