introduced context in query

alkampfergit · Nov 24, 2024 · c4d1d42 · c4d1d42
1 parent 477c591
commit c4d1d42
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/KernelMemory.Extensions.ConsoleTest/Samples/CustomSearchPipelineBase.cs b/src/KernelMemory.Extensions.ConsoleTest/Samples/CustomSearchPipelineBase.cs
@@ -316,7 +316,7 @@ private static IKernelMemoryBuilder CreateBasicKernelMemoryBuilder(
         services.AddSingleton<HyDeQueryHandler>();
         var hydeConfig = new HiDeQueryHandlerConfiguration()
         {
-            Prompt = "Given a question, generate a paragraph of text that answers the question in the context of computer security and IT security"
+            Prompt = "Given a question, generate a paragraph of text that answers the question"
         };
         services.AddSingleton(hydeConfig);
         services.AddSingleton<KeywordSearchQueryHandler>();

diff --git a/src/KernelMemory.Extensions/QueryPipeline/HyDeQueryHandler.cs b/src/KernelMemory.Extensions/QueryPipeline/HyDeQueryHandler.cs
@@ -47,6 +47,11 @@ protected override async Task OnHandleAsync(UserQuestion userQuestion, Cancellat
         // Perform a vector search in default memory
         StringBuilder prompt = new StringBuilder(_configuration.Prompt.Length + userQuestion.Question.Length + "Question: ".Length + "Paragraph: ".Length + 20);
         prompt.AppendLine(_configuration.Prompt);
+        if (!string.IsNullOrEmpty(userQuestion.Context))
+        {
+            prompt.AppendLine("The context of the question is: " + userQuestion.Context);
+        }
+
         prompt.AppendLine("Question: " + userQuestion.Question);
         prompt.AppendLine("Paragraph: ");