diff --git a/crates/collab/migrations_llm/20240809130000_change_rate_limit_columns_to_bigint.sql b/crates/collab/migrations_llm/20240809130000_change_rate_limit_columns_to_bigint.sql new file mode 100644 index 0000000000000000000000000000000000000000..f1def8209a742743a6c708365afc05f7ab911e18 --- /dev/null +++ b/crates/collab/migrations_llm/20240809130000_change_rate_limit_columns_to_bigint.sql @@ -0,0 +1,4 @@ +ALTER TABLE models + ALTER COLUMN max_requests_per_minute TYPE bigint, + ALTER COLUMN max_tokens_per_minute TYPE bigint, + ALTER COLUMN max_tokens_per_day TYPE bigint; diff --git a/crates/collab/src/llm/db/queries/providers.rs b/crates/collab/src/llm/db/queries/providers.rs index 975bf607ce71d0cea48a8c42a0ba9c99151e5ce4..c53b6aff7f785085ace41f59ff9a7a1ea5d5d5b5 100644 --- a/crates/collab/src/llm/db/queries/providers.rs +++ b/crates/collab/src/llm/db/queries/providers.rs @@ -4,9 +4,9 @@ use std::str::FromStr; use strum::IntoEnumIterator as _; pub struct ModelRateLimits { - pub max_requests_per_minute: i32, - pub max_tokens_per_minute: i32, - pub max_tokens_per_day: i32, + pub max_requests_per_minute: i64, + pub max_tokens_per_minute: i64, + pub max_tokens_per_day: i64, } impl LlmDatabase { diff --git a/crates/collab/src/llm/db/tables/model.rs b/crates/collab/src/llm/db/tables/model.rs index eb07ab9473dfe10e2eae9036a754c9536b55eedd..ea8bd5d456e51efd38b8c4e964c8c9ad6efe209c 100644 --- a/crates/collab/src/llm/db/tables/model.rs +++ b/crates/collab/src/llm/db/tables/model.rs @@ -10,9 +10,9 @@ pub struct Model { pub id: ModelId, pub provider_id: ProviderId, pub name: String, - pub max_requests_per_minute: i32, - pub max_tokens_per_minute: i32, - pub max_tokens_per_day: i32, + pub max_requests_per_minute: i64, + pub max_tokens_per_minute: i64, + pub max_tokens_per_day: i64, } #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]