feat: Pipeline 전면 개선 — 병렬실행, Batch Review, 총평, 대화기억, 스마트라우팅

- GeminiCaller: cmd/c 제거, 인자 분리, Semaphore(4) 동시성 제어, GeminiCallError - TaskPipeline: asyncio.gather 병렬 코딩, batch_review 1회, summarize 총평 - FileApplier: Coder 출력 파싱 → 실제 파일 적용 (경로 보안 체크) - Discord Bot: on_message 자동채팅, 의도분류(chat/task/clarify), 대화기억(10메시지) - Prompts: router.md (의도분류), summarizer.md (총평) - Workflows: agent_chat 환경 경로 업데이트
2026-03-06 20:46:58 +09:00
parent 4c0f5ec9c7
commit 752d851f9f
10 changed files with 783 additions and 190 deletions
--- a/api/discord_bot.py
+++ b/api/discord_bot.py
@@ -1,21 +1,26 @@
 """Discord Bot 어댑터.

-사용자 명령을 받아 FastAPI API를 호출하고 결과를 디스코드로 보고합니다.
+사용자 명령을 받아 파이프라인 실행 또는 즉답.
+지정 채널(봇 이름 채널)에서는 접두사 없이 자연스럽게 대화합니다.
+대화 기억: 채널별 최근 메시지를 컨텍스트로 주입.
 """

 import asyncio
+import json
 import logging
+import re

 import discord
 from discord.ext import commands

 import config
+from core.gemini_caller import GeminiCallError

 logger = logging.getLogger("variet.discord")

 # Bot 설정
 intents = discord.Intents.default()
