systemprompt_agent/repository/context/
mod.rs1pub mod message;
2mod mutations;
3mod queries;
4
5use sqlx::PgPool;
6use std::sync::Arc;
7use systemprompt_database::DbPool;
8
9#[derive(Debug, Clone)]
10pub struct ContextRepository {
11 pool: Arc<PgPool>,
12 write_pool: Arc<PgPool>,
13 db_pool: DbPool,
14}
15
16impl ContextRepository {
17 pub fn new(db: &DbPool) -> anyhow::Result<Self> {
18 let pool = db.pool_arc()?;
19 let write_pool = db.write_pool_arc()?;
20 Ok(Self {
21 pool,
22 write_pool,
23 db_pool: db.clone(),
24 })
25 }
26}