Skip to content

C
center-book
提交 15a93038 authored 作者: 王鹏飞's avatar 王鹏飞
浏览文件
操作

refactor: 重构useWenku.js

上级 4776215b
显示空白字符变更
内嵌 并排
正在显示 包含 120 行增加 和 82 行删除
src/common/wangeditor-customer/menu/common/AIWriteModal.jsx
浏览文件 @ 15a93038
......@@ -11,7 +11,7 @@ import {
import { ConfigProvider, Modal, Input, Button } from 'antd'
const { TextArea } = Input
import './AISearchModal.less'
import { usePaperOutline, useOutlineToPaper } from '@/hooks/useWenku'
import { usePaper } from '@/hooks/useWenku'
import { useCopyToClipboard } from 'react-use'
export default function AIModal() {
......@@ -20,7 +20,7 @@ export default function AIModal() {
const [textIndent, setTextIndent] = useState(0)
const prePromptRef = useRef(null)
const messageScrollRef = useRef(null)
const { messages, setMessages, isLoading, query } = usePaperOutline()
const { messages, setMessages, isLoading, generateOutline, generatePaper } = usePaper()
useEffect(() => {
if (prePromptRef.current) {
const width = prePromptRef.current.offsetWidth + 10
......@@ -43,7 +43,7 @@ export default function AIModal() {
}
}
const handleSearch = () => {
query(prePrompt + content)
generateOutline(prePrompt + content)
setContent('')
}
......@@ -55,13 +55,11 @@ export default function AIModal() {
}
// 生成论文
const { query: outlineToPaper } = useOutlineToPaper()
const handleGeneratePaper = async (msg) => {
setMessages((prevMessages) => {
return [...prevMessages, { content: '正在生成长文...', role: 'ai', tips: '预计10分钟', queryID: msg.queryID }]
})
const paper = await outlineToPaper({ userQuery: msg.userQuery, queryID: msg.queryID, outline: msg.content })
const paper = await generatePaper({ userQuery: msg.userQuery, queryID: msg.queryID, outline: msg.content })
setMessages((prevMessages) => {
prevMessages.pop()
return [...prevMessages, { content: '已为您生成初稿，请点击查看', role: 'ai', queryID: msg.queryID, paper }]
......@@ -101,7 +99,7 @@ export default function AIModal() {
size="small"
icon={<UndoOutlined />}
disabled={isLoading}
onClick={() => query(msg.userQuery)}>
onClick={() => generateOutline(msg.userQuery)}>
换个大纲
</Button>
<Button
......
src/hooks/useAsyncTask.js 0 → 100644
浏览文件 @ 15a93038
import { useState, useCallback } from 'react'
/**
* 通用异步任务处理hook
* @param {Function} taskFn 异步任务函数
* @returns {Object} { isLoading, error, execute }
*/
export function useAsyncTask(taskFn) {
const [isLoading, setIsLoading] = useState(false)
const [error, setError] = useState(null)
const execute = useCallback(
async (...args) => {
try {
setIsLoading(true)
setError(null)
return await taskFn(...args)
} catch (err) {
setError(err)
throw err
} finally {
setIsLoading(false)
}
},
[taskFn]
)
return { isLoading, error, execute }
}
src/hooks/useWenku.js
浏览文件 @ 15a93038
import { useState, useEffect, useRef } from 'react'
import { aiSearch, paperOutline, outlineToPaper, download } from '@/api/wenku'
import { useAsyncTask } from './useAsyncTask'
/**
* @typedef {Object} Message
* @property {string} content
* @property {'user' | 'ai'} role
* @property {Array<{title: string, url: string}>} [searchReferList]
* @property {string} [userQuery]
* @property {Chapter[]} [chapters]
* @property {string} [logID]
* @property {string} [queryID]
*/
/**
* @typedef {Object} Chapter
* @property {string} title
* @property {number} level
* @property {string} desc
* @property {string} chapter
*/
/**
* @typedef {Object} Paper
* @property {string} content
* @property {string} [downloadLink]
* @property {string} [docID]
*/
/**
* @typedef {Object} DownloadResult
* @property {string} download_link
*/
// AI搜索
export function useSearch() {
const [messages, setMessages] = useState([])
const [isLoading, setIsLoading] = useState(false)
const query = async (userQuery) => {
try {
setIsLoading(true)
const { isLoading, execute } = useAsyncTask(async (userQuery) => {
setMessages((prevMessages) => [...prevMessages, { content: userQuery, role: 'user' }])
const currentMessage = { content: '', searchReferList: [], role: 'ai', userQuery }
await aiSearch({
body: JSON.stringify({ userQuery }),
onmessage(message) {
......@@ -29,17 +60,13 @@ export function useSearch() {
})
}
} catch (error) {
console.log(error)
console.error('Error parsing message:', error)
}
},
})
} catch (error) {
console.log(error)
} finally {
setIsLoading(false)
}
}
return { messages, setMessages, isLoading, query }
})
return { messages, setMessages, isLoading, query: execute }
}
function parseInput(inputStr) {
......@@ -95,15 +122,34 @@ function parseInput(inputStr) {
return result
}
// 生成论文大纲
export function usePaperOutline() {
/**
* 论文生成全流程管理
* @returns {{
* messages: Message[],
* paper: Paper,
* isLoading: boolean,
* generateOutline: (userQuery: string) => Promise<void>,
* generatePaper: (data: any) => Promise<Paper>,
* downloadPaper: (docID: string) => Promise<string>
* }}
*/
export function usePaper() {
const [messages, setMessages] = useState([])
const [isLoading, setIsLoading] = useState(false)
const query = async (userQuery) => {
try {
setIsLoading(true)
const [paper, setPaper] = useState(null)
const [downloadLink, setDownloadLink] = useState('')
const timerRef = useRef(null)
const clearTimer = () => {
if (timerRef.current) {
clearTimeout(timerRef.current)
timerRef.current = null
}
}
const { isLoading: isOutlineLoading, execute: generateOutline } = useAsyncTask(async (userQuery) => {
setMessages((prevMessages) => [...prevMessages, { content: userQuery, role: 'user' }])
const currentMessage = { content: '', role: 'ai', userQuery, chapters: [], logID: '', queryID: '' }
await paperOutline({
body: JSON.stringify({ userQuery }),
onmessage(message) {
......@@ -124,75 +170,34 @@ export function usePaperOutline() {
})
}
} catch (error) {
console.log(error)
console.error('Error parsing message:', error)
}
},
})
} catch (error) {
console.log(error)
} finally {
setIsLoading(false)
}
}
return { messages, setMessages, isLoading, query }
}
})
// 大纲生成论文
export function useOutlineToPaper() {
const { queryWithPolling } = useDownload()
const [isLoading, setIsLoading] = useState(false)
const [paper, setPaper] = useState('')
const query = async (data) => {
try {
setIsLoading(true)
const { isLoading: isPaperLoading, execute: generatePaper } = useAsyncTask(async (data) => {
const { raw } = await outlineToPaper(data)
setPaper(raw)
const downloadLink = await queryWithPolling({
docID: '280fe43ee63a580216fc700abb68a98271feac83' || raw.docID,
const downloadLink = await downloadWithPolling({
docID: raw.docID,
})
setPaper({ ...raw, downloadLink })
return { ...raw, downloadLink }
} catch (error) {
console.log(error)
} finally {
setIsLoading(false)
}
}
return { paper, setPaper, isLoading, query }
}
// 文档下载
export function useDownload() {
const [downloadLink, setDownloadLink] = useState('')
const [isLoading, setIsLoading] = useState(false)
const timerRef = useRef(null)
const clearTimer = () => {
if (timerRef.current) {
clearTimeout(timerRef.current)
timerRef.current = null
}
}
})
const query = async (data) => {
try {
setIsLoading(true)
const { raw } = await download(data)
const { isLoading: isDownloadLoading, execute: downloadPaper } = useAsyncTask(async (docID) => {
const { raw } = await download({ docID })
if (raw) {
setDownloadLink(raw.download_link)
return raw.download_link
}
} catch (error) {
console.error(error)
} finally {
setIsLoading(false)
}
}
})
const queryWithPolling = async (data, interval = 3000) => {
const downloadWithPolling = async (data, interval = 3000) => {
const poll = async (resolve, reject) => {
try {
const link = await query(data)
const link = await downloadPaper(data.docID)
if (link) {
resolve(link)
return
......@@ -209,11 +214,17 @@ export function useDownload() {
}
useEffect(() => {
// 清除定时器的逻辑
return () => {
clearTimer()
}
return () => clearTimer()
}, [])
return { downloadLink, isLoading, query, queryWithPolling, clearTimer }
return {
messages,
setMessages,
paper,
downloadLink,
isLoading: isOutlineLoading || isPaperLoading || isDownloadLoading,
generateOutline,
generatePaper,
downloadPaper,
}
}
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论