Skip to content
8 changes: 8 additions & 0 deletions cli/src/cli_schema.rs
Original file line number Diff line number Diff line change
Expand Up @@ -253,6 +253,14 @@ pub enum TraceSubcommand {
#[arg(long, value_enum, default_value_t = OutputFormat::Text)]
format: OutputFormat,
},

#[command(
about = "Synchronize the current repository's Agent Trace database with the control plane"
)]
Sync {
#[arg(long, value_enum, default_value_t = OutputFormat::Text)]
format: OutputFormat,
},
}

#[derive(Subcommand, Debug, Clone, PartialEq, Eq)]
Expand Down
8 changes: 7 additions & 1 deletion cli/src/services/agent_trace_export/mod.rs
Original file line number Diff line number Diff line change
Expand Up @@ -712,7 +712,13 @@ mod tests {
let db_path = unique_test_db_path("messages-limit");
let db = RepositoryAgentTraceDb::new_at(&db_path).expect("test DB should open");
for id in 1..=10 {
insert_message_row_with_id(&db, id, &format!("msg-{id}"));
db.insert_message(InsertMessageInsert {
session_id: "sess-1".to_string(),
message_id: format!("msg-{id}"),
role: MessageRole::Assistant,
generated_at_unix_ms: 1_000 + id,
})
.expect("seed message should insert");
}

let reader = AgentTraceExportReader::new(&db);
Expand Down
Loading
Loading