diff --git a/LLama.Examples/NewVersion/KernelMemory.cs b/LLama.Examples/NewVersion/KernelMemory.cs
index 3358f3f0..c647d716 100644
--- a/LLama.Examples/NewVersion/KernelMemory.cs
+++ b/LLama.Examples/NewVersion/KernelMemory.cs
@@ -1,11 +1,11 @@
-using Microsoft.SemanticMemory.Handlers;
-using Microsoft.SemanticMemory;
-using System;
+using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using LLamaSharp.KernelMemory;
+using Microsoft.KernelMemory;
+using Microsoft.KernelMemory.Handlers;
namespace LLama.Examples.NewVersion
{
@@ -16,7 +16,7 @@ namespace LLama.Examples.NewVersion
Console.WriteLine("Example from: https://github.com/microsoft/kernel-memory/blob/main/examples/101-using-core-nuget/Program.cs");
Console.Write("Please input your model path: ");
var modelPath = Console.ReadLine();
- var memory = new MemoryClientBuilder()
+ var memory = new KernelMemoryBuilder()
.WithLLamaSharpDefaults(new LLamaSharpConfig(modelPath))
.With(new TextPartitioningOptions
{
diff --git a/LLama.KernelMemory/BuilderExtensions.cs b/LLama.KernelMemory/BuilderExtensions.cs
index 8f7d6a7f..bd0c97f5 100644
--- a/LLama.KernelMemory/BuilderExtensions.cs
+++ b/LLama.KernelMemory/BuilderExtensions.cs
@@ -1,4 +1,4 @@
-using Microsoft.SemanticMemory;
+using Microsoft.KernelMemory;
using System;
using System.Collections.Generic;
using System.Linq;
@@ -18,7 +18,7 @@ namespace LLamaSharp.KernelMemory
/// The MemoryClientBuilder instance.
/// The LLamaSharpConfig instance.
/// The MemoryClientBuilder instance with LLamaSharpTextEmbeddingGeneration added.
- public static MemoryClientBuilder WithLLamaSharpTextEmbeddingGeneration(this MemoryClientBuilder builder, LLamaSharpConfig config)
+ public static KernelMemoryBuilder WithLLamaSharpTextEmbeddingGeneration(this KernelMemoryBuilder builder, LLamaSharpConfig config)
{
builder.WithCustomEmbeddingGeneration(new LLamaSharpTextEmbeddingGeneration(config));
return builder;
@@ -30,7 +30,7 @@ namespace LLamaSharp.KernelMemory
/// The MemoryClientBuilder instance.
/// The LLamaSharpConfig instance.
/// The MemoryClientBuilder instance with LLamaSharpTextGeneration added.
- public static MemoryClientBuilder WithLLamaSharpTextGeneration(this MemoryClientBuilder builder, LLamaSharpConfig config)
+ public static KernelMemoryBuilder WithLLamaSharpTextGeneration(this KernelMemoryBuilder builder, LLamaSharpConfig config)
{
builder.WithCustomTextGeneration(new LlamaSharpTextGeneration(config));
return builder;
@@ -42,7 +42,7 @@ namespace LLamaSharp.KernelMemory
/// The MemoryClientBuilder instance.
/// The LLamaSharpConfig instance.
/// The MemoryClientBuilder instance with LLamaSharpTextEmbeddingGeneration and LLamaSharpTextGeneration added.
- public static MemoryClientBuilder WithLLamaSharpDefaults(this MemoryClientBuilder builder, LLamaSharpConfig config)
+ public static KernelMemoryBuilder WithLLamaSharpDefaults(this KernelMemoryBuilder builder, LLamaSharpConfig config)
{
builder.WithLLamaSharpTextEmbeddingGeneration(config);
builder.WithLLamaSharpTextGeneration(config);
diff --git a/LLama.KernelMemory/LLamaSharp.KernelMemory.csproj b/LLama.KernelMemory/LLamaSharp.KernelMemory.csproj
index 1575e3eb..54766b02 100644
--- a/LLama.KernelMemory/LLamaSharp.KernelMemory.csproj
+++ b/LLama.KernelMemory/LLamaSharp.KernelMemory.csproj
@@ -7,7 +7,7 @@
-
+
diff --git a/LLama.KernelMemory/LlamaSharpTextGeneration.cs b/LLama.KernelMemory/LlamaSharpTextGeneration.cs
index 9030861d..e0cdaa8c 100644
--- a/LLama.KernelMemory/LlamaSharpTextGeneration.cs
+++ b/LLama.KernelMemory/LlamaSharpTextGeneration.cs
@@ -1,6 +1,6 @@
using LLama;
using LLama.Common;
-using Microsoft.SemanticMemory.AI;
+using Microsoft.KernelMemory.AI;
using System;
using System.Collections.Generic;
using System.Linq;