111 lines
4.5 KiB
Python
111 lines
4.5 KiB
Python
import asyncio
|
|
from typing import Optional
|
|
|
|
from .base import BaseAgent
|
|
from ..db import create_task, update_task_status, get_agent_config, add_log
|
|
from .. import service_proxy
|
|
|
|
class StockAgent(BaseAgent):
|
|
agent_id = "stock"
|
|
display_name = "주식 트레이더"
|
|
|
|
async def on_schedule(self) -> None:
|
|
if self.state not in ("idle", "break"):
|
|
return
|
|
|
|
task_id = create_task(self.agent_id, "news_summary", {"limit": 15})
|
|
await self.transition("working", "AI 뉴스 요약 생성 중...", task_id)
|
|
|
|
try:
|
|
# AI 요약 호출 (뉴스 수집 + LLM 처리는 stock-lab이 담당)
|
|
result = await service_proxy.summarize_stock_news(limit=15)
|
|
|
|
await self.transition("reporting", "뉴스 요약 전송 중...")
|
|
|
|
# 새 통합 텔레그램 API 사용
|
|
from ..telegram import send_agent_message
|
|
tg_result = await send_agent_message(
|
|
agent_id=self.agent_id,
|
|
kind="report",
|
|
title="아침 시장 브리핑",
|
|
body=result["summary"],
|
|
task_id=task_id,
|
|
metadata={
|
|
"tokens": result["tokens"]["total"],
|
|
"duration_ms": result["duration_ms"],
|
|
"model": result["model"],
|
|
},
|
|
)
|
|
|
|
update_task_status(task_id, "succeeded", {
|
|
"summary": result["summary"],
|
|
"article_count": result.get("article_count", 0),
|
|
"tokens": result["tokens"],
|
|
"model": result["model"],
|
|
"duration_ms": result["duration_ms"],
|
|
"telegram_sent": tg_result.get("ok", False),
|
|
"telegram_message_id": tg_result.get("message_id"),
|
|
})
|
|
|
|
if not tg_result.get("ok"):
|
|
desc = tg_result.get("description") or "unknown"
|
|
code = tg_result.get("error_code")
|
|
add_log(self.agent_id, f"Telegram send failed: [{code}] {desc}", "warning", task_id)
|
|
if self._ws_manager:
|
|
await self._ws_manager.send_notification(
|
|
self.agent_id, "telegram_failed", task_id, "텔레그램 전송 실패"
|
|
)
|
|
|
|
await self.transition("idle", "뉴스 요약 완료")
|
|
|
|
except Exception as e:
|
|
add_log(self.agent_id, f"News summary failed: {e}", "error", task_id)
|
|
update_task_status(task_id, "failed", {"error": str(e)})
|
|
await self.transition("idle", f"오류: {e}")
|
|
|
|
async def on_command(self, command: str, params: dict) -> dict:
|
|
if command == "test_telegram":
|
|
from ..telegram import send_agent_message
|
|
result = await send_agent_message(
|
|
agent_id=self.agent_id,
|
|
kind="info",
|
|
title="연결 테스트",
|
|
body="텔레그램 연동이 정상적으로 동작합니다.",
|
|
)
|
|
return {
|
|
"ok": result.get("ok", False),
|
|
"message": "텔레그램 전송 성공" if result.get("ok") else "텔레그램 전송 실패",
|
|
"telegram_message_id": result.get("message_id"),
|
|
}
|
|
|
|
if command == "fetch_news":
|
|
await self.on_schedule()
|
|
return {"ok": True, "message": "뉴스 수집 시작"}
|
|
|
|
if command == "add_alert":
|
|
symbol = params.get("symbol")
|
|
target_price = params.get("target_price")
|
|
if not symbol or target_price is None:
|
|
return {"ok": False, "message": "symbol과 target_price는 필수입니다"}
|
|
config = get_agent_config(self.agent_id)
|
|
alerts = config["custom_config"].get("alerts", [])
|
|
alerts.append({
|
|
"symbol": symbol,
|
|
"name": params.get("name", symbol),
|
|
"target_price": target_price,
|
|
"direction": params.get("direction", "above"),
|
|
})
|
|
from ..db import update_agent_config
|
|
update_agent_config(self.agent_id, custom_config={**config["custom_config"], "alerts": alerts})
|
|
return {"ok": True, "message": f"알람 추가: {params['symbol']}"}
|
|
|
|
if command == "list_alerts":
|
|
config = get_agent_config(self.agent_id)
|
|
alerts = config["custom_config"].get("alerts", [])
|
|
return {"ok": True, "alerts": alerts}
|
|
|
|
return {"ok": False, "message": f"Unknown command: {command}"}
|
|
|
|
async def on_approval(self, task_id: str, approved: bool, feedback: str = "") -> None:
|
|
pass
|