3e2405dff9
- upstream sinmb79/blog-writer v3.2.1 코드 베이스 적용 - config_resolver, CLI, writer_bot, shorts pipeline 등 신규 기능 포함 - load_dotenv Windows 경로 → Docker 호환 load_dotenv() 변경 (25개 파일) - runtime_guard.py Docker 환경 bypass 추가 - config/blogs.json: eli-ai 블로그 정체성 (8개 카테고리) - config/sources.json: 38개 RSS 소스 유지 - config/engine.json: writing provider → gemini (2.5-flash) - config/safety_keywords.json: 모든 글 수동 승인 (score 101) - bots/scheduler.py: 시스템 프롬프트 eli 블로그 기준으로 업데이트 - bots/publisher_bot.py: .env refresh token OAuth 폴백 로직 추가 - requirements.txt: google-generativeai, groq 활성화 - Dockerfile + docker-compose.yml: NAS Docker 배포 설정 - CLAUDE.md: 프로젝트 메타데이터 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
"""
|
|
Remote Claude Bot
|
|
텔레그램 메시지를 Claude Agent SDK에 전달하고 결과를 돌려보냅니다.
|
|
실행: python bots/remote_claude.py
|
|
"""
|
|
import asyncio
|
|
import logging
|
|
import os
|
|
from pathlib import Path
|
|
|
|
from dotenv import load_dotenv
|
|
from telegram import Update
|
|
from telegram.ext import Application, MessageHandler, CommandHandler, filters, ContextTypes
|
|
from claude_agent_sdk import query, ClaudeAgentOptions, ResultMessage
|
|
|
|
load_dotenv()
|
|
|
|
BASE_DIR = Path(__file__).parent.parent
|
|
TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN', '')
|
|
TELEGRAM_CHAT_ID = int(os.getenv('TELEGRAM_CHAT_ID', '0'))
|
|
REMOTE_CLAUDE_POLLING_ENABLED = os.getenv('REMOTE_CLAUDE_POLLING_ENABLED', '').lower() in {'1', 'true', 'yes', 'on'}
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s [%(levelname)s] %(message)s',
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
MAX_MSG_LEN = 4000
|
|
|
|
|
|
def split_message(text: str) -> list[str]:
|
|
chunks = []
|
|
while len(text) > MAX_MSG_LEN:
|
|
chunks.append(text[:MAX_MSG_LEN])
|
|
text = text[MAX_MSG_LEN:]
|
|
if text:
|
|
chunks.append(text)
|
|
return chunks
|
|
|
|
|
|
async def run_claude(prompt: str) -> str:
|
|
result_text = ""
|
|
try:
|
|
async for message in query(
|
|
prompt=prompt,
|
|
options=ClaudeAgentOptions(
|
|
cwd=str(BASE_DIR),
|
|
allowed_tools=["Read", "Write", "Edit", "Bash", "Glob", "Grep"],
|
|
permission_mode="bypassPermissions",
|
|
max_turns=30,
|
|
)
|
|
):
|
|
if isinstance(message, ResultMessage):
|
|
result_text = message.result
|
|
except Exception as e:
|
|
result_text = f"오류: {e}"
|
|
return result_text or "(완료)"
|
|
|
|
|
|
async def handle_message(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if update.message.chat_id != TELEGRAM_CHAT_ID:
|
|
return
|
|
|
|
prompt = update.message.text.strip()
|
|
logger.info(f"명령 수신: {prompt[:80]}")
|
|
|
|
await update.message.reply_text("처리 중...")
|
|
await context.bot.send_chat_action(chat_id=TELEGRAM_CHAT_ID, action="typing")
|
|
|
|
result = await run_claude(prompt)
|
|
|
|
for chunk in split_message(result):
|
|
await update.message.reply_text(chunk)
|
|
|
|
|
|
async def cmd_help(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
if update.message.chat_id != TELEGRAM_CHAT_ID:
|
|
return
|
|
await update.message.reply_text(
|
|
"Remote Claude Bot\n\n"
|
|
"자연어로 아무 지시나 입력하세요.\n\n"
|
|
"예시:\n"
|
|
"• scheduler.py 상태 확인해줘\n"
|
|
"• 수집봇 지금 실행해줘\n"
|
|
"• .env 파일 내용 보여줘\n"
|
|
"• requirements.txt에 패키지 추가해줘\n"
|
|
"• 오늘 로그 확인해줘\n\n"
|
|
"/help — 이 메시지"
|
|
)
|
|
|
|
|
|
def main():
|
|
if not REMOTE_CLAUDE_POLLING_ENABLED:
|
|
logger.info("Remote Claude Bot polling 비활성화 — 기본 운영은 scheduler.py Telegram 리스너 사용")
|
|
return
|
|
|
|
if not TELEGRAM_BOT_TOKEN:
|
|
logger.error("TELEGRAM_BOT_TOKEN이 없습니다.")
|
|
return
|
|
|
|
app = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
|
|
app.add_handler(CommandHandler('help', cmd_help))
|
|
app.add_handler(MessageHandler(
|
|
filters.TEXT & ~filters.COMMAND & filters.Chat(TELEGRAM_CHAT_ID),
|
|
handle_message
|
|
))
|
|
|
|
logger.info(f"Remote Claude Bot 시작 (chat_id={TELEGRAM_CHAT_ID})")
|
|
app.run_polling(drop_pending_updates=True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|