async Task <TResult> OnEmbedding <TResult>( EmbeddingId embedding, Func <IMongoCollection <Embedding>, Task <TResult> > callback, CancellationToken token) => await callback(await _embeddings.GetStates(embedding, token).ConfigureAwait(false)) .ConfigureAwait(false);