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, ">") .replace(/"/g, """) .replace(/'/g, "'"); }; // 处理代码高亮 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 `
${highlightedCode}
`; }) // 处理不带语言标识的代码块 .replace(/```([\s\S]*?)```/g, (match, code) => { const highlightedCode = highlightCode(code, ''); return `
${highlightedCode}
`; }) // 处理行内代码 .replace(/`([^`]+)`/g, (match, code) => `${escapeHtml(code)}`) // 处理标题 (h1 ~ h6) .replace(/^#{1,6}\s+(.+)$/gm, (match, content) => { const level = match.trim().split("#").length - 1; return `${content.trim()}`; }) // 处理换行 .replace(/\n/g, "
") // 处理连续空格 .replace(/ {2,}/g, (match) => " ".repeat(match.length)) // 处理粗体 .replace(/\*\*(.*?)\*\*/g, "$1") // 处理斜体 .replace(/\*(.*?)\*/g, "$1") // 处理链接 .replace(/\[([^\]]+)\]\(([^)]+)\)/g, '$1') // 处理无序列表：先将每行列表项转换为
  • 标签 .replace(/^[*-]\s+(.+)$/gm, "
  • $1
    • ") // 然后将连续的
  • 包裹在