[FIX] Fixed issue in prompt construction (#767)

Fixed issue in prompt construction Signed-off-by: Deepak <[email protected]>
Zipstack · Oct 10, 2024 · 4e59cb3 · 4e59cb3
1 parent 822dc36
commit 4e59cb3
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/prompt-service/src/unstract/prompt_service/helper.py b/prompt-service/src/unstract/prompt_service/helper.py
@@ -287,10 +287,7 @@ def construct_prompt(
     context: str,
     platform_postamble: str,
 ) -> str:
-    prompt = (
-        f"{preamble}\n\nContext:\n---------------\n{context}\n"
-        f"-----------------\n\nQuestion or Instruction: {prompt}\n"
-    )
+    prompt = f"{preamble}\n\nQuestion or Instruction: {prompt}"
     if grammar_list is not None and len(grammar_list) > 0:
         prompt += "\n"
         for grammar in grammar_list:
@@ -305,7 +302,10 @@ def construct_prompt(
                     {", ".join(synonyms)} in both the quesiton and the context.'  # noqa
     if platform_postamble:
         platform_postamble += "\n\n"
-    prompt += f"\n\n{postamble}\n\n{platform_postamble}Answer:"
+    prompt += (
+        f"\n\n{postamble}\n\nContext:\n---------------\n{context}\n"
+        f"-----------------\n\n{platform_postamble}Answer:"
+    )
     return prompt