AI

import { AIChatPlugin, AIPlugin } from '@udecode/plate-ai/react';
import {
  BaseBoldPlugin,
  BaseCodePlugin,
  BaseItalicPlugin,
  BaseStrikethroughPlugin,
  BaseUnderlinePlugin,
} from '@udecode/plate-basic-marks';
import { BaseBlockquotePlugin } from '@udecode/plate-block-quote';
import {
  BaseCodeBlockPlugin,
  BaseCodeLinePlugin,
  BaseCodeSyntaxPlugin,
} from '@udecode/plate-code-block';
import { BaseParagraphPlugin, createSlateEditor } from '@udecode/plate';
import { BaseHeadingPlugin, HEADING_LEVELS } from '@udecode/plate-heading';
import { BaseHorizontalRulePlugin } from '@udecode/plate-horizontal-rule';
import { BaseIndentListPlugin } from '@udecode/plate-indent-list';
import { BaseLinkPlugin } from '@udecode/plate-link';
import { MarkdownPlugin } from '@udecode/plate-markdown';Copy

export const createAIEditor = () => {
  const editor = createSlateEditor({
    id: 'ai',
    plugins: [
      BaseBlockquotePlugin,
      BaseBoldPlugin,
      BaseCodeBlockPlugin,
      BaseCodeLinePlugin,
      BaseCodePlugin,
      BaseCodeSyntaxPlugin,
      BaseItalicPlugin,
      BaseStrikethroughPlugin,
      BaseUnderlinePlugin,
      BaseHeadingPlugin,
      BaseHorizontalRulePlugin,
      BaseLinkPlugin,
      BaseParagraphPlugin,
      BaseIndentListPlugin.extend({
        inject: {
          targetPlugins: [
            BaseParagraphPlugin.key,
            ...HEADING_LEVELS,
            BaseBlockquotePlugin.key,
            BaseCodeBlockPlugin.key,
          ],
        },
        options: {
          listStyleTypes: {
            todo: {
              liComponent: TodoLiStatic,
              markerComponent: TodoMarkerStatic,
              type: 'todo',
            },
          },
        },
      }),
      MarkdownPlugin.configure({
        options: {
          remarkPlugins: [remarkMath, remarkGfm, remarkMdx],
        },
      }),
    ],
  });
 
 
  return editor;
};
 
const systemCommon = `\
你是一个先进的AI智能笔记助手，旨在提升笔记管理的生产力和创造力。
直接响应用户提示，提供清晰、简洁且相关的内容。保持中立、有帮助的语气。
 
规则：
- <Document>代表用户正在处理的整个笔记文档
- <Reminder>是关于如何响应INSTRUCTIONS的提示，不适用于问题回答
- 其他内容均为用户提示
- 你的响应应针对用户提示，提供精确的笔记管理协助
- 对于INSTRUCTIONS：严格遵循<Reminder>。仅提供需要插入或替换的内容，不要包含解释或评论
- 对于QUESTIONS：提供有帮助且简洁的回答，必要时可包含简短说明
- 关键：区分INSTRUCTIONS和QUESTIONS。指令通常要求修改或添加内容，问题则询问信息或澄清
`;
 
const systemDefault = `\
${systemCommon}
- <Block>代表用户当前处理的文本块
- 确保输出能无缝融入现有<Block>结构
- 关键：仅提供单个文本块，不要创建多个段落或独立块
<Block>
{block}
</Block>
`;
 
const systemSelecting = `\
${systemCommon}
- <Block>包含用户选区的文本块，提供上下文
- 确保输出能无缝融入现有<Block>结构
- <Selection>是用户在块中选定的特定文本，需要修改或询问
- 考虑<Block>提供的上下文，但仅修改<Selection>。响应应直接替换<Selection>
<Block>
{block}
</Block>
<Selection>
{selection}
</Selection>
`;
 
const systemBlockSelecting = `\
${systemCommon}
- <Selection>代表用户选中的完整文本块，需要修改或询问
- 响应应直接替换整个<Selection>
- 除非明确指示，否则保持选中块的整体结构和格式
- 关键：仅提供替换<Selection>的内容。除非特别要求，不要添加额外块或改变块结构
<Selection>
{block}
</Selection>
`;
 