-intents.message_content = True  # MESSAGE CONTENT INTENT 필요
+intents.message_content = True

 bot = commands.Bot(
    command_prefix=config.DISCORD_COMMAND_PREFIX,
@@ -23,56 +28,217 @@ bot = commands.Bot(
    help_command=commands.DefaultHelpCommand(no_category="명령어"),
 )

-# In-memory: Discord 채널 ↔ Task 매핑
+# In-memory
 _channel_tasks: dict[int, list[str]] = {}
+_auto_chat_channel_ids: set[int] = set()


+# ──────────────────────────────────────────────
+#  대화 기억 (Conversation Memory)
+# ──────────────────────────────────────────────
+
+async def _get_channel_history(channel: discord.TextChannel, limit: int = 10) -> str:
+    """채널의 최근 메시지를 대화 히스토리 문자열로 변환."""
+    messages = []
+    async for msg in channel.history(limit=limit + 1):  # 현재 메시지 포함이므로 +1
+        if msg.author.bot:
+            role = "assistant"
+        else:
+            role = "user"
+        messages.append(f"[{role}] {msg.content[:300]}")
+
+    # 시간순 (오래된 것 먼저)
+    messages.reverse()
+
+    # 마지막(현재 메시지)은 제외 — 이미 context로 전달되니까
+    if messages:
+        messages = messages[:-1]
+
+    if not messages:
+        return ""
+
+    return "=== CONVERSATION HISTORY ===\n" + "\n".join(messages) + "\n\n"
+
+
+# ──────────────────────────────────────────────
+#  의도 분류 (Intent Router)
+# ──────────────────────────────────────────────
+
+async def _classify_intent(message: str) -> dict:
+    """Gemini로 사용자 메시지의 의도를 분류."""
+    from core.gemini_caller import GeminiCaller
+
+    gemini = GeminiCaller(str(config.PROJECT_ROOT))
+    try:
+        raw = await gemini.call("router", message, timeout=30)
+    except GeminiCallError as e:
+        logger.warning(f"Intent 분류 실패: {e}")
+        return {"intent": "chat", "reason": "분류 실패"}
+
+    try:
+        json_match = re.search(r'\{[^}]+\}', raw)
+        if json_match:
+            return json.loads(json_match.group())
+    except (json.JSONDecodeError, AttributeError):
+        pass
+
+    logger.warning(f"Intent JSON 파싱 실패, chat으로 처리: {raw[:100]}")
+    return {"intent": "chat", "reason": "파싱 실패 — 기본 chat"}
+
+
+# ──────────────────────────────────────────────
+#  이벤트 핸들러
+# ──────────────────────────────────────────────
+
@bot.event
 async def on_ready():
-    """봇 접속 완료."""
+    """봇 접속 완료 — 봇 이름 채널 자동 감지."""
    logger.info(f"Discord Bot 접속 완료: {bot.user} (ID: {bot.user.id})")
    logger.info(f"서버 {len(bot.guilds)}개 연결됨")
+
+    bot_name = bot.user.name.lower().replace(" ", "-")
+    bot_name_underscore = bot_name.replace("-", "_")
+    bot_name_dash = bot_name.replace("_", "-")
+
+    for guild in bot.guilds:
+        for channel in guild.text_channels:
+            ch_name = channel.name.lower()
+            if ch_name in (bot_name, bot_name_underscore, bot_name_dash):
+                _auto_chat_channel_ids.add(channel.id)
+                logger.info(
+                    f"자동 채팅 채널 감지: #{channel.name} (ID: {channel.id}) "
+                    f"in {guild.name}"
+                )
+
+    if not _auto_chat_channel_ids:
+        logger.info(f"봇 이름({bot_name}) 채널 없음 — !명령어만 사용 가능")
+
    await bot.change_presence(
        activity=discord.Activity(
            type=discord.ActivityType.listening,
-            name=f"{config.DISCORD_COMMAND_PREFIX}agent",
+            name="대화" if _auto_chat_channel_ids else f"{config.DISCORD_COMMAND_PREFIX}agent",
        )
    )


-@bot.command(name="agent", help="AI Agent에게 작업 요청\n예: !agent README에 설치 방법 추가해줘")
-async def agent_command(ctx: commands.Context, *, request: str):
-    """작업 요청 → Pipeline 실행 → 결과 보고."""
+@bot.event
+async def on_message(message: discord.Message):
+    """모든 메시지 수신 — 봇 전용 채널이면 스마트 라우팅."""
+    if message.author == bot.user or message.author.bot:
+        return

-    # 1. 접수 메시지
+    # ! 명령어는 기존 핸들러로
+    if message.content.startswith(config.DISCORD_COMMAND_PREFIX):
+        await bot.process_commands(message)
+        return
+
+    # 봇 전용 채널이 아니면 무시
+    if message.channel.id not in _auto_chat_channel_ids:
+        return
+
+    await _route_message(message)
+
+
+# ──────────────────────────────────────────────
+#  스마트 라우팅
+# ──────────────────────────────────────────────
+
+async def _route_message(message: discord.Message):
+    """메시지 의도를 분류하고 적절한 핸들러로 라우팅."""
+    user_text = message.content.strip()
+    if not user_text:
+        return
+
+    # 짧은 메시지 → 분류 없이 바로 chat
+    if len(user_text) <= 15:
+        await _handle_chat(message, user_text)
+        return
+
+    # 의도 분류
+    async with message.channel.typing():
+        intent_result = await _classify_intent(user_text)
+
+    intent = intent_result.get("intent", "chat")
+    reason = intent_result.get("reason", "")
+    logger.info(f"의도 분류: {intent} ({reason}) — \"{user_text[:50]}\"")
+
+    if intent == "task":
+        await _handle_task(message, user_text)
+    elif intent == "clarify":
+        await _handle_clarify(message, user_text, reason)
+    else:
+        await _handle_chat(message, user_text)
+
+
+# ──────────────────────────────────────────────
+#  핸들러: Chat (즉답 + 대화 기억)
+# ──────────────────────────────────────────────
+
+async def _handle_chat(message: discord.Message, text: str):
+    """즉답 — 대화 히스토리 포함하여 Gemini 직접 호출."""
+    async with message.channel.typing():
+        try:
+            from core.gemini_caller import GeminiCaller
+
+            # 대화 기억: 최근 10개 메시지
+            history = await _get_channel_history(message.channel, limit=10)
+            context = f"{history}{text}"
+
+            gemini = GeminiCaller(str(config.PROJECT_ROOT))
+            response = await gemini.call("default", context, timeout=60)
+
+            if not response:
+                await message.reply("⚠️ 응답을 생성하지 못했어요. 다시 시도해 주세요.")
+                return
+
+            # Discord 2000자 제한
+            if len(response) <= 2000:
+                await message.reply(response)
+            else:
+                for i in range(0, len(response), 4000):
+                    chunk = response[i:i + 4000]
+                    embed = discord.Embed(description=chunk, color=0x3498DB)
+                    await message.channel.send(embed=embed)
+
+        except GeminiCallError as e:
+            logger.error(f"Chat Gemini 오류: {e}")
+            await message.reply(f"⚠️ AI 호출 오류: {str(e)[:200]}")
+        except Exception as e:
+            logger.error(f"Chat 오류: {e}", exc_info=True)
+            await message.reply(f"❌ 오류: {str(e)[:200]}")
+
+
+# ──────────────────────────────────────────────
+#  핸들러: Task (파이프라인 실행)
+# ──────────────────────────────────────────────
+
+async def _handle_task(message: discord.Message, text: str):
+    """작업 요청 — 새 파이프라인 (병렬 Code + Batch Review + 총평)."""
+    import uuid
+
+    task_id = uuid.uuid4().hex[:8]
+
+    # 접수 메시지
    embed = discord.Embed(
        title="📋 작업 접수",
-        description=f"```{request[:200]}```",
+        description=f"```{text[:200]}```",
        color=0x3498DB,
    )
-    embed.set_footer(text="분석 중...")
-    status_msg = await ctx.send(embed=embed)
+    embed.set_footer(text=f"ID: {task_id} — 분석 중...")
+    status_msg = await message.channel.send(embed=embed)

    try:
-        # 2. Pipeline 직접 실행 (같은 프로세스)
        from core.task_pipeline import TaskPipeline
-        import uuid

-        task_id = uuid.uuid4().hex[:8]
+        pipeline = TaskPipeline(project_path=str(config.PROJECT_ROOT))
+        pipeline.setup()

-        # Planning
+        # ── Plan ──
        embed.color = 0xF39C12
        embed.set_footer(text=f"🔍 작업 분해 중... (ID: {task_id})")
        await status_msg.edit(embed=embed)

-        pipeline = TaskPipeline(
-            project_path=str(config.PROJECT_ROOT),
-        )
-        pipeline.setup()
-
-        # Plan 단계
-        plan = await pipeline.plan(request)
-
+        plan = await pipeline.plan(text)
        tasks = plan.get("tasks", [])
        plan_text = plan.get("summary", str(plan))[:500]

@@ -92,82 +258,178 @@ async def agent_command(ctx: commands.Context, *, request: str):
                inline=False,
            )
        plan_embed.set_footer(text=f"ID: {task_id}")
