testutil

type MockDSUPersistence struct {
	LoadFunc func() (*disjoint_set.DSU, error)
	SaveFunc func(dsu *disjoint_set.DSU) error

	SaveCount int
	LastDSU   *disjoint_set.DSU
	// contains filtered or unexported fields
}

type MockEmbeddingClient struct {
	GenerateEmbeddingFunc func(ctx context.Context, text string) ([]float32, error)

	CallCount int
	LastText  string
	// contains filtered or unexported fields
}

type MockLLMClient struct {
	ClassifyFunc func(ctx context.Context, text string) (string, error)

	CallCount int
	LastText  string
	// contains filtered or unexported fields
}

type MockVectorClient struct {
	SearchFunc func(ctx context.Context, vector []float32, topK int) ([]types.VectorMatch, error)
	UpsertFunc func(ctx context.Context, id string, vector []float32, metadata map[string]any) error

	CallCount   int
	UpsertCount int
	Storage     map[string]struct {
		Vector   []float32
		Metadata map[string]any
	}
	// contains filtered or unexported fields
}