systemprompt_agent/repository/task/
task_messages.rs

1use super::TaskRepository;
2use crate::models::a2a::{Message, Part};
3use crate::repository::context::message::{
4    PersistMessageWithTxParams, get_message_parts, get_messages_by_context, get_messages_by_task,
5    get_next_sequence_number, get_next_sequence_number_in_tx, persist_message_with_tx,
6};
7use systemprompt_traits::RepositoryError;
8
9impl TaskRepository {
10    pub async fn get_next_sequence_number(
11        &self,
12        task_id: &systemprompt_identifiers::TaskId,
13    ) -> Result<i32, RepositoryError> {
14        get_next_sequence_number(&self.pool, task_id).await
15    }
16
17    pub async fn get_messages_by_task(
18        &self,
19        task_id: &systemprompt_identifiers::TaskId,
20    ) -> Result<Vec<Message>, RepositoryError> {
21        get_messages_by_task(&self.pool, task_id).await
22    }
23
24    pub async fn get_message_parts(
25        &self,
26        message_id: &systemprompt_identifiers::MessageId,
27    ) -> Result<Vec<Part>, RepositoryError> {
28        get_message_parts(&self.pool, message_id).await
29    }
30
31    pub async fn get_messages_by_context(
32        &self,
33        context_id: &systemprompt_identifiers::ContextId,
34    ) -> Result<Vec<Message>, RepositoryError> {
35        get_messages_by_context(&self.pool, context_id).await
36    }
37
38    pub async fn get_next_sequence_number_in_tx(
39        &self,
40        tx: &mut dyn systemprompt_database::DatabaseTransaction,
41        task_id: &systemprompt_identifiers::TaskId,
42    ) -> Result<i32, RepositoryError> {
43        get_next_sequence_number_in_tx(tx, task_id).await
44    }
45
46    pub async fn persist_message_with_tx(
47        &self,
48        params: PersistMessageWithTxParams<'_>,
49    ) -> Result<(), RepositoryError> {
50        persist_message_with_tx(params).await
51    }
52}
systemprompt_agent/repository/task/task_messages.rs

systemprompt_agent/repository/task/
task_messages.rs
