Skip to main content

systemprompt_agent/repository/context/
mod.rs

1pub 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}