refactor: extract for actions type and map

unit-mesh · Sep 29, 2024 · 2c25ecc · 2c25ecc
1 parent 83d09c5
commit 2c25ecc
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 43 deletions.
diff --git a/web/core/lib/editor/menu/menu-bubble.tsx b/web/core/lib/editor/menu/menu-bubble.tsx
@@ -36,7 +36,7 @@ export const MenuBubble = ({ editor }: {
 
     innerSmartMenus.push({
       name: '润色',
-      template: `优化表达：###{{${DefinedVariable.SELECTION}}}###`,
+      template: `请润色并重写如下的内容：###{{${DefinedVariable.SELECTION}}}###`,
       facetType: FacetType.BUBBLE_MENU,
       changeForm: ChangeForm.DIFF,
       outputForm: OutputForm.TEXT

diff --git a/web/core/lib/editor/prompts/article-prompts.ts b/web/core/lib/editor/prompts/article-prompts.ts
@@ -1,6 +1,4 @@
 import {
-	BuiltInFunc,
-	ChangeForm,
 	DefinedVariable,
 	FacetType,
 	OutputForm,
@@ -39,29 +37,6 @@ export const ToolbarMenuPrompts: PromptAction[] = [
 ];
 
 export const BubbleMenuPrompts: PromptAction[] = [
-	{
-		name: 'Polish',
-		i18Name: true,
-		template: `You are an assistant helping to polish sentence. Output in markdown format. \n ###{{${DefinedVariable.SELECTION}}}###`,
-		facetType: FacetType.BUBBLE_MENU,
-		outputForm: OutputForm.STREAMING,
-	},
-	{
-		name: 'Simplify Content',
-		i18Name: true,
-		template: `You are an assistant helping to simplify content. Output in markdown format. \n ###{{${DefinedVariable.SELECTION}}}###`,
-		facetType: FacetType.BUBBLE_MENU,
-		outputForm: OutputForm.STREAMING,
-		changeForm: ChangeForm.DIFF,
-	},
-	{
-		name: 'Similar Chunk',
-		i18Name: true,
-		template: `{{${DefinedVariable.SELECTION}}}`,
-		builtinFunction: BuiltInFunc.SIMILAR_CHUNKS,
-		facetType: FacetType.BUBBLE_MENU,
-		outputForm: OutputForm.STREAMING,
-	},
 	{
 		name: 'Translate',
 		i18Name: true,

diff --git a/web/core/lib/editor/prompts/prompts-manager.ts b/web/core/lib/editor/prompts/prompts-manager.ts
@@ -19,15 +19,16 @@ export class PromptsManager {
 		return PromptsManager.instance;
 	}
 
+	actionsMap = {
+		"article": ArticlePrompts,
+		"requirements": RequirementsPrompts
+	}
+
 	getActions(type: FacetType, articleType: TypeOptions): PromptAction[] {
 		let typedPrompts: PromptAction[] = []
 
-		switch (articleType?.value) {
-			case "requirements":
-				typedPrompts = RequirementsPrompts;
-				break;
-			default:
-				typedPrompts = ArticlePrompts;
+		if (articleType?.value) {
+			typedPrompts = this.actionsMap[articleType.value]
 		}
 
 		const actions = typedPrompts.filter(prompt => prompt.facetType === type);
@@ -40,14 +41,13 @@ export class PromptsManager {
 		})
 	}
 
-	save(prompt: PromptAction[]) {
-		// todo: implement this
-	}
-
 	variableList(): string[] {
 		return Object.values(DefinedVariable);
 	}
 
+	updateActionsMap(articleType: string, prompts: PromptAction[]) {
+		this.actionsMap[articleType] = prompts
+	}
 
 	compile(string: string, data: object) {
 		const template = Handlebars.compile(string)

diff --git a/web/core/lib/editor/prompts/requirements-prompts.ts b/web/core/lib/editor/prompts/requirements-prompts.ts
@@ -18,13 +18,6 @@ const BubbleMenuPrompts: PromptAction[] = [
 		facetType: FacetType.BUBBLE_MENU,
 		outputForm: OutputForm.STREAMING,
 		changeForm: ChangeForm.DIFF,
-	},
-	{
-		name: 'Polish',
-		i18Name: true,
-		template: `You are an assistant helping to polish sentence. Output in markdown format. \n ###{{${DefinedVariable.SELECTION}}}###`,
-		facetType: FacetType.BUBBLE_MENU,
-		outputForm: OutputForm.STREAMING,
 	}
 ];