Improve LLM
This commit is contained in:
parent
02712f41e8
commit
80c093b7ba
16 changed files with 898 additions and 278 deletions
|
|
@ -240,9 +240,11 @@ async fn ensure_user_fields(
|
|||
let has_is_admin = fields.iter().any(|f| f["name"] == "is_admin");
|
||||
let has_subscription = fields.iter().any(|f| f["name"] == "subscription");
|
||||
let has_newsletter = fields.iter().any(|f| f["name"] == "newsletter");
|
||||
let has_ai_tokens_used = fields.iter().any(|f| f["name"] == "ai_tokens_used");
|
||||
let has_ai_tokens_week = fields.iter().any(|f| f["name"] == "ai_tokens_week");
|
||||
|
||||
if has_is_admin && has_subscription && has_newsletter {
|
||||
info!("PocketBase users collection already has is_admin, subscription, and newsletter fields");
|
||||
if has_is_admin && has_subscription && has_newsletter && has_ai_tokens_used && has_ai_tokens_week {
|
||||
info!("PocketBase users collection already has all required fields");
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
|
|
@ -269,6 +271,20 @@ async fn ensure_user_fields(
|
|||
}));
|
||||
}
|
||||
|
||||
if !has_ai_tokens_used {
|
||||
new_fields.push(serde_json::json!({
|
||||
"name": "ai_tokens_used",
|
||||
"type": "number",
|
||||
}));
|
||||
}
|
||||
|
||||
if !has_ai_tokens_week {
|
||||
new_fields.push(serde_json::json!({
|
||||
"name": "ai_tokens_week",
|
||||
"type": "number",
|
||||
}));
|
||||
}
|
||||
|
||||
let patch_resp = client
|
||||
.patch(&url)
|
||||
.header("Authorization", format!("Bearer {token}"))
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue