358 lines
13 KiB
JavaScript
358 lines
13 KiB
JavaScript
import * as aiAttachmentAssociationModel from '../../utils/aiAttachmentAssociationModel.js'
|
||
import { syncExpenseClaimFilesToDraft } from '../../utils/expenseClaimAttachmentSync.js'
|
||
import { collectReceiptFiles } from '../../views/scripts/travelReimbursementAttachmentModel.js'
|
||
import {
|
||
createExpenseClaimItem,
|
||
extractExpenseClaimItems,
|
||
fetchExpenseClaimDetail,
|
||
fetchExpenseClaims,
|
||
uploadExpenseClaimItemAttachment
|
||
} from '../../services/reimbursements.js'
|
||
import { recognizeOcrFiles } from '../../services/ocr.js'
|
||
import {
|
||
AI_ATTACHMENT_ASSOCIATION_CONFIRM_ACTION,
|
||
buildInlineAttachmentOcrDetails
|
||
} from './workbenchAiMessageModel.js'
|
||
import { isLikelyAiModeOcrFile } from './workbenchAiComposerModel.js'
|
||
|
||
function buildAiAttachmentAssociationThinkingEvents(status = 'running') {
|
||
const completed = status === 'completed'
|
||
const failed = status === 'failed'
|
||
const eventStatus = failed ? 'failed' : completed ? 'completed' : 'running'
|
||
return [
|
||
{
|
||
eventId: 'attachment-ocr',
|
||
title: '识别上传票据',
|
||
content: '提取票据里的日期、地点和行程信息。',
|
||
status: eventStatus
|
||
},
|
||
{
|
||
eventId: 'claim-lookup',
|
||
title: '查询可关联报销单',
|
||
content: '查找草稿、待补充和退回状态的可归集单据。',
|
||
status: eventStatus
|
||
},
|
||
{
|
||
eventId: 'claim-match',
|
||
title: '匹配票据与报销单',
|
||
content: '根据票据时间、城市和报销事由判断最可能的关联单据。',
|
||
status: eventStatus
|
||
}
|
||
]
|
||
}
|
||
|
||
function resolveAiAttachmentAssociationClaimNo(payload = {}) {
|
||
return String(payload?.claim_no || payload?.claimNo || '').trim()
|
||
}
|
||
|
||
function buildAiAttachmentAssociationResultThinkingEvents(status = 'running') {
|
||
const completed = status === 'completed'
|
||
const failed = status === 'failed'
|
||
const eventStatus = failed ? 'failed' : completed ? 'completed' : 'running'
|
||
return [
|
||
{
|
||
eventId: 'attachment-confirm',
|
||
title: '确认自动归集',
|
||
content: '正在读取匹配单据并准备写入附件。',
|
||
status: eventStatus
|
||
},
|
||
{
|
||
eventId: 'attachment-upload',
|
||
title: '归集票据附件',
|
||
content: '把本次上传的票据写入报销单明细。',
|
||
status: eventStatus
|
||
}
|
||
]
|
||
}
|
||
|
||
export function useWorkbenchAiAttachmentAssociationFlow({
|
||
aiAttachmentAssociationRuntime,
|
||
conversationMessages,
|
||
createAiAttachmentAssociationId,
|
||
createInlineMessage,
|
||
inlineConversationAutoScrollPinned,
|
||
persistCurrentConversation,
|
||
replaceInlineMessage,
|
||
scrollInlineConversationToBottom,
|
||
sending,
|
||
streamOrSetInlineAssistantContent,
|
||
toast
|
||
}) {
|
||
async function collectAiModeReceiptContext(files = []) {
|
||
const safeFiles = Array.isArray(files) ? files : []
|
||
const attachmentNames = safeFiles
|
||
.map((file) => String(file?.name || '').trim())
|
||
.filter(Boolean)
|
||
const ocrFiles = safeFiles.filter((file) => isLikelyAiModeOcrFile(file))
|
||
const ocrSourceFileNames = ocrFiles
|
||
.map((file) => String(file?.name || '').trim())
|
||
.filter(Boolean)
|
||
|
||
const baseContext = {
|
||
attachmentNames,
|
||
attachmentCount: attachmentNames.length,
|
||
ocrSourceFileNames,
|
||
ocrSummary: '',
|
||
ocrDocuments: []
|
||
}
|
||
|
||
if (!ocrFiles.length) {
|
||
return baseContext
|
||
}
|
||
|
||
try {
|
||
const collected = await collectReceiptFiles({
|
||
files: ocrFiles,
|
||
recognizeOcrFiles
|
||
})
|
||
return {
|
||
...baseContext,
|
||
ocrSummary: String(collected.ocrSummary || '').trim(),
|
||
ocrDocuments: Array.isArray(collected.ocrDocuments) ? collected.ocrDocuments : []
|
||
}
|
||
} catch (error) {
|
||
console.warn('AI mode OCR request failed:', error)
|
||
return {
|
||
...baseContext,
|
||
ocrError: error?.message || 'OCR识别失败，已继续使用附件名称。'
|
||
}
|
||
}
|
||
}
|
||
|
||
function findAiAttachmentAssociationRuntime(options = {}) {
|
||
const normalizedAssociationId = String(options.associationId || options.association_id || '').trim()
|
||
const normalizedClaimNo = String(options.claimNo || options.claim_no || '').trim()
|
||
if (normalizedAssociationId) {
|
||
const runtime = aiAttachmentAssociationRuntime.get(normalizedAssociationId)
|
||
if (runtime) {
|
||
return { associationId: normalizedAssociationId, runtime }
|
||
}
|
||
}
|
||
|
||
if (normalizedClaimNo) {
|
||
for (const [runtimeId, runtime] of aiAttachmentAssociationRuntime.entries()) {
|
||
if (String(runtime?.claimNo || '').trim() === normalizedClaimNo && runtime?.files?.length) {
|
||
return { associationId: runtimeId, runtime }
|
||
}
|
||
}
|
||
}
|
||
|
||
return { associationId: '', runtime: null }
|
||
}
|
||
|
||
function updateAiAttachmentAssociationActionState(message = {}, associationId = '', state = {}, options = {}) {
|
||
const normalizedAssociationId = String(associationId || '').trim()
|
||
const normalizedClaimNo = String(options.claimNo || options.claim_no || '').trim()
|
||
if (!message || !Array.isArray(message.suggestedActions) || (!normalizedAssociationId && !normalizedClaimNo)) {
|
||
return
|
||
}
|
||
message.suggestedActions = message.suggestedActions.map((action) => {
|
||
const actionAssociationId = String(action?.payload?.association_id || action?.payload?.associationId || '').trim()
|
||
const actionClaimNo = resolveAiAttachmentAssociationClaimNo(action?.payload || {})
|
||
const isSameAssociation = normalizedAssociationId && actionAssociationId === normalizedAssociationId
|
||
const isSameClaim = normalizedClaimNo && actionClaimNo === normalizedClaimNo
|
||
if (String(action?.action_type || '').trim() !== AI_ATTACHMENT_ASSOCIATION_CONFIRM_ACTION || (!isSameAssociation && !isSameClaim)) {
|
||
return action
|
||
}
|
||
return {
|
||
...action,
|
||
...state
|
||
}
|
||
})
|
||
}
|
||
|
||
function buildAiAttachmentAssociationDetailActions(runtime = {}) {
|
||
const claimNo = String(runtime.claimNo || '').trim()
|
||
const claimId = String(runtime.claimId || '').trim()
|
||
if (!claimNo && !claimId) {
|
||
return []
|
||
}
|
||
return [{
|
||
label: '查看单据',
|
||
description: '打开已归集票据的报销单。',
|
||
icon: 'mdi mdi-open-in-new',
|
||
action_type: 'open_application_detail',
|
||
payload: {
|
||
claim_id: claimId,
|
||
claim_no: claimNo,
|
||
document_type: 'expense'
|
||
}
|
||
}]
|
||
}
|
||
|
||
async function confirmAiAttachmentAssociation(actionPayload = {}, sourceMessage = null) {
|
||
const requestedAssociationId = String(actionPayload.association_id || actionPayload.associationId || '').trim()
|
||
const payloadClaimNo = resolveAiAttachmentAssociationClaimNo(actionPayload)
|
||
const runtimeResult = findAiAttachmentAssociationRuntime({
|
||
associationId: requestedAssociationId,
|
||
claimNo: payloadClaimNo
|
||
})
|
||
const associationId = runtimeResult.associationId
|
||
const runtime = runtimeResult.runtime
|
||
const actionClaimNo = payloadClaimNo || String(runtime?.claimNo || '').trim()
|
||
if (!associationId || !runtime?.files?.length) {
|
||
toast('当前会话里没有可归集的附件原件，请重新上传票据后再确认。')
|
||
return
|
||
}
|
||
|
||
updateAiAttachmentAssociationActionState(sourceMessage, associationId, {
|
||
label: '正在归集...',
|
||
disabled: true
|
||
}, { claimNo: actionClaimNo })
|
||
persistCurrentConversation()
|
||
sending.value = true
|
||
|
||
const pendingMessage = createInlineMessage('assistant', '', {
|
||
pending: true,
|
||
stewardPlan: {
|
||
streamStatus: 'streaming',
|
||
thinkingEvents: buildAiAttachmentAssociationResultThinkingEvents('running')
|
||
}
|
||
})
|
||
conversationMessages.value.push(pendingMessage)
|
||
scrollInlineConversationToBottom()
|
||
|
||
try {
|
||
const syncResult = await syncExpenseClaimFilesToDraft({
|
||
claimId: runtime.claimId,
|
||
files: runtime.files,
|
||
fetchExpenseClaimDetail,
|
||
createExpenseClaimItem,
|
||
uploadExpenseClaimItemAttachment
|
||
})
|
||
const finalMessageText = aiAttachmentAssociationModel.buildAiAttachmentAssociationResultMessage({
|
||
claimNo: runtime.claimNo,
|
||
fileNames: runtime.fileNames,
|
||
uploadedCount: syncResult.uploadedCount,
|
||
skippedCount: syncResult.skippedCount
|
||
})
|
||
await streamOrSetInlineAssistantContent(pendingMessage.id, finalMessageText)
|
||
replaceInlineMessage(
|
||
pendingMessage.id,
|
||
createInlineMessage('assistant', finalMessageText, {
|
||
id: pendingMessage.id,
|
||
stewardPlan: {
|
||
streamStatus: 'completed',
|
||
thinkingEvents: buildAiAttachmentAssociationResultThinkingEvents('completed')
|
||
},
|
||
suggestedActions: buildAiAttachmentAssociationDetailActions(runtime)
|
||
})
|
||
)
|
||
updateAiAttachmentAssociationActionState(sourceMessage, associationId, {
|
||
label: '已自动关联',
|
||
disabled: true
|
||
}, { claimNo: actionClaimNo })
|
||
aiAttachmentAssociationRuntime.delete(associationId)
|
||
persistCurrentConversation()
|
||
} catch (error) {
|
||
const finalMessageText = error?.message || '自动归集失败，请稍后重试。'
|
||
replaceInlineMessage(
|
||
pendingMessage.id,
|
||
createInlineMessage('assistant', finalMessageText, {
|
||
id: pendingMessage.id,
|
||
stewardPlan: {
|
||
streamStatus: 'failed',
|
||
thinkingEvents: buildAiAttachmentAssociationResultThinkingEvents('failed')
|
||
}
|
||
})
|
||
)
|
||
updateAiAttachmentAssociationActionState(sourceMessage, associationId, {
|
||
label: '重新自动关联',
|
||
disabled: false
|
||
}, { claimNo: actionClaimNo })
|
||
toast(finalMessageText)
|
||
persistCurrentConversation()
|
||
} finally {
|
||
sending.value = false
|
||
scrollInlineConversationToBottom({ force: inlineConversationAutoScrollPinned.value })
|
||
}
|
||
}
|
||
|
||
async function requestAiAttachmentAssociationReply(prompt, entry = {}, files = []) {
|
||
let shouldAutoScrollOnFinish = true
|
||
const pendingMessage = createInlineMessage('assistant', '', {
|
||
pending: true,
|
||
stewardPlan: {
|
||
streamStatus: 'streaming',
|
||
thinkingEvents: buildAiAttachmentAssociationThinkingEvents('running')
|
||
}
|
||
})
|
||
conversationMessages.value.push(pendingMessage)
|
||
scrollInlineConversationToBottom()
|
||
|
||
try {
|
||
const collected = await collectReceiptFiles({
|
||
files,
|
||
recognizeOcrFiles
|
||
})
|
||
const attachmentOcrDetails = buildInlineAttachmentOcrDetails(collected, files)
|
||
const claimsPayload = await fetchExpenseClaims({ page: 1, pageSize: 100 })
|
||
const claims = extractExpenseClaimItems(claimsPayload)
|
||
const match = aiAttachmentAssociationModel.resolveAiAttachmentAssociationMatch(claims, collected.ocrDocuments)
|
||
const associationRecord = match.best?.record || match.recommended?.record || null
|
||
const associationId = associationRecord?.claimId
|
||
? createAiAttachmentAssociationId()
|
||
: ''
|
||
if (associationId) {
|
||
aiAttachmentAssociationRuntime.set(associationId, {
|
||
files,
|
||
fileNames: files.map((file) => file?.name || '').filter(Boolean),
|
||
claimId: String(associationRecord.claimId || '').trim(),
|
||
claimNo: String(associationRecord.claimNo || '').trim(),
|
||
ocrPayload: collected.ocrPayload,
|
||
ocrSummary: collected.ocrSummary,
|
||
ocrDocuments: collected.ocrDocuments,
|
||
ocrFilePreviews: collected.ocrFilePreviews
|
||
})
|
||
}
|
||
const finalMessageText = aiAttachmentAssociationModel.buildAiAttachmentAssociationMessage({
|
||
match,
|
||
fileNames: files.map((file) => file?.name || ''),
|
||
ocrDocuments: collected.ocrDocuments
|
||
})
|
||
await streamOrSetInlineAssistantContent(pendingMessage.id, finalMessageText)
|
||
shouldAutoScrollOnFinish = inlineConversationAutoScrollPinned.value
|
||
replaceInlineMessage(
|
||
pendingMessage.id,
|
||
createInlineMessage('assistant', finalMessageText, {
|
||
id: pendingMessage.id,
|
||
stewardPlan: {
|
||
streamStatus: 'completed',
|
||
thinkingEvents: buildAiAttachmentAssociationThinkingEvents('completed')
|
||
},
|
||
attachmentOcrDetails,
|
||
suggestedActions: aiAttachmentAssociationModel.buildAiAttachmentAssociationActions(match, associationId, {
|
||
includeOcrDetails: Boolean(attachmentOcrDetails)
|
||
})
|
||
})
|
||
)
|
||
persistCurrentConversation()
|
||
} catch (error) {
|
||
shouldAutoScrollOnFinish = inlineConversationAutoScrollPinned.value
|
||
const finalMessageText = error?.message || '票据识别或单据匹配失败，请稍后再试。'
|
||
replaceInlineMessage(
|
||
pendingMessage.id,
|
||
createInlineMessage('assistant', finalMessageText, {
|
||
id: pendingMessage.id,
|
||
stewardPlan: {
|
||
streamStatus: 'failed',
|
||
thinkingEvents: buildAiAttachmentAssociationThinkingEvents('failed')
|
||
}
|
||
})
|
||
)
|
||
toast(finalMessageText)
|
||
persistCurrentConversation()
|
||
} finally {
|
||
sending.value = false
|
||
scrollInlineConversationToBottom({ force: shouldAutoScrollOnFinish })
|
||
}
|
||
}
|
||
|
||
return {
|
||
collectAiModeReceiptContext,
|
||
confirmAiAttachmentAssociation,
|
||
requestAiAttachmentAssociationReply,
|
||
resolveAiAttachmentAssociationClaimNo
|
||
}
|
||
}
|