pvs_agent/src/utils/ai-service.js

import OpenAI from 'openai'
import hljs from 'highlight.js'
import 'highlight.js/styles/atom-one-dark.css' // 导入一个暗色主题样式

export const openai = new OpenAI({
  // 若没有配置环境变量，请用百炼API Key将下行替换为：apiKey: "sk-xxx",
  apiKey: 'sk-e9855234f47346049809ce23ed3ebe3f',
  baseURL: 'https://dashscope.aliyuncs.com/compatible-mode/v1',
  dangerouslyAllowBrowser: true
})

export async function getFileContent(data) {
  try {
    const res = await fetch('http://180.76.147.97:18899/upload_file', {
      method: 'post',
      body: data
    }).then((res) => res.json())
    return res
  } catch (error) {}
}

export async function getFormKey(data) {
  try {
    const res = await fetch('http://180.76.147.97:18899/uie', {
      method: 'post',
      body: JSON.stringify(data)
    }).then((res) => res.json())
    return res
  } catch (error) {}
}

export async function hepl(data) {
  try {
    const res = await fetch(' http://180.76.147.97:18899/test', {
      method: 'post',
      body: JSON.stringify(data)
    }).then((res) => res.json())
    return res
  } catch (error) {}
}

export async function sendMessage(message) {
  try {
    // 创建新的 AbortController
    const controller = new AbortController()
    const a = +new Date()
    const response = await openai.chat.completions.create({
      model: 'qwen-plus', //模型列表：https://help.aliyun.com/zh/model-studio/getting-started/models
      messages: message,
      stream: true
    })
    return response
  } catch (error) {
    console.error('API call failed:', error)
    return '[FILE]'
  } finally {
  }
}

function formatMessages(currentMessage, Summary, html) {
  const messages = [
    {
      role: 'user',
      content: currentMessage
    }
  ]
  // const messages = this.context.filter(msg => msg.role === 'user').map((msg) => ({
  //     role: msg.role,
  //     content: msg.content,
  // }));
  // console.log(currentMessage, this.currentPageInfo, input);

  // // 如果存在页面信息，添加到当前消息的上下文
  // if (this.currentPageInfo && !input) {
  //     currentMessage = `基于之前总结的页面内容（标题：${JSON.stringify(this.currentPageInfo)}），${currentMessage}`;
  // }
  // messages.push()
  if (Summary) return messages
  if (html)
    messages.unshift({
      role: 'user',
      content: `页面主要内容${html}`
    })
  return messages
}

export function getSummaryPrompt(pageInfo) {
  return `请帮我总结以下网页内容的要点：

页面标题：${pageInfo.title}
网站：${pageInfo.siteName}
URL：${pageInfo.url}

主要内容：
${pageInfo.mainContent}

要求：
1. 用简洁清晰的语言提取3-5个核心要点
2. 保持客观中立的语气
3. 按重要性排序
4. 如果内容是新闻，需要包含时间、地点、人物等关键信息
5. 如果内容是教程，需要突出操作步骤和关键提示
6. 如果内容是产品介绍，需要包含主要特点和优势
7. 返回内容做好换行，以及展示样式

请以"以下是对该页面内容的总结："开头，然后用要点的形式列出主要内容。`
}

export function getFileSummaryPrompt(file, name) {
  return `请帮我总结以下文件内容的要点：

文件名称：${name}

主要内容：
${file}

要求：
1. 用简洁清晰的语言提取3-5个核心要点
2. 保持客观中立的语气
3. 按重要性排序
4. 返回内容做好换行，以及展示样式

请以"以下是对该文件内容的总结："开头，然后用要点的形式列出主要内容。`
}

