diff --git a/LLama/LLamaEmbedder.cs b/LLama/LLamaEmbedder.cs index 13a3e1c2..c29b6b25 100644 --- a/LLama/LLamaEmbedder.cs +++ b/LLama/LLamaEmbedder.cs @@ -97,9 +97,13 @@ namespace LLama private float[] GetEmbeddingsArray() { - var embeddings = NativeApi.llama_get_embeddings_seq(Context.NativeHandle, LLamaSeqId.Zero); - if (embeddings.Length == 0) - return Array.Empty(); + var embeddings = NativeApi.llama_get_embeddings(Context.NativeHandle); + if (embeddings == null || embeddings.Length == 0) + { + embeddings = NativeApi.llama_get_embeddings_seq(Context.NativeHandle, LLamaSeqId.Zero); + if (embeddings == null || embeddings.Length == 0) + return Array.Empty(); + } return embeddings.ToArray(); }