systemprompt-agent 0.2.0

Core Agent protocol module for systemprompt.io
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use systemprompt_extension::prelude::*;

#[derive(Debug, Clone, Copy, Default)]
pub struct AgentExtension;

impl Extension for AgentExtension {
    fn metadata(&self) -> ExtensionMetadata {
        ExtensionMetadata {
            id: "agent",
            name: "Agent",
            version: env!("CARGO_PKG_VERSION"),
        }
    }

    fn migration_weight(&self) -> u32 {
        40
    }

    fn schemas(&self) -> Vec<SchemaDefinition> {
        vec![
            SchemaDefinition::inline("user_contexts", include_str!("../schema/user_contexts.sql"))
                .with_required_columns(vec![
                    "context_id".into(),
                    "user_id".into(),
                    "created_at".into(),
                ]),
            SchemaDefinition::inline("agent_tasks", include_str!("../schema/agent_tasks.sql"))
                .with_required_columns(vec![
                    "task_id".into(),
                    "context_id".into(),
                    "status".into(),
                    "created_at".into(),
                ]),
            SchemaDefinition::inline("task_messages", include_str!("../schema/task_messages.sql"))
                .with_required_columns(vec![
                    "id".into(),
                    "task_id".into(),
                    "role".into(),
                    "created_at".into(),
                ]),
            SchemaDefinition::inline("message_parts", include_str!("../schema/message_parts.sql"))
                .with_required_columns(vec!["id".into(), "message_id".into(), "part_kind".into()]),
            SchemaDefinition::inline(
                "task_artifacts",
                include_str!("../schema/task_artifacts.sql"),
            )
            .with_required_columns(vec![
                "id".into(),
                "task_id".into(),
                "artifact_id".into(),
            ]),
            SchemaDefinition::inline(
                "artifact_parts",
                include_str!("../schema/artifact_parts.sql"),
            )
            .with_required_columns(vec![
                "id".into(),
                "artifact_id".into(),
                "part_kind".into(),
            ]),
            SchemaDefinition::inline(
                "context_agents",
                include_str!("../schema/context_agents.sql"),
            )
            .with_required_columns(vec![
                "id".into(),
                "context_id".into(),
                "agent_name".into(),
            ]),
            SchemaDefinition::inline(
                "context_notifications",
                include_str!("../schema/context_notifications.sql"),
            )
            .with_required_columns(vec![
                "id".into(),
                "context_id".into(),
                "notification_type".into(),
            ]),
            SchemaDefinition::inline(
                "task_push_notification_configs",
                include_str!("../schema/task_push_notification_configs.sql"),
            )
            .with_required_columns(vec!["id".into(), "task_id".into()]),
            SchemaDefinition::inline("agents", include_str!("../schema/agents.sql"))
                .with_required_columns(vec!["agent_id".into(), "name".into(), "enabled".into()]),
            SchemaDefinition::inline("agent_skills", include_str!("../schema/agent_skills.sql"))
                .with_required_columns(vec!["skill_id".into(), "name".into()]),
            SchemaDefinition::inline(
                "task_execution_steps",
                include_str!("../schema/task_execution_steps.sql"),
            )
            .with_required_columns(vec![
                "step_id".into(),
                "task_id".into(),
                "step_type".into(),
            ]),
            SchemaDefinition::inline("services", include_str!("../schema/services.sql"))
                .with_required_columns(vec!["name".into(), "module_name".into(), "status".into()]),
            SchemaDefinition::inline(
                "user_session_analytics",
                include_str!("../schema/user_session_analytics.sql"),
            ),
        ]
    }

    fn dependencies(&self) -> Vec<&'static str> {
        vec!["users", "oauth", "mcp"]
    }

    fn migrations(&self) -> Vec<Migration> {
        vec![
            Migration::new(
                1,
                "drop_playbooks_table",
                include_str!("../schema/migrations/001_drop_playbooks.sql"),
            ),
            Migration::new(
                2,
                "add_server_type_column",
                include_str!("../schema/migrations/002_add_server_type.sql"),
            ),
        ]
    }
}

register_extension!(AgentExtension);