Skip to main content

systemprompt_database/
extension.rs

1use systemprompt_extension::prelude::*;
2
3#[derive(Debug, Clone, Copy, Default)]
4pub struct DatabaseExtension;
5
6impl Extension for DatabaseExtension {
7    fn metadata(&self) -> ExtensionMetadata {
8        ExtensionMetadata {
9            id: "database",
10            name: "Database",
11            version: env!("CARGO_PKG_VERSION"),
12        }
13    }
14
15    fn migration_weight(&self) -> u32 {
16        1
17    }
18
19    fn schemas(&self) -> Vec<SchemaDefinition> {
20        vec![
21            SchemaDefinition::inline(
22                "extension_migrations",
23                include_str!("../schema/extension_migrations.sql"),
24            ),
25            SchemaDefinition::inline("functions", include_str!("../schema/functions.sql")),
26        ]
27    }
28
29    fn dependencies(&self) -> Vec<&'static str> {
30        vec![]
31    }
32}
33
34register_extension!(DatabaseExtension);