feat: AI助手现在可以知道用户当前所在页面

Author: LeafYeeXYZ

src/components/AI.tsx

@@ -29,7 +29,10 @@ const GREETTING =
 	'你好, 我是 PsychPen 的 AI 助手, 可以帮你讲解 PsychPen 的使用方法、探索你的数据集、导出数据、跳转页面、变量标准化/中心化/离散化、生成新变量、筛选数据等. 请问有什么可以帮你的?'
 const INSTRUCTION =
 	'你是在线统计分析和数据可视化软件"PsychPen"中的AI助手. \n\n你将收到用户的提问、当前用户导入到软件中的数据集中的变量的信息、PsychPen的使用和开发文档、可以供你调用的工具 (函数) 信息. \n\n你的任务是按照用户的要求, 对用户进行回复, 或调用工具 (函数). 在调用工具 (函数) 前, 请确保你已经明确知晓了用户的意图, 否则请通过进一步和用户对话来确认细节. \n\n你的回复中如果包含数学公式和符号, 请使用 TeX 语法, 并将行内公式用 `$` 包裹 (类似于 Markdown 的行内代码), 将块级公式用 `$$` 包裹 (类似于 Markdown 的代码块).'
-function GET_PROMPT(vars: Variable[]): string {
+function GET_PROMPT(
+	vars: Variable[],
+	page: string,
+): string {
 	const varsInfo = vars.map((col) => {
 		if (col.type === '称名或等级数据') {
 			return `| ${col.name} | ${col.type} | ${col.valid} | ${col.missing} | ${col.unique} | - | - | - | - | - | - |`
@@ -47,9 +50,10 @@ function GET_PROMPT(vars: Variable[]): string {
 			: '未定义任何子变量'
 		return `| ${col.name} | ${col.type} | ${col.valid} | ${col.missing} | ${col.unique} | ${col.mean?.toFixed(4) || '-'} | ${col.std?.toFixed(4) || '-'} | ${col.q2?.toFixed(4) || '-'} | ${col.min?.toFixed(4) || '-'} | ${col.max?.toFixed(4) || '-'} | ${subVarInfo} |`
 	})
+	const userText = `\n\n# 用户信息\n\n用户当前所处的页面为: ${page}`
 	const varsText = `\n\n# 变量信息\n\n| 变量名 | 变量类型 | 有效值数量 | 缺失值数量 | 唯一值数量 | 均值 | 标准差 | 中位数 | 最小值 | 最大值 | 子变量信息 |\n| --- | --- | --- | --- | --- | --- | --- | --- | --- | --- | --- |\n${varsInfo.join('\n')}`
 	const docsText = `\n\n# 使用文档\n\n\`\`\`markdown\n${readme}\n\`\`\``
-	return INSTRUCTION + varsText + docsText
+	return INSTRUCTION + userText + varsText + docsText
 }
 
 export function AI() {
@@ -96,7 +100,7 @@ export function AI() {
 				setMessages([...old, user])
 				setInput('')
 			})
-			const system = GET_PROMPT(dataCols)
+			const system = GET_PROMPT(dataCols, nav.currentPageInfo())
 
 			const stream = await ai.chat.completions.create({
 				model: model,

src/lib/hooks/useNav.tsx

@@ -231,6 +231,8 @@ export const TOOLS_VIEW_SUB_PAGES_ELEMENTS: Record<
 }
 
 type NavState = {
+  currentPageInfo: () => string
+
 	mainPage: React.ReactElement
 	activeMainPage: MAIN_PAGES_LABELS
 	setMainPage: (page: MAIN_PAGES_LABELS) => void
@@ -258,7 +260,24 @@ const DEFAULT_STATISTICS_SUB_PAGE =
 	STATISTICS_SUB_PAGES_LABELS.ONE_SAMPLE_T_TEST
 const DEFAULT_TOOLS_SUB_PAGE = TOOLS_VIEW_SUB_PAGES_LABELS.STATISTIC_TO_PVALUE
 
-export const useNav = create<NavState>()((setState) => ({
+export const useNav = create<NavState>()((setState, getState) => ({
+	currentPageInfo: () => {
+		const state = getState()
+		switch (state.activeMainPage) {
+			case MAIN_PAGES_LABELS.DATA:
+				return '数据视图'
+			case MAIN_PAGES_LABELS.VARIABLE:
+				return `变量视图的"${state.activeVariableViewSubPage}"页面`
+			case MAIN_PAGES_LABELS.PLOTS:
+				return `绘图视图的"${state.activePlotsViewSubPage}"页面`
+			case MAIN_PAGES_LABELS.STATISTICS:
+				return `统计视图的"${state.activeStatisticsViewSubPage}"页面`
+			case MAIN_PAGES_LABELS.TOOLS:
+				return `工具视图的"${state.activeToolsViewSubPage}"页面`
+			default:
+				throw new Error('发现未知页面')
+		}
+	},
 	mainPage: MAIN_PAGES_ELEMENTS[MAIN_PAGES_LABELS.DATA],
 	activeMainPage: MAIN_PAGES_LABELS.DATA,
 	setMainPage: (page) =>