From b8bd7d7eb27aaaadae728599f64a0874f66196ea Mon Sep 17 00:00:00 2001
From: Mohamed Bassem <me@mbassem.com>
Date: Sun, 12 Jan 2025 15:31:30 +0000
Subject: feat: Support customizing the summarization prompt. Fixes #731

---
 packages/shared/prompts.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'packages/shared/prompts.ts')

diff --git a/packages/shared/prompts.ts b/packages/shared/prompts.ts
index 82c9ba6e..40e706b2 100644
--- a/packages/shared/prompts.ts
+++ b/packages/shared/prompts.ts
@@ -53,11 +53,15 @@ You must respond in JSON with the key "tags" and the value is an array of string
 
 export function buildSummaryPrompt(
   lang: string,
+  customPrompts: string[],
   content: string,
   contextLength: number,
 ) {
   const constructPrompt = (c: string) => `
-    Summarize the following content in 3-4 sentences in ${lang}, responding ONLY with the summary:
+    Summarize the following content responding ONLY with the summary. You MUST follow the following rules:
+- Summary must be in 3-4 sentences.
+- The summary language must be in ${lang}.
+${customPrompts && customPrompts.map((p) => `- ${p}`).join("\n")}
     ${c}`;
 
   const promptSize = calculateNumTokens(constructPrompt(""));
-- 
cgit v1.2.3-70-g09d2