export function formatMessage(text) {
  if (!text) return ''

  // 用于转义代码块中的 HTML 字符
  const escapeHtml = (str) => {
    return str
      .replace(/&/g, '&')
      .replace(/</g, '&lt;')
      .replace(/>/g, '&gt;')
      .replace(/"/g, '&quot;')
      .replace(/'/g, '&#039;')
  }

  // 处理代码高亮
  const highlightCode = (code, language) => {
    if (language && language.trim() !== '') {
      try {
        return hljs.highlight(code, { language: language.trim() }).value
      } catch (e) {
        // 如果指定的语言不支持，回退到自动检测
        return hljs.highlightAuto(code).value
      }
    }
    return hljs.highlightAuto(code).value
  }

  return (
    text
      // 处理带有语言标识的代码块
      .replace(/```(\w+)?\n([\s\S]*?)```/g, (match, lang, code) => {
        const highlightedCode = highlightCode(code, lang || '')
        return `<pre style="background-color: #282c34; padding: 10px; border-radius: 5px; overflow-x: auto;"><code class="hljs ${lang || ''}">${highlightedCode}</code></pre>`
      })
      // 处理不带语言标识的代码块
      .replace(/```([\s\S]*?)```/g, (match, code) => {
        const highlightedCode = highlightCode(code, '')
        return `<pre style="background-color: #282c34; padding: 10px; border-radius: 5px; overflow-x: auto;"><code class="hljs">${highlightedCode}</code></pre>`
      })
      // 处理行内代码
      .replace(
        /`([^`]+)`/g,
        (match, code) =>
          `<code style="background-color: #f0f0f0; padding: 2px 4px; border-radius: 3px;">${escapeHtml(code)}</code>`
      )
      // 处理标题 (h1 ~ h6)
      .replace(/^#{1,6}\s+(.+)$/gm, (match, content) => {
        const level = match.trim().split('#').length - 1
        return `<h${level}>${content.trim()}</h${level}>`
      })
      // 处理换行
      .replace(/\n/g, '<br>')
      // 处理连续空格
      .replace(/ {2,}/g, (match) => '&nbsp;'.repeat(match.length))
      // 处理粗体
      .replace(/\*\*(.*?)\*\*/g, '<strong>$1</strong>')
      // 处理斜体
      .replace(/\*(.*?)\*/g, '<em>$1</em>')
      // 处理链接
      .replace(
        /\[([^\]]+)\]\(([^)]+)\)/g,
        '<a href="$2" target="_blank">$1</a>'
      )
      // 处理无序列表：先将每行列表项转换为 <li> 标签
      .replace(/^[*-]\s+(.+)$/gm, '<li>$1</li>')
      // 然后将连续的 <li> 包裹在 <ul> 中
      .replace(/(<li>.*<\/li>)/gs, '<ul>$1</ul>')
      // 处理有序列表：先将每行列表项转换为 <li> 标签
      .replace(/^\d+\.\s+(.+)$/gm, '<li>$1</li>')
      // 然后将连续的 <li> 包裹在 <ol> 中
      .replace(/(<li>.*<\/li>)/gs, '<ol>$1</ol>')
      // 处理分隔线
      .replace(/^---+$/gm, '<hr>')
      // 处理引用
      .replace(/^>\s+(.+)$/gm, '<blockquote>$1</blockquote>')
  )
}

export function buildExcelUnderstandingPrompt(data, fileName, pageInfo) {
  // if (!data || data.length < 2) {
  //     return "这是一个空的Excel文件，请检查文件内容。";
  // }

  // const headers = data[0];
  // const rows = data.slice(1);
  // data[0].forEach((header, i) => {
  //     if (!this.excelData[header]) this.excelData[header] = []
  //     this.excelData[header].push(data[1][i])
  // })

  return `我将向你展示一个通过SheetJS库读取的Excel文件内容和一个form表单。请帮我理解这些数据：

文件名：${fileName}
列标题：${data.join(', ')}

表单内容：
${pageInfo}
             
要求：
1. 请根据表单中的表单项和列标题进行匹配，一定以表单为准,列标题只能匹配一次，没有和列标题匹配到的表单项不返回！
2. 生成表单项与列标题对应的数组，并使用findBy告诉我通过表单项的什么字段信息匹配到的，使用findByValue告诉我匹配到的表单项字段值，使用excelColumn字段告诉我excel文件中列标题的值。
3. 表单项有id根据id匹配，findBy是id，并通过findByValue告诉我id的值，没有id根据label匹配，findBy是label，并通过findByValue给我label元素的值，没有label根据placeholder匹配，findBy是placeholder，并通过findByValue告诉我placeholder的值，没有placeholder，再根据其他内容匹配
3. 并去除没有匹配到的表单项和excel文件中没有匹配到的列，
4. 通过type字段告诉我输入项的类型
5. 仅返回数组，不要返回任何其他内容。`
}

// 5. 如果表单项有label标签，同时返回label, 通过label字段告诉我label元素的文本
export function buildObjPrompt(obj, pageInfo) {
  return `我将向你展示一个对象和一个form表单。请帮我理解这些数据：

对象：${JSON.stringify(obj)}
表单内容：
${pageInfo}
             
要求：
1. 请根据表单中的表单项和对象进行匹配，一定以表单为准，key同时在对象和表单中出现，才能返回！
2. 表单项有label，根据label匹配,没有label根据placeholder匹配，没有placeholder，根据id匹配，再根据其他内容匹配
3. 并根据实际可操作的表单项的所有信息与上传的对象的key进行匹配，生成表单项与key的数组，并使用findBy告诉我通过表单项的什么字段信息匹配到的，使用findByValue告诉我匹配到的表单项字段值，使用excelColumn字段告诉我对应的key值。在一个字段内返回
4. 并去除没有匹配到的表单项和对象中没有匹配到的key，
5. 通过type字段告诉我输入项的类型，如果对象中key对应的是日期，type统一返回date
6. 仅返回数组，不要返回任何其他内容。`
}

function escapeHtml(html) {
  const div = document.createElement('div')
  div.textContent = html
  return div.innerHTML
}