-        await ctx.send(embed=plan_embed)
+        await message.channel.send(embed=plan_embed)

-        # Code + Review 단계
-        if tasks:
-            for i, task in enumerate(tasks, 1):
-                progress_embed = discord.Embed(
-                    title=f"⚙️ 실행 중 ({i}/{len(tasks)})",
-                    description=task.get("title", task.get("description", ""))[:200],
-                    color=0xE67E22,
+        if not tasks:
+            await message.channel.send(
+                embed=discord.Embed(
+                    title="⚠️ 실행할 태스크가 없습니다",
+                    description="요청을 더 구체적으로 해주세요.",
+                    color=0xF39C12,
                )
-                await ctx.send(embed=progress_embed)
+            )
+            return

-                code_output = await pipeline.code(task)
-                review = await pipeline.review(task, code_output)
-
-                passed = review.get("passed", True)
-                review_emoji = "✅" if passed else "⚠️"
-                review_embed = discord.Embed(
-                    title=f"{review_emoji} 리뷰 결과 ({i}/{len(tasks)})",
-                    description=review.get("summary", str(review))[:500],
-                    color=0x2ECC71 if passed else 0xE74C3C,
-                )
-                await ctx.send(embed=review_embed)
-
-        # 완료
-        done_embed = discord.Embed(
-            title="✅ 작업 완료",
-            description=f"총 {len(tasks)}개 태스크 처리 완료",
-            color=0x2ECC71,
+        # ── Code 병렬 실행 ──
+        code_embed = discord.Embed(
+            title=f"⚙️ 코딩 중... ({len(tasks)}개 병렬 실행)",
+            description="\n".join(
+                f"• {t.get('title', t.get('description', '?'))[:60]}"
+                for t in tasks
+            ),
+            color=0xE67E22,
        )
-        done_embed.set_footer(text=f"ID: {task_id}")
-        await ctx.send(embed=done_embed)
+        code_msg = await message.channel.send(embed=code_embed)

-        # 채널 태스크 기록
-        _channel_tasks.setdefault(ctx.channel.id, []).append(task_id)
+        code_outputs = await pipeline.code_parallel(tasks)

+        # 코딩 완료 표시
+        code_embed.title = f"✅ 코딩 완료 ({len(tasks)}개)"
+        code_embed.color = 0x2ECC71
+        await code_msg.edit(embed=code_embed)
+
+        # ── 파일 적용 ──
+        from core.file_applier import parse_code_output, apply_changes
+
+        all_applied = []
+        for output in code_outputs:
+            if not output.startswith("[ERROR]"):
+                changes = parse_code_output(output)
+                if changes:
+                    applied = apply_changes(changes, str(config.PROJECT_ROOT))
+                    all_applied.extend(applied)
+
+        if all_applied:
+            files_text = "\n".join(
+                f"• `{f['path']}` ({f['action']})"
+                for f in all_applied[:15]
+            )
+            await message.channel.send(
+                embed=discord.Embed(
+                    title=f"📁 파일 적용 ({len(all_applied)}개)",
+                    description=files_text,
+                    color=0x3498DB,
+                )
+            )
+
+        # ── Batch Review ──
+        review_msg = await message.channel.send(
+            embed=discord.Embed(
+                title="🔍 전체 리뷰 중...",
+                color=0xF39C12,
+            )
+        )
+
+        review = await pipeline.batch_review(tasks, code_outputs)
+        passed = review.get("passed", True)
+        review_embed = discord.Embed(
+            title=f"{'✅' if passed else '⚠️'} 리뷰 결과",
+            description=review.get("summary", str(review))[:500],
+            color=0x2ECC71 if passed else 0xE74C3C,
+        )
+        await review_msg.edit(embed=review_embed)
+
+        # ── 총평 ──
+        summary = await pipeline.summarize(
+            text, plan, code_outputs, review, all_applied
+        )
+
+        summary_embed = discord.Embed(
+            title=f"📊 {summary.get('title', '작업 완료')}",
+            description=summary.get("summary", "작업이 완료되었습니다."),
+            color=0x9B59B6,
+        )
+
+        # 변경 파일 요약
+        changes = summary.get("changes", [])
+        if changes:
+            changes_text = "\n".join(
+                f"• `{c.get('file', '?')}` — {c.get('description', '')}"
+                for c in changes[:10]
+            )
+            summary_embed.add_field(
+                name="변경 사항",
+                value=changes_text[:1000],
+                inline=False,
+            )
+
+        # 주의사항
+        warnings = summary.get("warnings", [])
+        if warnings:
+            summary_embed.add_field(
+                name="⚠️ 주의",
+                value="\n".join(f"• {w}" for w in warnings),
+                inline=False,
+            )
+
+        # 다음 단계 제안
+        next_steps = summary.get("next_steps", [])
+        if next_steps:
+            summary_embed.add_field(
+                name="🔜 다음 단계",
+                value="\n".join(f"• {s}" for s in next_steps),
+                inline=False,
+            )
+
+        summary_embed.set_footer(text=f"ID: {task_id}")
+        await message.channel.send(embed=summary_embed)
+
+        _channel_tasks.setdefault(message.channel.id, []).append(task_id)
+
+    except GeminiCallError as e:
+        logger.error(f"파이프라인 Gemini 오류: {e}")
+        await message.channel.send(
+            embed=discord.Embed(
+                title="❌ AI 호출 오류",
+                description=f"```{str(e)[:500]}```",
+                color=0xE74C3C,
+            )
+        )
    except Exception as e:
        logger.error(f"작업 실행 오류: {e}", exc_info=True)
-        error_embed = discord.Embed(
-            title="❌ 오류 발생",
-            description=f"```{str(e)[:500]}```",
-            color=0xE74C3C,
+        await message.channel.send(
+            embed=discord.Embed(
+                title="❌ 오류 발생",
+                description=f"```{str(e)[:500]}```",
+                color=0xE74C3C,
+            )
        )
-        await ctx.send(embed=error_embed)
+
+
+# ──────────────────────────────────────────────
+#  핸들러: Clarify (되묻기)
+# ──────────────────────────────────────────────
+
+async def _handle_clarify(message: discord.Message, text: str, reason: str):
+    """의도 불명확 — 사용자에게 되묻기."""
+    embed = discord.Embed(
+        title="🤔 확인이 필요해요",
+        description=(
+            f"말씀하신 내용을 정확히 이해하고 싶어요.\n\n"
+            f"> {text[:200]}\n\n"
+            f"**💬 질문/대화**인가요, **🔧 작업 요청**인가요?\n"
+            f"`질문` 또는 `작업`으로 답해주세요."
+        ),
+        color=0xF39C12,
+    )
+    embed.set_footer(text=f"사유: {reason}")
+    await message.reply(embed=embed)
+
+
+# ──────────────────────────────────────────────
+#  기존 ! 명령어 (유지)
+# ──────────────────────────────────────────────
+
+@bot.command(name="agent", help="AI Agent에게 작업 요청\n예: !agent README에 설치 방법 추가해줘")
+async def agent_command(ctx: commands.Context, *, request: str):
+    """작업 요청 → Pipeline 실행."""
+    await _handle_task(ctx.message, request)


@bot.command(name="chat", help="Gemini에게 질문/대화\n예: !chat 네 소개를 해줘")
 async def chat_command(ctx: commands.Context, *, message: str):
-    """단순 대화 — Pipeline 없이 Gemini 직접 호출."""
-    async with ctx.typing():
-        try:
-            from core.gemini_caller import GeminiCaller
-
-            gemini = GeminiCaller(str(config.PROJECT_ROOT))
-            response = await gemini.call(
-                role="default",
-                context=message,
-                timeout=60,
-            )
-
-            # Discord 메시지 2000자 제한 처리
-            if len(response) <= 2000:
-                await ctx.send(response)
-            else:
-                # 긴 응답은 Embed로 분할
-                for i in range(0, len(response), 4000):
-                    chunk = response[i:i + 4000]
-                    embed = discord.Embed(
-                        description=chunk,
-                        color=0x3498DB,
-                    )
-                    await ctx.send(embed=embed)
-
-        except Exception as e:
-            logger.error(f"Chat 오류: {e}", exc_info=True)
-            await ctx.send(f"❌ 오류: {str(e)[:200]}")
+    """단순 대화."""
+    await _handle_chat(ctx.message, message)


@bot.command(name="ping", help="봇 응답 테스트")
@@ -188,6 +450,16 @@ async def info_command(ctx: commands.Context):
    embed.add_field(name="프로젝트", value=str(config.PROJECT_ROOT), inline=False)
    embed.add_field(name="명령어 접두사", value=config.DISCORD_COMMAND_PREFIX, inline=True)
    embed.add_field(name="서버 수", value=str(len(bot.guilds)), inline=True)
+
+    if _auto_chat_channel_ids:
+        channels = ", ".join(f"<#{ch_id}>" for ch_id in _auto_chat_channel_ids)
+        embed.add_field(name="자동 채팅 채널", value=channels, inline=False)
+
+    embed.add_field(
+        name="파이프라인",
+        value="Plan → Code(병렬) → Review(배치) → 총평",
+        inline=False,
+    )
    await ctx.send(embed=embed)