Function newKnowledgeManager

cli/ctxmgr/knowledge_query_test.go:18–33

newKnowledgeManager builds a manager with one knowledge base attached to session "sess" plus one non-knowledge context that must stay invisible.

(t *testing.T)

// newKnowledgeManager builds a manager with one knowledge base attached to
// session "sess" plus one non-knowledge context that must stay invisible.
func newKnowledgeManager(t *testing.T) (*Manager, *FileContext) {
	t.Helper()
	m := newTestManager(t)
	m.AttachEmbeddingProvider(embedding.NewNull())

	kb := knowledgeTestContext()
	m.contexts[kb.ID] = kb
	if err := m.AttachContext("sess", kb.ID, 1); err != nil {
		t.Fatal(err)
	}
	full := addTestContext(t, m, "plain-ctx", sampleFiles(), ModeFull, nil)
	if err := m.AttachContext("sess", full.ID, 2); err != nil {
		t.Fatal(err)
	}
	return m, kb
}

func TestAttachedKnowledge_FiltersByMode(t *testing.T) {
	m, kb := newKnowledgeManager(t)