153 lines
5.0 KiB
TypeScript
153 lines
5.0 KiB
TypeScript
import { defineEventHandler, createError, getCookie } from 'h3'
|
|
import { query, queryOne, execute } from '../../../utils/db'
|
|
import OpenAI from 'openai'
|
|
|
|
const openai = new OpenAI({
|
|
apiKey: process.env.OPENAI_API_KEY
|
|
})
|
|
|
|
/**
|
|
* 기존 취합 보고서에 AI 요약 일괄 생성
|
|
* POST /api/report/summary/regenerate-ai
|
|
*/
|
|
export default defineEventHandler(async (event) => {
|
|
const userId = getCookie(event, 'user_id')
|
|
if (!userId) {
|
|
throw createError({ statusCode: 401, message: '로그인이 필요합니다.' })
|
|
}
|
|
|
|
// AI 요약이 없는 취합 보고서 조회
|
|
const summaries = await query<any>(`
|
|
SELECT s.summary_id, s.project_id, s.report_year, s.report_week,
|
|
p.project_name
|
|
FROM wr_aggregated_report_summary s
|
|
JOIN wr_project_info p ON s.project_id = p.project_id
|
|
WHERE s.ai_work_summary IS NULL OR s.ai_plan_summary IS NULL
|
|
ORDER BY s.summary_id
|
|
`, [])
|
|
|
|
console.log(`AI 요약 생성 대상: ${summaries.length}건`)
|
|
|
|
let successCount = 0
|
|
let errorCount = 0
|
|
|
|
for (const summary of summaries) {
|
|
try {
|
|
// 해당 프로젝트/주차의 Task 조회
|
|
const tasks = await query<any>(`
|
|
SELECT
|
|
t.task_type,
|
|
t.task_description,
|
|
t.task_hours,
|
|
t.is_completed,
|
|
e.employee_name as author_name
|
|
FROM wr_weekly_report r
|
|
JOIN wr_weekly_report_task t ON r.report_id = t.report_id
|
|
JOIN wr_employee_info e ON r.author_id = e.employee_id
|
|
WHERE t.project_id = $1
|
|
AND r.report_year = $2
|
|
AND r.report_week = $3
|
|
AND r.report_status IN ('SUBMITTED', 'AGGREGATED')
|
|
ORDER BY t.task_type, e.employee_name
|
|
`, [summary.project_id, summary.report_year, summary.report_week])
|
|
|
|
if (tasks.length === 0) {
|
|
console.log(`Skip ${summary.summary_id}: no tasks`)
|
|
continue
|
|
}
|
|
|
|
// AI 요약 생성
|
|
const { workSummary, planSummary } = await generateAISummary(
|
|
tasks,
|
|
summary.project_name,
|
|
summary.report_year,
|
|
summary.report_week
|
|
)
|
|
|
|
// 업데이트
|
|
await execute(`
|
|
UPDATE wr_aggregated_report_summary
|
|
SET ai_work_summary = $1,
|
|
ai_plan_summary = $2,
|
|
ai_summary_at = NOW()
|
|
WHERE summary_id = $3
|
|
`, [workSummary, planSummary, summary.summary_id])
|
|
|
|
successCount++
|
|
console.log(`Generated AI summary for ${summary.project_name} (${summary.report_year}-W${summary.report_week})`)
|
|
|
|
} catch (e: any) {
|
|
console.error(`Error for summary ${summary.summary_id}:`, e.message)
|
|
errorCount++
|
|
}
|
|
}
|
|
|
|
return {
|
|
success: true,
|
|
total: summaries.length,
|
|
successCount,
|
|
errorCount
|
|
}
|
|
})
|
|
|
|
async function generateAISummary(tasks: any[], projectName: string, year: number, week: number) {
|
|
const workTasks = tasks.filter(t => t.task_type === 'WORK')
|
|
const planTasks = tasks.filter(t => t.task_type === 'PLAN')
|
|
|
|
const workPrompt = `주간보고 취합 전문가입니다. 아래 금주 실적을 간결하게 요약해주세요.
|
|
|
|
## 프로젝트: ${projectName}
|
|
## 기간: ${year}년 ${week}주차
|
|
|
|
## 금주 실적 (${workTasks.length}건)
|
|
${workTasks.map(t => `- [${t.author_name}] ${t.is_completed ? '완료' : '진행중'} | ${t.task_description} (${t.task_hours}h)`).join('\n') || '없음'}
|
|
|
|
## 요약 규칙
|
|
1. 완료된 작업과 진행 중인 작업을 구분하여 핵심만 요약
|
|
2. 동일/유사한 작업은 하나로 통합
|
|
3. 담당자 이름은 생략하고 내용 위주로 작성
|
|
4. 3~5줄 이내로 간결하게
|
|
5. 마크다운 리스트 형식으로 작성`
|
|
|
|
const planPrompt = `주간보고 취합 전문가입니다. 아래 차주 계획을 간결하게 요약해주세요.
|
|
|
|
## 프로젝트: ${projectName}
|
|
## 기간: ${year}년 ${week+1}주차 계획
|
|
|
|
## 차주 계획 (${planTasks.length}건)
|
|
${planTasks.map(t => `- [${t.author_name}] ${t.task_description} (${t.task_hours}h)`).join('\n') || '없음'}
|
|
|
|
## 요약 규칙
|
|
1. 주요 계획을 우선순위에 따라 요약
|
|
2. 동일/유사한 작업은 하나로 통합
|
|
3. 담당자 이름은 생략하고 내용 위주로 작성
|
|
4. 2~4줄 이내로 간결하게
|
|
5. 마크다운 리스트 형식으로 작성`
|
|
|
|
const [workRes, planRes] = await Promise.all([
|
|
openai.chat.completions.create({
|
|
model: 'gpt-4o-mini',
|
|
messages: [
|
|
{ role: 'system', content: '주간보고 취합 전문가입니다. 간결하게 요약합니다.' },
|
|
{ role: 'user', content: workPrompt }
|
|
],
|
|
temperature: 0.3,
|
|
max_tokens: 500
|
|
}),
|
|
openai.chat.completions.create({
|
|
model: 'gpt-4o-mini',
|
|
messages: [
|
|
{ role: 'system', content: '주간보고 취합 전문가입니다. 간결하게 요약합니다.' },
|
|
{ role: 'user', content: planPrompt }
|
|
],
|
|
temperature: 0.3,
|
|
max_tokens: 500
|
|
})
|
|
])
|
|
|
|
return {
|
|
workSummary: workRes.choices[0]?.message?.content || '요약 없음',
|
|
planSummary: planRes.choices[0]?.message?.content || '요약 없음'
|
|
}
|
|
}
|