|
|
|
@ -24,9 +24,12 @@ import { |
|
|
|
deleteSuperAgentSession, |
|
|
|
respondSuperAgentCheckpointStream, |
|
|
|
sendSuperAgentChat, |
|
|
|
uploadSuperAgentFile, |
|
|
|
} from "../actions/api"; |
|
|
|
import { createResourceFileFromUpload } from "../actions/resourceLibrary"; |
|
|
|
import { |
|
|
|
completeAttachmentUpload, |
|
|
|
createAttachmentUploadIntent, |
|
|
|
listAttachments, |
|
|
|
} from "../actions/agentResources"; |
|
|
|
import useSuperAgentSidebar from "./useSuperAgentSidebar"; |
|
|
|
import { |
|
|
|
TASK_STAGE, |
|
|
|
@ -62,6 +65,11 @@ import { |
|
|
|
} from "../utils/workspace"; |
|
|
|
import { normalizeCheckpoint } from "../utils/checkpoint"; |
|
|
|
import { isWorkflowBodyText } from "../utils/tenderWorkflow"; |
|
|
|
import { |
|
|
|
calculateFileSha256, |
|
|
|
normalizeAttachmentUploadIntent, |
|
|
|
uploadFileToIntent, |
|
|
|
} from "../utils/attachmentUpload"; |
|
|
|
|
|
|
|
const getActionData = (res) => { |
|
|
|
const payload = res?.payload ?? res; |
|
|
|
@ -76,41 +84,19 @@ const ensureActionSuccess = (res, fallbackMessage) => { |
|
|
|
return getActionData(res); |
|
|
|
}; |
|
|
|
|
|
|
|
const toReadyUploadedFile = (attachment) => ({ |
|
|
|
id: String(attachment.id || attachment.attachment_id || attachment.attachmentId), |
|
|
|
attachmentId: String(attachment.id || attachment.attachment_id || attachment.attachmentId), |
|
|
|
name: attachment.file_name || attachment.filename || "未命名文件", |
|
|
|
status: attachment.status, |
|
|
|
}); |
|
|
|
|
|
|
|
const buildSessionParams = (sessionId) => ({ |
|
|
|
conversationId: sessionId, |
|
|
|
sessionId, |
|
|
|
":sessionId": sessionId, |
|
|
|
}); |
|
|
|
|
|
|
|
const hasFilePathPrefix = (filePath = "") => { |
|
|
|
return /^(https?:)?\/\//i.test(filePath) || /^(data|blob):/i.test(filePath); |
|
|
|
}; |
|
|
|
|
|
|
|
const buildQiniuFilePath = (filePath = "") => { |
|
|
|
const cleanFilePath = String(filePath || "").trim(); |
|
|
|
if (!cleanFilePath || hasFilePathPrefix(cleanFilePath)) return cleanFilePath; |
|
|
|
|
|
|
|
const qiniuDomain = window?.env?.FS_QINIU_DOMAIN || ""; |
|
|
|
const cleanDomain = String(qiniuDomain || "").replace(/\/+$/, ""); |
|
|
|
if (!cleanDomain) return cleanFilePath; |
|
|
|
|
|
|
|
return `${cleanDomain}/${cleanFilePath.replace(/^\/+/, "")}`; |
|
|
|
}; |
|
|
|
|
|
|
|
//[文件上下文] 上传文件只向智能体传递访问地址，不附加固定业务文案。
|
|
|
|
const buildUploadedFileReference = (fileList = []) => { |
|
|
|
return fileList |
|
|
|
.map((file) => { |
|
|
|
const fileName = file?.name || file?.originalFilename || "未命名文件"; |
|
|
|
const rawFilePath = file?.url || file?.path || file?.objectKey || file?.localPath || ""; |
|
|
|
const filePath = buildQiniuFilePath(rawFilePath); |
|
|
|
if (!filePath) return ""; |
|
|
|
return `（文件名：${fileName}，文件路径：${filePath}）`; |
|
|
|
}) |
|
|
|
.filter(Boolean) |
|
|
|
.join("\n"); |
|
|
|
}; |
|
|
|
|
|
|
|
const useSuperAgentChat = ({ |
|
|
|
currentUserId, |
|
|
|
openAuthModal, |
|
|
|
@ -156,6 +142,7 @@ const useSuperAgentChat = ({ |
|
|
|
const streamingMessageKeyRef = useRef(""); |
|
|
|
const chatRunPromiseRef = useRef(Promise.resolve()); |
|
|
|
const chatRunActiveRef = useRef(false); |
|
|
|
const uploadAbortControllerRef = useRef(null); |
|
|
|
const creatingSessionRef = useRef(false); |
|
|
|
const sidebarRefreshTimerRef = useRef(null); |
|
|
|
const artifactProgressSessionRef = useRef(""); |
|
|
|
@ -168,6 +155,33 @@ const useSuperAgentChat = ({ |
|
|
|
activeSessionIdRef.current = activeSessionId; |
|
|
|
}, [activeSessionId]); |
|
|
|
|
|
|
|
useEffect(() => { |
|
|
|
const conversationId = String(activeConversationId || ""); |
|
|
|
if (!conversationId) { |
|
|
|
setUploadedFileList([]); |
|
|
|
return undefined; |
|
|
|
} |
|
|
|
let active = true; |
|
|
|
listAttachments(conversationId) |
|
|
|
.then((page) => { |
|
|
|
if (!active) return; |
|
|
|
setUploadedFileList( |
|
|
|
page.items |
|
|
|
.filter((attachment) => attachment.status === "ready" && !attachment.message_id) |
|
|
|
.map(toReadyUploadedFile) |
|
|
|
); |
|
|
|
}) |
|
|
|
.catch((error) => { |
|
|
|
superAgentWarn("chat.upload", "attachment_list_failed", { |
|
|
|
conversationId, |
|
|
|
error: error?.message, |
|
|
|
}); |
|
|
|
}); |
|
|
|
return () => { |
|
|
|
active = false; |
|
|
|
}; |
|
|
|
}, [activeConversationId]); |
|
|
|
|
|
|
|
const activeTask = useMemo( |
|
|
|
() => taskList.find((task) => task.id === activeTaskId) || null, |
|
|
|
[activeTaskId, taskList] |
|
|
|
@ -357,6 +371,9 @@ const useSuperAgentChat = ({ |
|
|
|
if (chatAbortControllerRef.current) { |
|
|
|
chatAbortControllerRef.current.abort(); |
|
|
|
} |
|
|
|
if (uploadAbortControllerRef.current) { |
|
|
|
uploadAbortControllerRef.current.abort(); |
|
|
|
} |
|
|
|
if (sidebarRefreshTimerRef.current) { |
|
|
|
window.clearTimeout(sidebarRefreshTimerRef.current); |
|
|
|
} |
|
|
|
@ -398,6 +415,7 @@ const useSuperAgentChat = ({ |
|
|
|
folderId = null, |
|
|
|
title = "", |
|
|
|
shouldResetMessages = false, |
|
|
|
returnConversationId = false, |
|
|
|
} = {}) => { |
|
|
|
if (!currentUserId) { |
|
|
|
openAuthModal("login"); |
|
|
|
@ -469,7 +487,7 @@ const useSuperAgentChat = ({ |
|
|
|
} |
|
|
|
|
|
|
|
await loadSidebar({ silent: true }); |
|
|
|
return nextSessionId; |
|
|
|
return returnConversationId ? nextSessionId : nextThreadId; |
|
|
|
}; |
|
|
|
|
|
|
|
const getCurrentSessionType = () => { |
|
|
|
@ -1545,14 +1563,14 @@ const useSuperAgentChat = ({ |
|
|
|
|
|
|
|
//[执行上传文件的实际逻辑]
|
|
|
|
const performUpload = (file, fileName, currentSessionType) => { |
|
|
|
const currentSessionId = activeSessionId || activeTaskId; |
|
|
|
const shouldCreateSession = !currentSessionId; |
|
|
|
const currentConversationId = activeConversationId || activeTaskId; |
|
|
|
const shouldCreateSession = !currentConversationId; |
|
|
|
const shouldUseTaskSession = |
|
|
|
currentSessionType === "task" || (shouldCreateSession && isTaskRoute); |
|
|
|
const shouldCreateNewTask = shouldCreateSession && shouldUseTaskSession; |
|
|
|
superAgentLog("chat.upload", "perform.start", { |
|
|
|
fileName, |
|
|
|
currentSessionId, |
|
|
|
currentConversationId, |
|
|
|
currentSessionType, |
|
|
|
shouldCreateSession, |
|
|
|
shouldUseTaskSession, |
|
|
|
@ -1583,23 +1601,32 @@ const useSuperAgentChat = ({ |
|
|
|
status: "preparing", |
|
|
|
}); |
|
|
|
try { |
|
|
|
const sessionId = |
|
|
|
currentSessionId || |
|
|
|
const conversationId = |
|
|
|
currentConversationId || |
|
|
|
(await createAndActivateSession({ |
|
|
|
isTaskSession: shouldUseTaskSession, |
|
|
|
title: shouldUseTaskSession |
|
|
|
? fileName.replace(/\.[^.]+$/, "") || "AI 标书任务" |
|
|
|
: "", |
|
|
|
returnConversationId: true, |
|
|
|
})); |
|
|
|
superAgentLog("chat.upload", "session.ready", { |
|
|
|
sessionId, |
|
|
|
conversationId, |
|
|
|
fileName, |
|
|
|
shouldUseTaskSession, |
|
|
|
}); |
|
|
|
const uploadResult = await uploadSuperAgentFile({ |
|
|
|
sessionId, |
|
|
|
userId: currentUserId, |
|
|
|
file, |
|
|
|
const sha256 = await calculateFileSha256(file); |
|
|
|
const uploadIntent = normalizeAttachmentUploadIntent( |
|
|
|
await createAttachmentUploadIntent(conversationId, { |
|
|
|
file_name: fileName, |
|
|
|
mime_type: file.type || "application/octet-stream", |
|
|
|
file_size: Number(file.size || 0), |
|
|
|
...(sha256 ? { sha256 } : {}), |
|
|
|
}) |
|
|
|
); |
|
|
|
uploadAbortControllerRef.current = new AbortController(); |
|
|
|
await uploadFileToIntent(uploadIntent, file, { |
|
|
|
signal: uploadAbortControllerRef.current.signal, |
|
|
|
onProgress: (progress) => { |
|
|
|
setUploadProgress({ |
|
|
|
fileName, |
|
|
|
@ -1613,57 +1640,36 @@ const useSuperAgentChat = ({ |
|
|
|
percent: 100, |
|
|
|
status: "processing", |
|
|
|
})); |
|
|
|
const uploadData = uploadResult?.data || uploadResult || {}; |
|
|
|
try { |
|
|
|
await createResourceFileFromUpload(uploadData); |
|
|
|
} catch (resourceError) { |
|
|
|
superAgentWarn("chat.upload", "resource_library_save_failed", { |
|
|
|
fileName, |
|
|
|
error: resourceError?.message, |
|
|
|
}); |
|
|
|
message.warning("文件已上传，但保存到资源库失败"); |
|
|
|
const uploadData = await completeAttachmentUpload(uploadIntent.attachmentId); |
|
|
|
const attachmentPage = await listAttachments(conversationId); |
|
|
|
const readyAttachment = attachmentPage.items.find((attachment) => |
|
|
|
String(attachment.id || attachment.attachment_id || attachment.attachmentId) === uploadIntent.attachmentId |
|
|
|
) || uploadData?.data || uploadData || {}; |
|
|
|
if (readyAttachment.status !== "ready") { |
|
|
|
throw new Error("附件尚未准备完成，请稍后重试"); |
|
|
|
} |
|
|
|
const uploadedFileName = |
|
|
|
uploadData.filename || |
|
|
|
uploadData.original_filename || |
|
|
|
uploadData.originalFilename || |
|
|
|
readyAttachment.file_name || |
|
|
|
readyAttachment.filename || |
|
|
|
readyAttachment.original_filename || |
|
|
|
readyAttachment.originalFilename || |
|
|
|
fileName; |
|
|
|
const rawUploadedFilePath = |
|
|
|
uploadData.path || |
|
|
|
uploadData.url || |
|
|
|
uploadData.object_key || |
|
|
|
uploadData.objectKey || |
|
|
|
""; |
|
|
|
const uploadedFilePath = buildQiniuFilePath(rawUploadedFilePath); |
|
|
|
const uploadedLocalPath = uploadData.local_path || uploadData.localPath || ""; |
|
|
|
const uploadedFileStorage = uploadData.storage || ""; |
|
|
|
const normalizedUploadedFileStorage = |
|
|
|
uploadedFileStorage || uploadData.storage_provider || uploadData.storageProvider || ""; |
|
|
|
superAgentLog("chat.upload", "upload.done", { |
|
|
|
sessionId, |
|
|
|
conversationId, |
|
|
|
uploadedFileName, |
|
|
|
uploadedFilePath, |
|
|
|
uploadedLocalPath, |
|
|
|
uploadedFileStorage: normalizedUploadedFileStorage, |
|
|
|
uploadResult, |
|
|
|
attachmentId: uploadIntent.attachmentId, |
|
|
|
}); |
|
|
|
const uploadedFile = toReadyUploadedFile({ |
|
|
|
...readyAttachment, |
|
|
|
id: uploadIntent.attachmentId, |
|
|
|
file_name: uploadedFileName, |
|
|
|
status: "ready", |
|
|
|
}); |
|
|
|
const uploadedFile = { |
|
|
|
id: `${sessionId}_${uploadedFileName}_${Date.now()}`, |
|
|
|
attachmentId: uploadData.id || "", |
|
|
|
name: uploadedFileName, |
|
|
|
url: uploadedFilePath, |
|
|
|
objectKey: uploadData.object_key || uploadData.objectKey || "", |
|
|
|
localPath: uploadedLocalPath, |
|
|
|
storage: normalizedUploadedFileStorage, |
|
|
|
}; |
|
|
|
|
|
|
|
if (shouldCreateNewTask) { |
|
|
|
updateActiveTask({ |
|
|
|
title: uploadedFileName.replace(/\.[^.]+$/, "") || "AI 标书任务", |
|
|
|
fileName: uploadedFileName, |
|
|
|
filePath: uploadedFilePath, |
|
|
|
localPath: uploadedLocalPath, |
|
|
|
fileStorage: normalizedUploadedFileStorage, |
|
|
|
}); |
|
|
|
} |
|
|
|
|
|
|
|
@ -1677,6 +1683,7 @@ const useSuperAgentChat = ({ |
|
|
|
}); |
|
|
|
message.error(error?.message || "上传文件失败"); |
|
|
|
} finally { |
|
|
|
uploadAbortControllerRef.current = null; |
|
|
|
setIsUploadingFile(false); |
|
|
|
setUploadProgress(null); |
|
|
|
} |
|
|
|
@ -1689,6 +1696,10 @@ const useSuperAgentChat = ({ |
|
|
|
setUploadedFileList((prev) => prev.filter((file) => file.id !== fileId)); |
|
|
|
}; |
|
|
|
|
|
|
|
const handleCancelUpload = () => { |
|
|
|
uploadAbortControllerRef.current?.abort(); |
|
|
|
}; |
|
|
|
|
|
|
|
const confirmBasicInfo = () => { |
|
|
|
superAgentLog("chat.confirm", "basic_info", { activeSessionId, activeTaskId }); |
|
|
|
setTaskStage(TASK_STAGE.SCORE); |
|
|
|
@ -1727,8 +1738,9 @@ const useSuperAgentChat = ({ |
|
|
|
const handleSend = () => { |
|
|
|
const userInput = inputValue.trim(); |
|
|
|
const currentFileList = uploadedFileList; |
|
|
|
const fileReferenceText = buildUploadedFileReference(currentFileList); |
|
|
|
const finalMessage = [fileReferenceText, userInput].filter(Boolean).join("\n"); |
|
|
|
const finalMessage = userInput || (currentFileList.length |
|
|
|
? `请处理已上传的附件：${currentFileList.map((file) => file.name).filter(Boolean).join("、")}` |
|
|
|
: ""); |
|
|
|
|
|
|
|
if (!finalMessage || isStreamingAnswer) return; |
|
|
|
|
|
|
|
@ -1770,7 +1782,10 @@ const useSuperAgentChat = ({ |
|
|
|
//[文件引用写入历史展示字段，刷新或切换会话后才能还原文件标签]
|
|
|
|
displayContent: finalMessage, |
|
|
|
attachmentIds: Array.from( |
|
|
|
new Set(currentFileList.map((file) => file.attachmentId).filter(Boolean)) |
|
|
|
new Set(currentFileList |
|
|
|
.filter((file) => file.status === "ready") |
|
|
|
.map((file) => file.attachmentId) |
|
|
|
.filter(Boolean)) |
|
|
|
), |
|
|
|
}); |
|
|
|
}; |
|
|
|
@ -1887,6 +1902,7 @@ const useSuperAgentChat = ({ |
|
|
|
handleTenderQuickStart, |
|
|
|
handleUploadFile, |
|
|
|
handleRemoveUploadedFile, |
|
|
|
handleCancelUpload, |
|
|
|
confirmBasicInfo, |
|
|
|
confirmScoreItems, |
|
|
|
confirmCatalog, |
|
|
|
|
Senor-Liu
1 month ago