fix: LlamaSharpEmbeddings: EmbeddingMode must be true exception fixed…

… by passing the embeddingMode condiguration to the LlamaEmbedder (#344)
tryAGI · Jun 14, 2024 · f105e01 · f105e01
1 parent 6c84131
commit f105e01
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/Providers/LLamaSharp/src/LLamaSharpEmbeddings.cs b/src/Providers/LLamaSharp/src/LLamaSharpEmbeddings.cs
@@ -1,5 +1,5 @@
-using LLama.Common;
-using LLama;
+using LLama;
+using LLama.Common;
 
 namespace LangChain.Providers.LLamaSharp;
 
@@ -23,7 +23,8 @@ public static LLamaSharpEmbeddings FromPath(string path, float temperature = 0)
         return new LLamaSharpEmbeddings(new LLamaSharpConfiguration
         {
             PathToModelFile = path,
-            Temperature = temperature
+            Temperature = temperature,
+            EmbeddingMode = true
         });
     }
 
@@ -43,6 +44,7 @@ public LLamaSharpEmbeddings(LLamaSharpConfiguration configuration) : base(id: "L
         {
             ContextSize = (uint)configuration.ContextSize,
             Seed = (uint)configuration.Seed,
+            Embeddings = configuration.EmbeddingMode
         };
         _model = LLamaWeights.LoadFromFile(parameters);
         _configuration = configuration;