From 89217f73caa16786fedacb33a0d441ed905d87ff Mon Sep 17 00:00:00 2001
From: Zoli Somogyi <zsomogyi.be@gmail.com>
Date: Fri, 19 Apr 2024 17:23:44 +0200
Subject: [PATCH] Embeddings correction (#674)

* Embeddings correction
---
 LLama/LLamaEmbedder.cs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

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<float>();
+            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<float>();
+            }
 
             return embeddings.ToArray();
         }