const userDefault = `<Reminder>
关键：不要使用块级格式，仅允许行内格式
关键：不要换行或新建段落
永远不要输出<Block>
</Reminder>
{prompt}`;
 
const userSelecting = `<Reminder>
如果是问题，提供关于<Selection>的有帮助且简洁的回答
如果是指令，仅提供替换<Selection>的文本，不要解释
确保内容能无缝融入<Block>。如果<Block>为空，写一个随机句子
永远不要输出<Block>或<Selection>
</Reminder>
{prompt} 关于 <Selection>`;
 
const userBlockSelecting = `<Reminder>
如果是问题，提供关于<Selection>的有帮助且简洁的回答
如果是指令，仅提供替换整个<Selection>的内容，不要解释
除非另有指示，否则保持整体结构
永远不要输出<Block>或<Selection>
</Reminder>
{prompt} 关于 <Selection>`;
 
export const PROMPT_TEMPLATES = {
  systemBlockSelecting,
  systemDefault,
  systemSelecting,
  userBlockSelecting,
  userDefault,
  userSelecting,
};
 
const plugins = [
  // ...其他插件
  MarkdownPlugin.configure({
    options: {
      remarkPlugins: [remarkMath, remarkGfm, remarkMdx],
    },
  }),
  AIPlugin,
  AIChatPlugin.configure({
    options: {
      createAIEditor,
      promptTemplate: ({ isBlockSelecting, isSelecting }) => {
        return isBlockSelecting
          ? PROMPT_TEMPLATES.userBlockSelecting
          : isSelecting
            ? PROMPT_TEMPLATES.userSelecting
            : PROMPT_TEMPLATES.userDefault;
      },
      systemTemplate: ({ isBlockSelecting, isSelecting }) => {
        return isBlockSelecting
          ? PROMPT_TEMPLATES.systemBlockSelecting
          : isSelecting
            ? PROMPT_TEMPLATES.systemSelecting
            : PROMPT_TEMPLATES.systemDefault;
      },
    },
    render: { afterEditable: () => <AIMenu /> },
  }),
];Copy

    const result = streamText({
      experimental_transform: smoothStream({
        chunking: (buffer) => {
          // 检测代码块标记
          if (/```[^\s]+/.test(buffer)) {
            isInCodeBlock = true
          }else if(isInCodeBlock && buffer.includes('```') ) {
            isInCodeBlock = false
          }
          // 测试用例：不应反序列化带有markdown语法的链接
          if (buffer.includes('http')) {
            isInLink = true;
          } else if (buffer.includes('https')) {
            isInLink = true;
          } else if (buffer.includes('\n') && isInLink) {
            isInLink = false;
          }
          if (buffer.includes('*') || buffer.includes('-')) {
            isInList = true;
          } else if (buffer.includes('\n') && isInList) {
            isInList = false;
          }
          // 简单表格检测：遇到|进入表格模式，遇到双换行退出
          if (!isInTable && buffer.includes('|')) {
            isInTable = true;
          } else if (isInTable && buffer.includes('\n\n')) {
            isInTable = false;
          }
 
          // 根据内容类型选择分块策略
          let match;
 
          if (isInCodeBlock || isInTable || isInLink) {
            // 对代码块和表格使用行分块
            match = CHUNKING_REGEXPS.line.exec(buffer);
          } else if (isInList) {
            // 对列表使用列表分块
            match = CHUNKING_REGEXPS.list.exec(buffer);
          } else {
            // 对常规文本使用词分块
            match = CHUNKING_REGEXPS.word.exec(buffer);
          }
          if (!match) {
            return null;
          }
 
          return buffer.slice(0, match.index) + match?.[0];
        },
      }),
      // 其他配置项
      
      // maxTokens: 2048,
      // messages: convertToCoreMessages(messages),
      // model: openai('gpt-4o'),
      // system: system,
    });
 Copy