feat: 워크스페이스 시스템 + 통합 프롬프트 + Docs 기록 관리
- workspace.py: 채널별 워크스페이스 모델 + JSON 영속 저장 - discord_bot.py: /workspace 슬래시 커맨드 (set/git/vikunja/info/remove/list) - 등록 채널만 자동 응답, 미등록 채널 무시 - Git/Vikunja 미설정 시 작업 차단 + 안내 - 통합 프롬프트 1회 호출 (router+planner+chat 통합) - docs_manager.py: Wiki 인덱스, 세션 기록, Changelog 자동 업데이트 - task_pipeline.py: 모든 Gemini 호출에 docs 컨텍스트 주입, 완료 시 기록 - unified.md: 분류+즉답/계획 통합 프롬프트
This commit is contained in:
@@ -1,8 +1,7 @@
|
||||
"""Discord Bot 어댑터.
|
||||
"""Discord Bot — 워크스페이스 기반 AI Agent.
|
||||
|
||||
사용자 명령을 받아 파이프라인 실행 또는 즉답.
|
||||
지정 채널(봇 이름 채널)에서는 접두사 없이 자연스럽게 대화합니다.
|
||||
대화 기억: 채널별 최근 메시지를 컨텍스트로 주입.
|
||||
슬래시 커맨드로 워크스페이스 관리.
|
||||
등록된 채널에서 자동 대화 + 통합 프롬프트 (1회 호출로 분류+응답).
|
||||
"""
|
||||
|
||||
import asyncio
|
||||
@@ -11,10 +10,12 @@ import logging
|
||||
import re
|
||||
|
||||
import discord
|
||||
from discord import app_commands
|
||||
from discord.ext import commands
|
||||
|
||||
import config
|
||||
from core.gemini_caller import GeminiCallError
|
||||
from core.workspace import WorkspaceManager
|
||||
from core.gemini_caller import GeminiCaller, GeminiCallError
|
||||
|
||||
logger = logging.getLogger("variet.discord")
|
||||
|
||||
@@ -28,31 +29,24 @@ bot = commands.Bot(
|
||||
help_command=commands.DefaultHelpCommand(no_category="명령어"),
|
||||
)
|
||||
|
||||
# In-memory
|
||||
_channel_tasks: dict[int, list[str]] = {}
|
||||
_auto_chat_channel_ids: set[int] = set()
|
||||
# 워크스페이스 매니저 (전역)
|
||||
ws_manager = WorkspaceManager()
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# 대화 기억 (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"
|
||||
async for msg in channel.history(limit=limit + 1):
|
||||
role = "assistant" if msg.author.bot else "user"
|
||||
messages.append(f"[{role}] {msg.content[:300]}")
|
||||
|
||||
# 시간순 (오래된 것 먼저)
|
||||
messages.reverse()
|
||||
|
||||
# 마지막(현재 메시지)은 제외 — 이미 context로 전달되니까
|
||||
if messages:
|
||||
messages = messages[:-1]
|
||||
messages = messages[:-1] # 현재 메시지 제외
|
||||
|
||||
if not messages:
|
||||
return ""
|
||||
@@ -61,29 +55,51 @@ async def _get_channel_history(channel: discord.TextChannel, limit: int = 10) ->
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# 의도 분류 (Intent Router)
|
||||
# 통합 프롬프트 (1회 호출: 분류 + 응답/계획)
|
||||
# ──────────────────────────────────────────────
|
||||
|
||||
async def _classify_intent(message: str) -> dict:
|
||||
"""Gemini로 사용자 메시지의 의도를 분류."""
|
||||
from core.gemini_caller import GeminiCaller
|
||||
async def _unified_call(text: str, history: str, project_path: str) -> dict:
|
||||
"""통합 프롬프트로 1회 호출 — chat/task/clarify 자동 분기."""
|
||||
gemini = GeminiCaller(project_path)
|
||||
|
||||
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": "분류 실패"}
|
||||
# docs 인덱스 주입
|
||||
from core.docs_manager import DocsManager
|
||||
docs = DocsManager(project_path)
|
||||
docs_index = docs.get_docs_index()
|
||||
|
||||
context = (
|
||||
f"{history}"
|
||||
f"## Project Docs\n{docs_index}\n\n"
|
||||
f"## User Message\n{text}"
|
||||
)
|
||||
|
||||
raw = await gemini.call("unified", context, timeout=120)
|
||||
|
||||
# JSON 추출
|
||||
try:
|
||||
json_match = re.search(r'\{[^}]+\}', raw)
|
||||
if json_match:
|
||||
return json.loads(json_match.group())
|
||||
# ```json ... ``` 패턴
|
||||
match = re.search(r'```json\s*\n(.*?)\n\s*```', raw, re.DOTALL)
|
||||
if match:
|
||||
return json.loads(match.group(1))
|
||||
|
||||
# 중첩 { } 찾기
|
||||
brace_depth = 0
|
||||
start = -1
|
||||
for i, ch in enumerate(raw):
|
||||
if ch == '{':
|
||||
if brace_depth == 0:
|
||||
start = i
|
||||
brace_depth += 1
|
||||
elif ch == '}':
|
||||
brace_depth -= 1
|
||||
if brace_depth == 0 and start >= 0:
|
||||
return json.loads(raw[start:i + 1])
|
||||
except (json.JSONDecodeError, AttributeError):
|
||||
pass
|
||||
|
||||
logger.warning(f"Intent JSON 파싱 실패, chat으로 처리: {raw[:100]}")
|
||||
return {"intent": "chat", "reason": "파싱 실패 — 기본 chat"}
|
||||
# 파싱 실패 → chat으로 처리
|
||||
logger.warning(f"통합 프롬프트 JSON 파싱 실패: {raw[:100]}")
|
||||
return {"mode": "chat", "response": raw}
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
@@ -92,159 +108,159 @@ async def _classify_intent(message: str) -> dict:
|
||||
|
||||
@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("_", "-")
|
||||
# 슬래시 커맨드 동기화
|
||||
try:
|
||||
synced = await bot.tree.sync()
|
||||
logger.info(f"슬래시 커맨드 {len(synced)}개 동기화 완료")
|
||||
except Exception as e:
|
||||
logger.error(f"슬래시 커맨드 동기화 실패: {e}")
|
||||
|
||||
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}) 채널 없음 — !명령어만 사용 가능")
|
||||
# 등록된 워크스페이스 채널 표시
|
||||
ws_list = ws_manager.list_all()
|
||||
if ws_list:
|
||||
for ws in ws_list:
|
||||
logger.info(f"워크스페이스 활성: #{ws.channel_id} → {ws.name} ({ws.path})")
|
||||
else:
|
||||
logger.info("등록된 워크스페이스 없음 - /workspace set 으로 등록하세요")
|
||||
|
||||
await bot.change_presence(
|
||||
activity=discord.Activity(
|
||||
type=discord.ActivityType.listening,
|
||||
name="대화" if _auto_chat_channel_ids else f"{config.DISCORD_COMMAND_PREFIX}agent",
|
||||
name="/workspace",
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_message(message: discord.Message):
|
||||
"""모든 메시지 수신 — 봇 전용 채널이면 스마트 라우팅."""
|
||||
"""메시지 수신 — 워크스페이스 채널이면 자동 응답."""
|
||||
if message.author == bot.user or message.author.bot:
|
||||
return
|
||||
|
||||
# ! 명령어는 기존 핸들러로
|
||||
# ! 명령어 처리
|
||||
if message.content.startswith(config.DISCORD_COMMAND_PREFIX):
|
||||
await bot.process_commands(message)
|
||||
return
|
||||
|
||||
# 봇 전용 채널이 아니면 무시
|
||||
if message.channel.id not in _auto_chat_channel_ids:
|
||||
# 워크스페이스 채널인지 확인
|
||||
if not ws_manager.is_workspace_channel(message.channel.id):
|
||||
return
|
||||
|
||||
await _route_message(message)
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# 스마트 라우팅
|
||||
# ──────────────────────────────────────────────
|
||||
|
||||
async def _route_message(message: discord.Message):
|
||||
"""메시지 의도를 분류하고 적절한 핸들러로 라우팅."""
|
||||
ws = ws_manager.get_workspace(message.channel.id)
|
||||
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)
|
||||
|
||||
result = await _unified_call(user_text, history, ws.path)
|
||||
except GeminiCallError as e:
|
||||
logger.error(f"Chat Gemini 오류: {e}")
|
||||
await message.reply(f"⚠️ AI 호출 오류: {str(e)[:200]}")
|
||||
return
|
||||
except Exception as e:
|
||||
logger.error(f"Chat 오류: {e}", exc_info=True)
|
||||
logger.error(f"통합 호출 오류: {e}", exc_info=True)
|
||||
await message.reply(f"❌ 오류: {str(e)[:200]}")
|
||||
return
|
||||
|
||||
mode = result.get("mode", "chat")
|
||||
logger.info(f"통합 분류: {mode} — \"{user_text[:50]}\"")
|
||||
|
||||
if mode == "task":
|
||||
# 설정 완료 체크
|
||||
if not ws.is_ready:
|
||||
await _send_setup_warning(message, ws)
|
||||
return
|
||||
await _handle_task(message, user_text, ws)
|
||||
elif mode == "clarify":
|
||||
question = result.get("question", "더 구체적으로 말씀해 주시겠어요?")
|
||||
embed = discord.Embed(
|
||||
title="🤔 확인이 필요해요",
|
||||
description=question,
|
||||
color=0xF39C12,
|
||||
)
|
||||
await message.reply(embed=embed)
|
||||
else:
|
||||
# chat — 즉답
|
||||
response = result.get("response", "응답을 생성하지 못했습니다.")
|
||||
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)
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# 핸들러: Task (파이프라인 실행)
|
||||
# 설정 경고
|
||||
# ──────────────────────────────────────────────
|
||||
|
||||
async def _handle_task(message: discord.Message, text: str):
|
||||
"""작업 요청 — 새 파이프라인 (병렬 Code + Batch Review + 총평)."""
|
||||
async def _send_setup_warning(message: discord.Message, ws):
|
||||
"""미설정 항목 안내."""
|
||||
missing = ws.missing_configs
|
||||
lines = []
|
||||
for item in missing:
|
||||
if item == "Git":
|
||||
lines.append("❌ **Git** 미설정 → `/workspace git` 으로 설정")
|
||||
elif item == "Vikunja":
|
||||
lines.append("❌ **Vikunja** 미설정 → `/workspace vikunja` 으로 설정")
|
||||
|
||||
embed = discord.Embed(
|
||||
title="⚠️ 워크스페이스 설정 미완료",
|
||||
description=(
|
||||
f"**{ws.name}** 워크스페이스의 설정이 완료되지 않아 작업을 실행할 수 없습니다.\n\n"
|
||||
+ "\n".join(lines)
|
||||
+ "\n\n설정 완료 후 다시 요청해주세요."
|
||||
),
|
||||
color=0xE74C3C,
|
||||
)
|
||||
await message.reply(embed=embed)
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# Task 핸들러 (파이프라인 실행)
|
||||
# ──────────────────────────────────────────────
|
||||
|
||||
async def _handle_task(message: discord.Message, text: str, ws):
|
||||
"""작업 요청 — 파이프라인 실행."""
|
||||
import uuid
|
||||
|
||||
task_id = uuid.uuid4().hex[:8]
|
||||
|
||||
# 접수 메시지
|
||||
embed = discord.Embed(
|
||||
title="📋 작업 접수",
|
||||
description=f"```{text[:200]}```",
|
||||
color=0x3498DB,
|
||||
)
|
||||
embed.set_footer(text=f"ID: {task_id} — 분석 중...")
|
||||
embed.set_footer(text=f"ID: {task_id} — 워크스페이스: {ws.name}")
|
||||
status_msg = await message.channel.send(embed=embed)
|
||||
|
||||
try:
|
||||
from core.task_pipeline import TaskPipeline
|
||||
|
||||
pipeline = TaskPipeline(project_path=str(config.PROJECT_ROOT))
|
||||
pipeline = TaskPipeline(
|
||||
project_path=ws.path,
|
||||
docs_subpath=ws.docs_path,
|
||||
)
|
||||
pipeline.setup()
|
||||
|
||||
# ── Plan ──
|
||||
# Plan
|
||||
embed.color = 0xF39C12
|
||||
embed.set_footer(text=f"🔍 작업 분해 중... (ID: {task_id})")
|
||||
await status_msg.edit(embed=embed)
|
||||
|
||||
plan = await pipeline.plan(text)
|
||||
tasks = plan.get("tasks", [])
|
||||
plan_text = plan.get("summary", str(plan))[:500]
|
||||
|
||||
plan_embed = discord.Embed(
|
||||
title="📝 작업 계획",
|
||||
description=f"```{plan_text}```",
|
||||
description=f"```{plan.get('summary', str(plan))[:500]}```",
|
||||
color=0x2ECC71,
|
||||
)
|
||||
if tasks:
|
||||
@@ -253,221 +269,289 @@ async def _handle_task(message: discord.Message, text: str):
|
||||
for t in tasks[:10]
|
||||
)
|
||||
plan_embed.add_field(
|
||||
name=f"태스크 ({len(tasks)}개)",
|
||||
value=task_list[:1000],
|
||||
inline=False,
|
||||
name=f"태스크 ({len(tasks)}개)", value=task_list[:1000], inline=False,
|
||||
)
|
||||
plan_embed.set_footer(text=f"ID: {task_id}")
|
||||
await message.channel.send(embed=plan_embed)
|
||||
|
||||
if not tasks:
|
||||
await message.channel.send(
|
||||
embed=discord.Embed(
|
||||
title="⚠️ 실행할 태스크가 없습니다",
|
||||
title="⚠️ 실행할 태스크 없음",
|
||||
description="요청을 더 구체적으로 해주세요.",
|
||||
color=0xF39C12,
|
||||
)
|
||||
)
|
||||
return
|
||||
|
||||
# ── Code 병렬 실행 ──
|
||||
# Code 병렬
|
||||
code_embed = discord.Embed(
|
||||
title=f"⚙️ 코딩 중... ({len(tasks)}개 병렬 실행)",
|
||||
title=f"⚙️ 코딩 중... ({len(tasks)}개 병렬)",
|
||||
description="\n".join(
|
||||
f"• {t.get('title', t.get('description', '?'))[:60]}"
|
||||
for t in tasks
|
||||
f"• {t.get('title', '?')[:60]}" for t in tasks
|
||||
),
|
||||
color=0xE67E22,
|
||||
)
|
||||
code_msg = await message.channel.send(embed=code_embed)
|
||||
|
||||
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))
|
||||
applied = apply_changes(changes, ws.path)
|
||||
all_applied.extend(applied)
|
||||
|
||||
if all_applied:
|
||||
files_text = "\n".join(
|
||||
f"• `{f['path']}` ({f['action']})"
|
||||
for f in all_applied[:15]
|
||||
)
|
||||
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,
|
||||
)
|
||||
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,
|
||||
)
|
||||
)
|
||||
|
||||
# Batch Review
|
||||
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 message.channel.send(
|
||||
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 = await pipeline.summarize(text, plan, code_outputs, review, all_applied)
|
||||
|
||||
summary_embed = discord.Embed(
|
||||
title=f"📊 {summary.get('title', '작업 완료')}",
|
||||
description=summary.get("summary", "작업이 완료되었습니다."),
|
||||
description=summary.get("summary", "완료"),
|
||||
color=0x9B59B6,
|
||||
)
|
||||
for field_name, key, emoji in [
|
||||
("변경 사항", "changes", ""),
|
||||
("⚠️ 주의", "warnings", ""),
|
||||
("🔜 다음 단계", "next_steps", ""),
|
||||
]:
|
||||
items = summary.get(key, [])
|
||||
if items:
|
||||
if key == "changes":
|
||||
val = "\n".join(f"• `{c.get('file','?')}` — {c.get('description','')}" for c in items[:10])
|
||||
else:
|
||||
val = "\n".join(f"• {s}" for s in items)
|
||||
summary_embed.add_field(name=field_name, value=val[:1000], inline=False)
|
||||
|
||||
# 변경 파일 요약
|
||||
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}")
|
||||
summary_embed.set_footer(text=f"ID: {task_id} | {ws.name}")
|
||||
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,
|
||||
)
|
||||
embed=discord.Embed(title="❌ AI 호출 오류",
|
||||
description=f"```{str(e)[:500]}```", color=0xE74C3C)
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error(f"작업 실행 오류: {e}", exc_info=True)
|
||||
logger.error(f"작업 오류: {e}", exc_info=True)
|
||||
await message.channel.send(
|
||||
embed=discord.Embed(
|
||||
title="❌ 오류 발생",
|
||||
description=f"```{str(e)[:500]}```",
|
||||
color=0xE74C3C,
|
||||
)
|
||||
embed=discord.Embed(title="❌ 오류",
|
||||
description=f"```{str(e)[:500]}```", color=0xE74C3C)
|
||||
)
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# 핸들러: Clarify (되묻기)
|
||||
# 슬래시 커맨드: /workspace
|
||||
# ──────────────────────────────────────────────
|
||||
|
||||
async def _handle_clarify(message: discord.Message, text: str, reason: str):
|
||||
"""의도 불명확 — 사용자에게 되묻기."""
|
||||
workspace_group = app_commands.Group(name="workspace", description="워크스페이스 관리")
|
||||
|
||||
|
||||
@workspace_group.command(name="set", description="이 채널에 워크스페이스 등록")
|
||||
@app_commands.describe(name="프로젝트 이름", path="로컬 프로젝트 경로")
|
||||
async def workspace_set(interaction: discord.Interaction, name: str, path: str):
|
||||
"""채널에 워크스페이스 등록."""
|
||||
from pathlib import Path as P
|
||||
if not P(path).exists():
|
||||
await interaction.response.send_message(
|
||||
f"❌ 경로가 존재하지 않습니다: `{path}`", ephemeral=True
|
||||
)
|
||||
return
|
||||
|
||||
ws = ws_manager.set_workspace(interaction.channel_id, name, path)
|
||||
|
||||
embed = discord.Embed(
|
||||
title="🤔 확인이 필요해요",
|
||||
title="✅ 워크스페이스 등록 완료",
|
||||
description=(
|
||||
f"말씀하신 내용을 정확히 이해하고 싶어요.\n\n"
|
||||
f"> {text[:200]}\n\n"
|
||||
f"**💬 질문/대화**인가요, **🔧 작업 요청**인가요?\n"
|
||||
f"`질문` 또는 `작업`으로 답해주세요."
|
||||
f"**{name}** → `{path}`\n\n"
|
||||
f"이 채널에서 봇과 대화할 수 있습니다.\n"
|
||||
f"작업 실행을 위해 Git과 Vikunja를 설정해주세요:"
|
||||
),
|
||||
color=0xF39C12,
|
||||
color=0x2ECC71,
|
||||
)
|
||||
embed.set_footer(text=f"사유: {reason}")
|
||||
await message.reply(embed=embed)
|
||||
embed.add_field(name="Git 설정", value="`/workspace git`", inline=True)
|
||||
embed.add_field(name="Vikunja 설정", value="`/workspace vikunja`", inline=True)
|
||||
await interaction.response.send_message(embed=embed)
|
||||
|
||||
|
||||
@workspace_group.command(name="git", description="Git 연결 설정")
|
||||
@app_commands.describe(url="Git 서버 URL", token="API 토큰", repo="Owner/Repo", branch="기본 브랜치")
|
||||
async def workspace_git(interaction: discord.Interaction, url: str, token: str,
|
||||
repo: str = "", branch: str = "main"):
|
||||
"""워크스페이스에 Git 설정."""
|
||||
ws = ws_manager.get_workspace(interaction.channel_id)
|
||||
if not ws:
|
||||
await interaction.response.send_message(
|
||||
"❌ 이 채널에 워크스페이스가 없습니다. `/workspace set`을 먼저 실행하세요.",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
ws_manager.set_git(interaction.channel_id, url, token, repo, branch)
|
||||
|
||||
embed = discord.Embed(
|
||||
title="✅ Git 연결 완료",
|
||||
description=f"**{ws.name}** → {url}\nRepo: `{repo or '미지정'}`\nBranch: `{branch}`",
|
||||
color=0x2ECC71,
|
||||
)
|
||||
await interaction.response.send_message(embed=embed)
|
||||
|
||||
|
||||
@workspace_group.command(name="vikunja", description="Vikunja 프로젝트 관리 연결")
|
||||
@app_commands.describe(url="Vikunja URL", token="API 토큰", project_id="프로젝트 ID")
|
||||
async def workspace_vikunja(interaction: discord.Interaction, url: str, token: str,
|
||||
project_id: int):
|
||||
"""워크스페이스에 Vikunja 설정."""
|
||||
ws = ws_manager.get_workspace(interaction.channel_id)
|
||||
if not ws:
|
||||
await interaction.response.send_message(
|
||||
"❌ 이 채널에 워크스페이스가 없습니다. `/workspace set`을 먼저 실행하세요.",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
ws_manager.set_vikunja(interaction.channel_id, url, token, project_id)
|
||||
|
||||
embed = discord.Embed(
|
||||
title="✅ Vikunja 연결 완료",
|
||||
description=f"**{ws.name}** → {url}\nProject ID: `{project_id}`",
|
||||
color=0x2ECC71,
|
||||
)
|
||||
await interaction.response.send_message(embed=embed)
|
||||
|
||||
|
||||
@workspace_group.command(name="info", description="현재 워크스페이스 정보 표시")
|
||||
async def workspace_info(interaction: discord.Interaction):
|
||||
"""현재 채널 워크스페이스 상태."""
|
||||
ws = ws_manager.get_workspace(interaction.channel_id)
|
||||
if not ws:
|
||||
await interaction.response.send_message(
|
||||
"이 채널에 등록된 워크스페이스가 없습니다.\n`/workspace set`으로 등록하세요.",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
git_status = f"✅ {ws.git.url}" if ws.git.is_configured else "❌ 미설정"
|
||||
vik_status = f"✅ {ws.vikunja.url} (project {ws.vikunja.project_id})" if ws.vikunja.is_configured else "❌ 미설정"
|
||||
|
||||
embed = discord.Embed(
|
||||
title=f"📂 {ws.name}",
|
||||
description=f"경로: `{ws.path}`\nDocs: `{ws.docs_path}`",
|
||||
color=0x3498DB if ws.is_ready else 0xF39C12,
|
||||
)
|
||||
embed.add_field(name="Git", value=git_status, inline=False)
|
||||
embed.add_field(name="Vikunja", value=vik_status, inline=False)
|
||||
embed.add_field(
|
||||
name="상태",
|
||||
value="✅ 모든 설정 완료 — 작업 가능" if ws.is_ready else "⚠️ 설정 미완료 — 작업 차단됨",
|
||||
inline=False,
|
||||
)
|
||||
await interaction.response.send_message(embed=embed)
|
||||
|
||||
|
||||
@workspace_group.command(name="remove", description="워크스페이스 등록 해제")
|
||||
async def workspace_remove(interaction: discord.Interaction):
|
||||
"""워크스페이스 제거."""
|
||||
ws = ws_manager.get_workspace(interaction.channel_id)
|
||||
if not ws:
|
||||
await interaction.response.send_message(
|
||||
"이 채널에 등록된 워크스페이스가 없습니다.", ephemeral=True
|
||||
)
|
||||
return
|
||||
|
||||
name = ws.name
|
||||
ws_manager.remove_workspace(interaction.channel_id)
|
||||
|
||||
await interaction.response.send_message(
|
||||
embed=discord.Embed(
|
||||
title="🗑️ 워크스페이스 해제",
|
||||
description=f"**{name}** 등록이 해제되었습니다.\n이 채널에서 봇이 더 이상 자동 응답하지 않습니다.",
|
||||
color=0x95A5A6,
|
||||
)
|
||||
)
|
||||
|
||||
|
||||
@workspace_group.command(name="list", description="등록된 전체 워크스페이스 목록")
|
||||
async def workspace_list(interaction: discord.Interaction):
|
||||
"""전체 워크스페이스 목록."""
|
||||
all_ws = ws_manager.list_all()
|
||||
if not all_ws:
|
||||
await interaction.response.send_message(
|
||||
"등록된 워크스페이스가 없습니다.\n`/workspace set`으로 등록하세요.",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
embed = discord.Embed(title="📂 워크스페이스 목록", color=0x3498DB)
|
||||
for ws in all_ws:
|
||||
status = "✅" if ws.is_ready else "⚠️"
|
||||
embed.add_field(
|
||||
name=f"{status} {ws.name}",
|
||||
value=f"채널: <#{ws.channel_id}>\n경로: `{ws.path}`",
|
||||
inline=False,
|
||||
)
|
||||
await interaction.response.send_message(embed=embed)
|
||||
|
||||
|
||||
# 슬래시 커맨드 그룹 등록
|
||||
bot.tree.add_command(workspace_group)
|
||||
|
||||
|
||||
# ──────────────────────────────────────────────
|
||||
# 기존 ! 명령어 (유지)
|
||||
# 기존 ! 명령어 (유지, 하위호환)
|
||||
# ──────────────────────────────────────────────
|
||||
|
||||
@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):
|
||||
"""단순 대화."""
|
||||
await _handle_chat(ctx.message, message)
|
||||
|
||||
|
||||
@bot.command(name="ping", help="봇 응답 테스트")
|
||||
async def ping_command(ctx: commands.Context):
|
||||
"""연결 상태 확인."""
|
||||
latency = round(bot.latency * 1000)
|
||||
await ctx.send(f"🏓 Pong! ({latency}ms)")
|
||||
|
||||
|
||||
@bot.command(name="info", help="시스템 정보")
|
||||
async def info_command(ctx: commands.Context):
|
||||
"""시스템 정보 표시."""
|
||||
embed = discord.Embed(
|
||||
title="🤖 Variet Agent",
|
||||
description="AI Agent Team — Gemini CLI 기반 자동화 개발 에이전트",
|
||||
description="AI Agent Team — 워크스페이스 기반 자동화 개발 에이전트",
|
||||
color=0x9B59B6,
|
||||
)
|
||||
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)
|
||||
|
||||
all_ws = ws_manager.list_all()
|
||||
embed.add_field(name="워크스페이스", value=f"{len(all_ws)}개 등록", inline=True)
|
||||
embed.add_field(name="서버", value=str(len(bot.guilds)), inline=True)
|
||||
embed.add_field(
|
||||
name="파이프라인",
|
||||
value="Plan → Code(병렬) → Review(배치) → 총평",
|
||||
value="통합분류(1회) → Code(병렬) → Review(배치) → 총평 → 기록",
|
||||
inline=False,
|
||||
)
|
||||
await ctx.send(embed=embed)
|
||||
|
||||
|
||||
async def start_bot():
|
||||
"""Discord Bot 시작 (async)."""
|
||||
"""Discord Bot 시작."""
|
||||
token = config.DISCORD_BOT_TOKEN
|
||||
if not token:
|
||||
logger.error("DISCORD_BOT_TOKEN이 설정되지 않았습니다. .env 파일을 확인하세요.")
|
||||
logger.error("DISCORD_BOT_TOKEN이 설정되지 않았습니다.")
|
||||
return
|
||||
|
||||
logger.info("Discord Bot 시작 중...")
|
||||
|
||||
Reference in New Issue
Block a user