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(
26                "functions",
27                include_str!("../schema/functions.sql"),
28            ),
29        ]
30    }
31
32    fn dependencies(&self) -> Vec<&'static str> {
33        vec![]
34    }
35}
36
37register_extension!(DatabaseExtension);