Struct DocumentTree 

pub struct DocumentTree { /* private fields */ }

Hierarchical document tree for multi-level summarization

Implementations

impl DocumentTree

pub fn new(document_id: DocumentId, config: HierarchicalConfig) -> Result<Self>

Create a new document tree

pub fn with_llm_client( document_id: DocumentId, config: HierarchicalConfig, llm_client: Arc<dyn LLMClient>, ) -> Result<Self>

Create a new document tree with LLM client

pub fn set_llm_client(&mut self, llm_client: Option<Arc<dyn LLMClient>>)

Set LLM client for the tree

pub fn with_parallel_processing( document_id: DocumentId, config: HierarchicalConfig, _parallel_processor: ParallelProcessor, ) -> Result<Self>

Create a new document tree with parallel processing support

pub fn with_parallel_and_llm( document_id: DocumentId, config: HierarchicalConfig, _parallel_processor: ParallelProcessor, llm_client: Arc<dyn LLMClient>, ) -> Result<Self>

Create a new document tree with both parallel processing and LLM client

pub async fn build_from_chunks(&mut self, chunks: Vec<TextChunk>) -> Result<()>

Build the hierarchical tree from text chunks

pub async fn generate_llm_summary( &self, text: &str, level: usize, context: &str, ) -> Result<String>

Generate summary suitable for parallel processing Generate summary using LLM for the given text and level

pub async fn generate_llm_summaries_batch( &self, texts: &[(&str, usize, &str)], ) -> Result<Vec<String>>

Generate summaries in batch for multiple texts

pub fn query(&self, query: &str, max_results: usize) -> Result<Vec<QueryResult>>

Query the tree for relevant nodes at different levels

pub fn get_ancestors(&self, node_id: &NodeId) -> Vec<&TreeNode>

Get ancestors of a node (path to root)

pub fn get_descendants(&self, node_id: &NodeId) -> Vec<&TreeNode>

Get descendants of a node (all children recursively)

pub fn get_node(&self, node_id: &NodeId) -> Option<&TreeNode>

Get a node by ID

pub fn get_level_nodes(&self, level: usize) -> Vec<&TreeNode>

Get all nodes at a specific level

pub fn get_root_nodes(&self) -> Vec<&TreeNode>

Get the root nodes of the tree

pub fn document_id(&self) -> &DocumentId

Get the document ID

pub fn get_statistics(&self) -> TreeStatistics

Get tree statistics

pub fn to_json(&self) -> Result<String>

Serialize the tree to JSON format

pub fn from_json(json_str: &str) -> Result<Self>

Load tree from JSON format