bitrouter 1.0.0-alpha.28

BitRouter: an LLM API router. CLI + assembly library.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100

//! Bind eval exchange rows to the authenticated owning user.

use sea_orm_migration::prelude::*;

#[derive(DeriveMigrationName)]
pub struct Migration;

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        for table in ["eval_subjects", "eval_results", "eval_snapshots"] {
            manager
                .alter_table(
                    Table::alter()
                        .table(Alias::new(table))
                        .add_column(
                            ColumnDef::new(Alias::new("owner_user_id"))
                                .string()
                                .not_null()
                                .default("local"),
                        )
                        .to_owned(),
                )
                .await?;
        }
        manager
            .create_index(
                Index::create()
                    .if_not_exists()
                    .name("idx_eval_subjects_owner")
                    .table(Alias::new("eval_subjects"))
                    .col(Alias::new("owner_user_id"))
                    .to_owned(),
            )
            .await?;
        manager
            .create_index(
                Index::create()
                    .if_not_exists()
                    .name("idx_eval_results_owner")
                    .table(Alias::new("eval_results"))
                    .col(Alias::new("owner_user_id"))
                    .to_owned(),
            )
            .await?;
        manager
            .create_index(
                Index::create()
                    .if_not_exists()
                    .name("idx_eval_results_owner_idempotency")
                    .table(Alias::new("eval_results"))
                    .col(Alias::new("owner_user_id"))
                    .col(Alias::new("idempotency_key"))
                    .unique()
                    .to_owned(),
            )
            .await?;
        manager
            .create_index(
                Index::create()
                    .if_not_exists()
                    .name("idx_eval_snapshots_owner")
                    .table(Alias::new("eval_snapshots"))
                    .col(Alias::new("owner_user_id"))
                    .to_owned(),
            )
            .await?;
        Ok(())
    }

    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        for (table, index) in [
            ("eval_snapshots", "idx_eval_snapshots_owner"),
            ("eval_results", "idx_eval_results_owner_idempotency"),
            ("eval_results", "idx_eval_results_owner"),
            ("eval_subjects", "idx_eval_subjects_owner"),
        ] {
            manager
                .drop_index(
                    Index::drop()
                        .if_exists()
                        .name(index)
                        .table(Alias::new(table))
                        .to_owned(),
                )
                .await?;
        }
        for table in ["eval_snapshots", "eval_results", "eval_subjects"] {
            manager
                .alter_table(
                    Table::alter()
                        .table(Alias::new(table))
                        .drop_column(Alias::new("owner_user_id"))
                        .to_owned(),
                )
                .await?;
        }
        Ok(())
    }
}