06625bfe94
Release Notes: - N/A --------- Co-authored-by: Marshall <marshall@zed.dev> Co-authored-by: Marshall Bowers <elliott.codes@gmail.com>
25 lines
576 B
Rust
25 lines
576 B
Rust
use crate::llm::db::ProviderId;
|
|
use sea_orm::entity::prelude::*;
|
|
|
|
/// An LLM provider.
|
|
#[derive(Clone, Debug, PartialEq, DeriveEntityModel)]
|
|
#[sea_orm(table_name = "providers")]
|
|
pub struct Model {
|
|
#[sea_orm(primary_key)]
|
|
pub id: ProviderId,
|
|
pub name: String,
|
|
}
|
|
|
|
#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
|
|
pub enum Relation {
|
|
#[sea_orm(has_many = "super::model::Entity")]
|
|
Models,
|
|
}
|
|
|
|
impl Related<super::model::Entity> for Entity {
|
|
fn to() -> RelationDef {
|
|
Relation::Models.def()
|
|
}
|
|
}
|
|
|
|
impl ActiveModelBehavior for ActiveModel {}
|