Function test_build_context_string_truncation

atomic-repository/src/ai/mod.rs:623–638

()

    #[test]
    fn test_build_context_string_truncation() {
        // Create enough nodes to exceed MAX_CHARS
        let nodes: Vec<KgNode> = (0..500)
            .map(|i| {
                KgNode::new(
                    format!("entity:{}", i),
                    "entity",
                    format!("entity-{}", i),
                    "graph",
                )
                .with_summary(format!("A fairly long summary for entity number {} to help fill up the buffer quickly and test truncation behaviour", i))
            })
            .collect();
        let context = build_context_string(&nodes, &[], "big query");
        assert!(context.len() <= 4000);
    }

    #[test]
    fn test_build_context_string_groups_by_kind() {