style(memory): cargo fmt on source_tree module (#709)

CI rustfmt caught a few multi-line function signatures and import
groups that rustfmt wants collapsed / reordered. No behaviour change.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
This commit is contained in:
sanil-23
2026-04-22 20:40:03 +05:30
co-authored by Claude Opus 4.7
parent 8e55ddd71e
commit b7395a85c6
4 changed files with 8 additions and 30 deletions
@@ -25,7 +25,7 @@ use crate::openhuman::config::Config;
use crate::openhuman::memory::tree::source_tree::registry::new_summary_id;
use crate::openhuman::memory::tree::source_tree::store;
use crate::openhuman::memory::tree::source_tree::summariser::{
SummaryContext, SummaryInput, Summariser,
Summariser, SummaryContext, SummaryInput,
};
use crate::openhuman::memory::tree::source_tree::types::{
Buffer, SummaryNode, Tree, TreeKind, TOKEN_BUDGET,
@@ -333,11 +333,7 @@ fn refresh_last_sealed_tx(
/// Fetch contributions for `item_ids`. At level 0 we pull from
/// `mem_tree_chunks` + `mem_tree_score`; at ≥1 we pull from
/// `mem_tree_summaries`.
fn hydrate_inputs(
config: &Config,
level: u32,
item_ids: &[String],
) -> Result<Vec<SummaryInput>> {
fn hydrate_inputs(config: &Config, level: u32, item_ids: &[String]) -> Result<Vec<SummaryInput>> {
if level == 0 {
hydrate_leaf_inputs(config, item_ids)
} else {
@@ -87,9 +87,7 @@ mod tests {
use crate::openhuman::memory::tree::source_tree::registry::get_or_create_source_tree;
use crate::openhuman::memory::tree::source_tree::summariser::inert::InertSummariser;
use crate::openhuman::memory::tree::store::upsert_chunks;
use crate::openhuman::memory::tree::types::{
chunk_id, Chunk, Metadata, SourceKind, SourceRef,
};
use crate::openhuman::memory::tree::types::{chunk_id, Chunk, Metadata, SourceKind, SourceRef};
use tempfile::TempDir;
fn test_config() -> (TempDir, Config) {
+4 -20
View File
@@ -60,11 +60,7 @@ pub(crate) fn insert_tree_conn(conn: &Connection, tree: &Tree) -> Result<()> {
}
/// Fetch a tree by `(kind, scope)`. Returns `None` if no such tree exists.
pub fn get_tree_by_scope(
config: &Config,
kind: TreeKind,
scope: &str,
) -> Result<Option<Tree>> {
pub fn get_tree_by_scope(config: &Config, kind: TreeKind, scope: &str) -> Result<Option<Tree>> {
with_connection(config, |conn| get_tree_by_scope_conn(conn, kind, scope))
}
@@ -114,12 +110,7 @@ pub(crate) fn update_tree_after_seal_tx(
max_level = ?2,
last_sealed_at_ms = ?3
WHERE id = ?4",
params![
root_id,
max_level,
sealed_at.timestamp_millis(),
tree_id,
],
params![root_id, max_level, sealed_at.timestamp_millis(), tree_id,],
)
.with_context(|| format!("Failed to update tree {tree_id} after seal"))?;
Ok(())
@@ -306,11 +297,7 @@ pub fn get_buffer(config: &Config, tree_id: &str, level: u32) -> Result<Buffer>
with_connection(config, |conn| get_buffer_conn(conn, tree_id, level))
}
pub(crate) fn get_buffer_conn(
conn: &Connection,
tree_id: &str,
level: u32,
) -> Result<Buffer> {
pub(crate) fn get_buffer_conn(conn: &Connection, tree_id: &str, level: u32) -> Result<Buffer> {
let mut stmt = conn.prepare(
"SELECT tree_id, level, item_ids_json, token_sum, oldest_at_ms
FROM mem_tree_buffers WHERE tree_id = ?1 AND level = ?2",
@@ -361,10 +348,7 @@ pub(crate) fn clear_buffer_tx(tx: &Transaction<'_>, tree_id: &str, level: u32) -
/// List all non-empty buffers ordered by `oldest_at_ms ASC`. Used by the
/// time-based flush pass.
pub fn list_stale_buffers(
config: &Config,
older_than: DateTime<Utc>,
) -> Result<Vec<Buffer>> {
pub fn list_stale_buffers(config: &Config, older_than: DateTime<Utc>) -> Result<Vec<Buffer>> {
with_connection(config, |conn| {
let mut stmt = conn.prepare(
"SELECT tree_id, level, item_ids_json, token_sum, oldest_at_ms
@@ -10,7 +10,7 @@ use anyhow::Result;
use async_trait::async_trait;
use crate::openhuman::memory::tree::source_tree::summariser::{
SummaryContext, SummaryInput, SummaryOutput, Summariser,
Summariser, SummaryContext, SummaryInput, SummaryOutput,
};
use crate::openhuman::memory::tree::types::approx_token_count;