asepharyana
2efd40ca88
- Added module-level documentation for memory tools (`remember`, `recall`, `forget`) to clarify their purpose. - Improved documentation in `recall.rs` and `remember.rs` to describe the functionality and flow of memory entry operations. - Updated `mod.rs` to include descriptions for the tool trait and execution context. - Enhanced `plan.rs` with detailed comments on plan-mode signaling tools. - Documented text search tools in `search.rs` to explain their functionality. - Improved sequential-thinking tool documentation in `seqthink.rs`. - Added safety filter documentation in `shell_filter` for credential and git operations. - Enhanced utility tools documentation, including `cd`, `dir_cache_update`, and `todowrite`. - Improved rendering documentation in view modules (`chat`, `markdown`, `status`, `workflow`) to clarify rendering flows and purposes.
34 lines
1.4 KiB
Rust
34 lines
1.4 KiB
Rust
//! Insert queries against the message log's `messages` table.
|
|
|
|
use rusqlite::{Connection, params};
|
|
use anyhow::Result;
|
|
use crate::dto::chat::message::{ChatMessage, Role};
|
|
|
|
/// Insert a chat message into the session's message log.
|
|
///
|
|
/// Flow: extract optional content/tool_call_id/tool_name → serialize
|
|
/// `tool_calls` to a JSON string if present → map `Role` to its string
|
|
/// column value → `INSERT` the row with the current timestamp.
|
|
///
|
|
/// Return: the new row's `rowid` on success, or the underlying error.
|
|
pub fn insert_message(conn: &Connection, session_id: &str, msg: &ChatMessage) -> Result<i64> {
|
|
let content = msg.content.as_deref();
|
|
let tool_call_id = msg.tool_call_id.as_deref();
|
|
let tool_name = msg.name.as_deref();
|
|
let tool_arguments = msg.tool_calls.as_ref().map(|calls| {
|
|
serde_json::to_string(calls).unwrap_or_default()
|
|
});
|
|
let created_at = chrono::Utc::now().timestamp_millis();
|
|
let role_str = match msg.role {
|
|
Role::User => "user",
|
|
Role::Assistant => "assistant",
|
|
Role::System => "system",
|
|
Role::Tool => "tool",
|
|
};
|
|
conn.execute(
|
|
"INSERT INTO messages (session_id, role, content, tool_call_id, tool_name, tool_arguments, created_at) VALUES (?1, ?2, ?3, ?4, ?5, ?6, ?7)",
|
|
params![session_id, role_str, content, tool_call_id, tool_name, tool_arguments, created_at],
|
|
)?;
|
|
Ok(conn.last_insert_rowid())
|
|
}
|