Skip to main content

agentforge_db/
lib.rs

1pub mod agent_repo;
2pub mod benchmark_repo;
3pub mod eval_repo;
4pub mod finetune_repo;
5pub mod pool;
6pub mod scenario_repo;
7pub mod shadow_repo;
8pub mod trace_repo;
9
10pub use agent_repo::AgentRepo;
11pub use benchmark_repo::BenchmarkRepo;
12pub use eval_repo::EvalRepo;
13pub use finetune_repo::FineTuneRepo;
14pub use pool::*;
15pub use scenario_repo::ScenarioRepo;
16pub use shadow_repo::ShadowRepo;
17pub use trace_repo::TraceRepo;
18
19use agentforge_core::AgentForgeError;
20
21/// Convert sqlx errors to AgentForge errors.
22pub(crate) fn db_err(e: sqlx::Error) -> AgentForgeError {
23    AgentForgeError::DatabaseError(e.to_string())
24}