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 0000000000..f1def8209a --- /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 975bf607ce..c53b6aff7f 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 eb07ab9473..ea8bd5d456 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)]