feat: 🎸 add copyQA contextMenu (#51)

bingryan · May 19, 2023 · eb74ca3 · eb74ca3
1 parent 93d534c
commit eb74ca3
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 0 deletions.
diff --git a/src/locales/en-US.ts b/src/locales/en-US.ts
@@ -27,6 +27,7 @@ export default {
 		edit: 'Edit',
 		delete: 'Delete',
 		copy: 'Copy',
+		copyQA: 'CopyQA',
 		newChat: 'New Chat',
 		editChat: 'Edit Chat',
 		saveSuccess: 'save success',

diff --git a/src/locales/zh-CN.ts b/src/locales/zh-CN.ts
@@ -27,6 +27,7 @@ export default {
 		edit: '编辑',
 		delete: '删除',
 		copy: '复制',
+		copyQA: '复制QA',
 		newChat: '新建聊天',
 		editChat: '编辑聊天',
 		saveSuccess: '导出成功',

diff --git a/src/store/chat.ts b/src/store/chat.ts
@@ -141,6 +141,25 @@ export const useChatStore = defineStore(ID, {
 			this.chatCache[index].uptime = uptime;
 			this.reloadRoute(id);
 		},
+		/**
+		 * get user's chat context by id and index
+		 * @param id chat id
+		 * @param dataIndex data index
+		 */
+		selectContextQA(id: number, dataIndex: number) {
+			const cacheIndex = this.getCacheIndex(id);
+			const contextData = [];
+			if (cacheIndex !== -1) {
+				const data = this.chatCache[cacheIndex].data[dataIndex];
+				contextData.push(data);
+				if (data.role === 'user') {
+					contextData.push(this.chatCache[cacheIndex].data[dataIndex + 1]);
+				} else {
+					contextData.unshift(this.chatCache[cacheIndex].data[dataIndex - 1]);
+				}
+			}
+			return contextData;
+		},
 
 		/**
 		 * remove single record at cache by id

diff --git a/src/views/chat/components/message/Text.vue b/src/views/chat/components/message/Text.vue
@@ -81,6 +81,12 @@
 		chatStore.removeCache(+id, index);
 	}
 
+	function copyQA(index: number) {
+		const contextList = chatStore.selectContextQA(+id, index);
+		const copyContent = `### ${contextList[0].content} \n${contextList[1].content}`;
+		copy(copyContent);
+	}
+
 	const contextMenuTheme = computed(() => {
 		return appStore.theme === 'dark' ? 'mac dark' : 'mac';
 	});
@@ -103,6 +109,13 @@
 					}),
 					onClick: () => copy(props.content),
 				},
+				{
+					label: t('common.copyQA'),
+					icon: h('i', {
+						class: 'fa-solid fa-copy',
+					}),
+					onClick: () => copyQA(index),
+				},
 			],
 			zIndex: 60,
 			minWidth: 100,