feat: add summarization prompts (#9)

rpidanny · May 12, 2023 · 693ee2b · 693ee2b
1 parent 24698a8
commit 693ee2b
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 1 deletion.
diff --git a/apps/chrome-extension/src/pages/Content/prompts.ts b/apps/chrome-extension/src/pages/Content/prompts.ts
@@ -1,4 +1,5 @@
 import {
+  BasicSummarizationPrompt,
   CodeErrorHelpPrompt,
   GeneralKnowledgePrompt,
   IPrompt,
@@ -8,6 +9,7 @@ import {
   QuestionWithContextPrompt,
   QuestionWithReasoningPrompt,
   SQLQueryPrompt,
+  SummarizationIntoListPrompt,
   WriteTestsPrompt,
   ZeroShotCoTAPEPrompt,
   ZeroShotCoTPrompt,
@@ -30,6 +32,8 @@ const generalPrompts: IPrompt[] = [
   QuestionWithContextPrompt,
   ZeroShotCoTPrompt,
   ZeroShotCoTAPEPrompt,
+  BasicSummarizationPrompt,
+  SummarizationIntoListPrompt,
   GeneralKnowledgePrompt,
 ];
 

diff --git a/...llm-prompt-templates/src/core/question.ts → .../llm-prompt-templates/src/core/general.ts b/...llm-prompt-templates/src/core/question.ts → .../llm-prompt-templates/src/core/general.ts
@@ -53,6 +53,36 @@ export const ZeroShotCoTAPEPrompt: IPrompt = {
 Let's work this out in a step by step way to be sure we have the right answer.`,
 };
 
+export const BasicSummarizationPrompt: IPrompt = {
+  name: 'Basic Summarization',
+  description: `This prompt asks the LLM to summarize a given text.`,
+  tags: [Tag.Summarization],
+  category: PromptCategory.General,
+  content: `Write a concise summary of the following text delimited by triple backquotes.
+Please provide your output in a manner that a 5 year old would understand.
+
+\`\`\`
+{text}
+\`\`\`
+
+SUMMARY:`,
+};
+
+export const SummarizationIntoListPrompt: IPrompt = {
+  name: 'Summarization Into List',
+  description: `This prompt asks the LLM to summarize a given text into a list of bullet points.`,
+  tags: [Tag.Summarization],
+  category: PromptCategory.General,
+  content: `Write a concise summary of the following text delimited by triple backquotes.
+Return your response in bullet points which covers the key points of the text.
+
+\`\`\`
+{text}
+\`\`\`
+
+BULLET POINT SUMMARY:`,
+};
+
 // https://arxiv.org/abs/2110.08387
 export const GeneralKnowledgePrompt: IPrompt = {
   name: 'General Knowledge',

diff --git a/libs/llm-prompt-templates/src/core/index.ts b/libs/llm-prompt-templates/src/core/index.ts
@@ -1,6 +1,6 @@
 export * from './categories.enum';
 export * from './code';
+export * from './general';
 export * from './interfaces';
 export * from './pal';
-export * from './question';
 export * from './tags.enum';
diff --git a/libs/llm-prompt-templates/src/core/tags.enum.ts b/libs/llm-prompt-templates/src/core/tags.enum.ts
@@ -11,4 +11,5 @@ export enum Tag {
   Error = 'error',
   Debugging = 'debugging',
   SQL = 'sql',
+  Summarization = 'summarization',
 }