systemprompt_logging/
lib.rs

1pub mod extension;
2pub mod layer;
3pub mod models;
4pub mod repository;
5pub mod services;
6pub mod trace;
7
8pub use extension::LoggingExtension;
9
10pub use layer::DatabaseLayer;
11pub use models::{LogEntry, LogFilter, LogLevel};
12pub use repository::{AnalyticsEvent, AnalyticsRepository, LoggingRepository};
13pub use services::{
14    is_startup_mode, publish_log, set_log_publisher, set_startup_mode, CliService,
15    DatabaseLogService, FilterSystemFields, LoggingMaintenanceService, RequestSpan,
16    RequestSpanBuilder, SystemSpan,
17};
18pub use trace::{
19    AiRequestInfo, AiRequestSummary, AiTraceService, ConversationMessage, ExecutionStep,
20    ExecutionStepSummary, McpExecutionSummary, McpToolExecution, TaskArtifact, TaskInfo,
21    ToolLogEntry, TraceEvent, TraceQueryService,
22};
23
24use std::sync::OnceLock;
25
26use systemprompt_database::DbPool;
27use tracing_subscriber::layer::SubscriberExt;
28use tracing_subscriber::util::SubscriberInitExt;
29use tracing_subscriber::{EnvFilter, Layer};
30
31static LOGGING_INITIALIZED: OnceLock<()> = OnceLock::new();
32
33const NOISE_FILTERS: &[&str] = &[
34    "tokio_cron_scheduler=warn",
35    "sqlx::postgres::notice=warn",
36    "sqlx::query=warn",
37    "handlebars=warn",
38    "systemprompt_database::lifecycle=info",
39    "systemprompt_templates=info",
40    "systemprompt_extension::registry=info",
41];
42
43fn build_filter(base: &str) -> EnvFilter {
44    let filter_str = std::iter::once(base.to_string())
45        .chain(NOISE_FILTERS.iter().map(ToString::to_string))
46        .collect::<Vec<_>>()
47        .join(",");
48    EnvFilter::new(filter_str)
49}
50
51pub fn init_logging(db_pool: DbPool) {
52    if LOGGING_INITIALIZED.set(()).is_err() {
53        return;
54    }
55
56    let console_filter = if is_startup_mode() {
57        EnvFilter::new("warn")
58    } else {
59        EnvFilter::try_from_default_env().unwrap_or_else(|_| build_filter("info"))
60    };
61
62    let fmt_layer = tracing_subscriber::fmt::layer()
63        .fmt_fields(FilterSystemFields::new())
64        .with_target(true)
65        .with_writer(std::io::stderr)
66        .with_filter(console_filter);
67
68    let db_layer =
69        DatabaseLayer::new(db_pool).with_filter(tracing_subscriber::filter::LevelFilter::INFO);
70
71    tracing_subscriber::registry()
72        .with(fmt_layer)
73        .with(db_layer)
74        .init();
75}
76
77pub fn init_console_logging() {
78    init_console_logging_with_level(None);
79}
80
81pub fn init_console_logging_with_level(level: Option<&str>) {
82    if LOGGING_INITIALIZED.set(()).is_err() {
83        return;
84    }
85
86    let env_filter = EnvFilter::try_from_default_env()
87        .unwrap_or_else(|_| build_filter(level.unwrap_or("info")));
88
89    tracing_subscriber::fmt().with_env_filter(env_filter).init();
90}
systemprompt_logging/lib.rs

systemprompt_logging/
lib.rs
