Function newKnowledgeTestCLI

cli/knowledge_adapter_test.go:22–49

newKnowledgeTestCLI builds a minimal ChatCLI whose context manager lives under a throwaway HOME, with one knowledge base created from a docs-flatten JSONL corpus and attached to the default session.

(t *testing.T)

// under a throwaway HOME, with one knowledge base created from a docs-flatten
// JSONL corpus and attached to the default session.
func newKnowledgeTestCLI(t *testing.T) *ChatCLI {
	t.Helper()
	t.Setenv("HOME", t.TempDir())

	handler, err := NewContextHandler(zap.NewNop())
	if err != nil {
		t.Fatalf("NewContextHandler: %v", err)
	}
	handler.GetManager().AttachEmbeddingProvider(embedding.NewNull())

	corpus := filepath.Join(t.TempDir(), "docs.jsonl")
	lines := strings.Join([]string{
		`{"id":"guide/install.md#0001","source":"guide/install.md","content":"# Install\nHow to install the CLI."}`,
		`{"id":"guide/install.md#0002","source":"guide/install.md","content":"## Homebrew\nbrew install chatcli"}`,
		`{"id":"api/auth.md#0001","source":"api/auth.md","content":"# Auth\noauth login bearer token flows"}`,
	}, "\n")
	if err := os.WriteFile(corpus, []byte(lines), 0o600); err != nil {
		t.Fatal(err)
	}
	fc, err := handler.GetManager().CreateContext(context.Background(), "chatcli-docs", "test corpus", []string{corpus}, "knowledge", nil, false)
	if err != nil {
		t.Fatalf("CreateContext: %v", err)
	}
	if err := handler.GetManager().AttachContext("default", fc.ID, 1); err != nil {
		t.Fatalf("AttachContext: %v", err)
	}
	return &ChatCLI{contextHandler: handler, logger: zap.NewNop()}
}

func TestKnowledgeAdapter_EndToEnd(t *testing.T) {
	cli := newKnowledgeTestCLI(t)