Function assemble_caps_items_per_source

core/src/context/assembler.rs:435–466

()

    #[test]
    fn assemble_caps_items_per_source() {
        let assembler = ContextAssembler::new(ContextBudget {
            max_items: 10,
            max_tokens: 100,
        })
        .with_source_policy(ContextSourcePolicy {
            max_items_per_source: Some(2),
            max_tokens_per_source: None,
        });
        let assembly = assembler.assemble(&[result(
            "test",
            vec![
                ContextItem::new("a", ContextType::Resource, "a")
                    .with_source("file://a")
                    .with_relevance(0.9)
                    .with_token_count(1),
                ContextItem::new("b", ContextType::Resource, "b")
                    .with_source("file://b")
                    .with_relevance(0.8)
                    .with_token_count(1),
                ContextItem::new("c", ContextType::Resource, "c")
                    .with_source("file://c")
                    .with_relevance(0.7)
                    .with_token_count(1),
            ],
        )]);

        assert_eq!(assembly.items.len(), 2);
        assert_eq!(assembly.items[0].id, "a");
        assert_eq!(assembly.items[1].id, "b");
        assert!(assembly.truncated);
    }

    #[test]
    fn assemble_caps_tokens_per_source_but_keeps_other_sources() {