Compare commits
6 Commits
feature/is
...
feature/is
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9bf72c63ec | ||
|
|
1399fa4d09 | ||
| f63fb53289 | |||
|
|
5050a4cf84 | ||
|
|
4987b6393a | ||
| 8faf974522 |
@@ -17,11 +17,16 @@ Updated: 2026-02-26
|
|||||||
|
|
||||||
- Main Agent: 최종 취합/우선순위/승인 게이트 오너
|
- Main Agent: 최종 취합/우선순위/승인 게이트 오너
|
||||||
- PM Agent: 시나리오/요구사항/티켓 관리
|
- PM Agent: 시나리오/요구사항/티켓 관리
|
||||||
- TPM Agent: PM-Dev-검증 간 구현 가능성/달성률 통제
|
- TPM Agent: PM-Dev-검증 간 구현 가능성/달성률 통제, 티켓 등록 및 구현 우선순위 지정 오너
|
||||||
- Dev Agent: 구현 수행, 블로커 발생 시 재계획 요청
|
- Dev Agent: 구현 수행, 블로커 발생 시 재계획 요청
|
||||||
- Verifier Agent: 문서/코드/테스트 산출물 검증
|
- Verifier Agent: 문서/코드/테스트 산출물 검증
|
||||||
- Runtime Verifier Agent: 실제 동작 모니터링, 이상 징후 이슈 발행, 수정 후 이슈 클로즈 판정
|
- Runtime Verifier Agent: 실제 동작 모니터링, 이상 징후 이슈 발행, 수정 후 이슈 클로즈 판정
|
||||||
|
|
||||||
|
Main Agent 아이디에이션 책임:
|
||||||
|
- 진행 중 신규 구현 아이디어를 별도 문서에 누적 기록한다.
|
||||||
|
- 기록 위치: [70_main_agent_ideation.md](./70_main_agent_ideation.md)
|
||||||
|
- 각 항목은 `IDEA-*` 식별자, 배경, 기대효과, 리스크, 후속 티켓 후보를 포함해야 한다.
|
||||||
|
|
||||||
## Main Decision Checkpoints (Mandatory)
|
## Main Decision Checkpoints (Mandatory)
|
||||||
|
|
||||||
- DCP-01 범위 확정: Phase 0 종료 전 Main Agent 승인 필수
|
- DCP-01 범위 확정: Phase 0 종료 전 Main Agent 승인 필수
|
||||||
@@ -140,6 +145,11 @@ Control checks:
|
|||||||
- Main Agent 승인 없는 재계획은 실행 금지
|
- Main Agent 승인 없는 재계획은 실행 금지
|
||||||
- 재계획 반영 시 문서(`REQ/TASK/TEST`) 동시 갱신 필수
|
- 재계획 반영 시 문서(`REQ/TASK/TEST`) 동시 갱신 필수
|
||||||
|
|
||||||
|
TPM 티켓 운영 규칙:
|
||||||
|
- TPM은 합의된 변경을 이슈로 등록하고 우선순위(`P0/P1/P2`)를 지정한다.
|
||||||
|
- PR 본문에는 TPM이 지정한 우선순위와 범위가 그대로 반영되어야 한다.
|
||||||
|
- 우선순위 변경은 TPM 제안 + Main Agent 승인으로만 가능하다.
|
||||||
|
|
||||||
## Runtime Verification Protocol
|
## Runtime Verification Protocol
|
||||||
|
|
||||||
- Runtime Verifier는 테스트 통과 이후 실제 동작(스테이징/실운영)을 모니터링한다.
|
- Runtime Verifier는 테스트 통과 이후 실제 동작(스테이징/실운영)을 모니터링한다.
|
||||||
@@ -150,6 +160,12 @@ Control checks:
|
|||||||
- Dev 수정 완료 + Verifier 재검증 통과 + Runtime Verifier 재관측 정상
|
- Dev 수정 완료 + Verifier 재검증 통과 + Runtime Verifier 재관측 정상
|
||||||
- 최종 클로즈 승인자는 Main Agent
|
- 최종 클로즈 승인자는 Main Agent
|
||||||
|
|
||||||
|
## Server Reflection Rule (No-Merge by Default)
|
||||||
|
|
||||||
|
- 서버 반영 기본 규칙은 `브랜치 푸시 + PR 생성/코멘트`까지로 제한한다.
|
||||||
|
- 기본 흐름에서 검증 승인 후 자동/수동 머지 실행은 금지한다.
|
||||||
|
- 예외는 사용자 명시 승인 시에만 허용되며, Main Agent가 예외 근거를 PR에 기록한다.
|
||||||
|
|
||||||
## Acceptance Matrix (PM Scenario -> Dev Tasks -> Verifier Checks)
|
## Acceptance Matrix (PM Scenario -> Dev Tasks -> Verifier Checks)
|
||||||
|
|
||||||
| PM Scenario | Requirement Coverage | Dev Tasks (Primary) | Verifier Checks (Must Pass) |
|
| PM Scenario | Requirement Coverage | Dev Tasks (Primary) | Verifier Checks (Must Pass) |
|
||||||
|
|||||||
@@ -86,3 +86,9 @@ Updated: 2026-02-26
|
|||||||
- Dev가 `REPLAN-REQUEST` 발행 시 TPM 심사 없이는 스코프/일정 변경 금지
|
- Dev가 `REPLAN-REQUEST` 발행 시 TPM 심사 없이는 스코프/일정 변경 금지
|
||||||
- `REPLAN-REQUEST`는 Main Agent 승인 전 \"제안\" 상태로 유지
|
- `REPLAN-REQUEST`는 Main Agent 승인 전 \"제안\" 상태로 유지
|
||||||
- 승인된 재계획은 `REQ/TASK/TEST` 문서를 동시 갱신해야 유효
|
- 승인된 재계획은 `REQ/TASK/TEST` 문서를 동시 갱신해야 유효
|
||||||
|
|
||||||
|
## 9) 서버 반영 규칙 (No-Merge by Default)
|
||||||
|
|
||||||
|
- 서버 반영은 `브랜치 푸시 + PR 코멘트(리뷰/논의/검증승인)`까지를 기본으로 한다.
|
||||||
|
- 기본 규칙에서 `tea pulls merge` 실행은 금지한다.
|
||||||
|
- 사용자 명시 승인 시에만 예외적으로 머지를 허용한다(예외 근거를 PR 코멘트에 기록).
|
||||||
|
|||||||
48
docs/ouroboros/70_main_agent_ideation.md
Normal file
48
docs/ouroboros/70_main_agent_ideation.md
Normal file
@@ -0,0 +1,48 @@
|
|||||||
|
<!--
|
||||||
|
Doc-ID: DOC-IDEA-001
|
||||||
|
Version: 1.0.0
|
||||||
|
Status: active
|
||||||
|
Owner: main-agent
|
||||||
|
Updated: 2026-02-26
|
||||||
|
-->
|
||||||
|
|
||||||
|
# 메인 에이전트 아이디에이션 백로그
|
||||||
|
|
||||||
|
목적:
|
||||||
|
- 구현 진행 중 떠오른 신규 구현 아이디어를 계획 반영 전 임시 저장한다.
|
||||||
|
- 본 문서는 사용자 검토 후 다음 계획 포함 여부를 결정하기 위한 검토 큐다.
|
||||||
|
|
||||||
|
운영 규칙:
|
||||||
|
- 각 아이디어는 `IDEA-*` 식별자를 사용한다.
|
||||||
|
- 필수 필드: 배경, 기대효과, 리스크, 후속 티켓 후보.
|
||||||
|
- 상태는 `proposed`, `under-review`, `accepted`, `rejected` 중 하나를 사용한다.
|
||||||
|
|
||||||
|
## 아이디어 목록
|
||||||
|
|
||||||
|
- `IDEA-001` (status: proposed)
|
||||||
|
- 제목: Kill-Switch 전역 상태를 프로세스 단일 전역에서 시장/세션 단위 상태로 분리
|
||||||
|
- 배경: 현재는 전역 block 플래그 기반이라 시장별 분리 제어가 제한될 수 있음
|
||||||
|
- 기대효과: KR/US 병행 운용 시 한 시장 장애가 다른 시장 주문을 불필요하게 막는 리스크 축소
|
||||||
|
- 리스크: 상태 동기화 복잡도 증가, 테스트 케이스 확장 필요
|
||||||
|
- 후속 티켓 후보: `TKT-P1-KS-SCOPE-SPLIT`
|
||||||
|
|
||||||
|
- `IDEA-002` (status: proposed)
|
||||||
|
- 제목: Exit Engine 입력 계약(ATR/peak/model_prob/liquidity) 표준 DTO를 데이터 파이프라인에 고정
|
||||||
|
- 배경: 현재 ATR/모델확률 일부가 fallback 기반이라 운영 일관성이 약함
|
||||||
|
- 기대효과: 백테스트-실거래 입력 동형성 강화, 회귀 분석 용이
|
||||||
|
- 리스크: 기존 스캐너/시나리오 엔진 연동 작업량 증가
|
||||||
|
- 후속 티켓 후보: `TKT-P1-EXIT-CONTRACT`
|
||||||
|
|
||||||
|
- `IDEA-003` (status: proposed)
|
||||||
|
- 제목: Runtime Verifier 자동 이슈 생성기(로그 패턴 -> 이슈 템플릿 자동화)
|
||||||
|
- 배경: 런타임 이상 리포트가 수동 작성 중심이라 누락 가능성 존재
|
||||||
|
- 기대효과: 이상 탐지 후 이슈 등록 리드타임 단축, 증적 표준화
|
||||||
|
- 리스크: 오탐 이슈 폭증 가능성, 필터링 룰 필요
|
||||||
|
- 후속 티켓 후보: `TKT-P1-RUNTIME-AUTO-ISSUE`
|
||||||
|
|
||||||
|
- `IDEA-004` (status: proposed)
|
||||||
|
- 제목: PR 코멘트 워크플로우 자동 점검(리뷰어->개발논의->검증승인 누락 차단)
|
||||||
|
- 배경: 현재 절차는 강력하지만 수행 확인이 수동
|
||||||
|
- 기대효과: 절차 누락 방지, 감사 추적 자동화
|
||||||
|
- 리스크: CLI/API 연동 유지보수 비용
|
||||||
|
- 후속 티켓 후보: `TKT-P0-WORKFLOW-GUARD`
|
||||||
@@ -21,6 +21,7 @@ Updated: 2026-02-26
|
|||||||
7. PM 시나리오/이슈 분류: [50_scenario_matrix_and_issue_taxonomy.md](./50_scenario_matrix_and_issue_taxonomy.md)
|
7. PM 시나리오/이슈 분류: [50_scenario_matrix_and_issue_taxonomy.md](./50_scenario_matrix_and_issue_taxonomy.md)
|
||||||
8. TPM 제어 프로토콜/수용 매트릭스: [50_tpm_control_protocol.md](./50_tpm_control_protocol.md)
|
8. TPM 제어 프로토콜/수용 매트릭스: [50_tpm_control_protocol.md](./50_tpm_control_protocol.md)
|
||||||
9. 저장소 강제 설정 체크리스트: [60_repo_enforcement_checklist.md](./60_repo_enforcement_checklist.md)
|
9. 저장소 강제 설정 체크리스트: [60_repo_enforcement_checklist.md](./60_repo_enforcement_checklist.md)
|
||||||
|
10. 메인 에이전트 아이디에이션 백로그: [70_main_agent_ideation.md](./70_main_agent_ideation.md)
|
||||||
|
|
||||||
## 운영 규칙
|
## 운영 규칙
|
||||||
|
|
||||||
|
|||||||
71
src/core/kill_switch.py
Normal file
71
src/core/kill_switch.py
Normal file
@@ -0,0 +1,71 @@
|
|||||||
|
"""Kill switch orchestration for emergency risk actions.
|
||||||
|
|
||||||
|
Order is fixed:
|
||||||
|
1) block new orders
|
||||||
|
2) cancel pending orders
|
||||||
|
3) refresh order state
|
||||||
|
4) reduce risk
|
||||||
|
5) snapshot and notify
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import inspect
|
||||||
|
from dataclasses import dataclass, field
|
||||||
|
from typing import Any, Awaitable, Callable
|
||||||
|
|
||||||
|
StepCallable = Callable[[], Any | Awaitable[Any]]
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass
|
||||||
|
class KillSwitchReport:
|
||||||
|
reason: str
|
||||||
|
steps: list[str] = field(default_factory=list)
|
||||||
|
errors: list[str] = field(default_factory=list)
|
||||||
|
|
||||||
|
|
||||||
|
class KillSwitchOrchestrator:
|
||||||
|
def __init__(self) -> None:
|
||||||
|
self.new_orders_blocked = False
|
||||||
|
|
||||||
|
async def _run_step(
|
||||||
|
self,
|
||||||
|
report: KillSwitchReport,
|
||||||
|
name: str,
|
||||||
|
fn: StepCallable | None,
|
||||||
|
) -> None:
|
||||||
|
report.steps.append(name)
|
||||||
|
if fn is None:
|
||||||
|
return
|
||||||
|
try:
|
||||||
|
result = fn()
|
||||||
|
if inspect.isawaitable(result):
|
||||||
|
await result
|
||||||
|
except Exception as exc: # pragma: no cover - intentionally resilient
|
||||||
|
report.errors.append(f"{name}: {exc}")
|
||||||
|
|
||||||
|
async def trigger(
|
||||||
|
self,
|
||||||
|
*,
|
||||||
|
reason: str,
|
||||||
|
cancel_pending_orders: StepCallable | None = None,
|
||||||
|
refresh_order_state: StepCallable | None = None,
|
||||||
|
reduce_risk: StepCallable | None = None,
|
||||||
|
snapshot_state: StepCallable | None = None,
|
||||||
|
notify: StepCallable | None = None,
|
||||||
|
) -> KillSwitchReport:
|
||||||
|
report = KillSwitchReport(reason=reason)
|
||||||
|
|
||||||
|
self.new_orders_blocked = True
|
||||||
|
report.steps.append("block_new_orders")
|
||||||
|
|
||||||
|
await self._run_step(report, "cancel_pending_orders", cancel_pending_orders)
|
||||||
|
await self._run_step(report, "refresh_order_state", refresh_order_state)
|
||||||
|
await self._run_step(report, "reduce_risk", reduce_risk)
|
||||||
|
await self._run_step(report, "snapshot_state", snapshot_state)
|
||||||
|
await self._run_step(report, "notify", notify)
|
||||||
|
|
||||||
|
return report
|
||||||
|
|
||||||
|
def clear_block(self) -> None:
|
||||||
|
self.new_orders_blocked = False
|
||||||
79
src/main.py
79
src/main.py
@@ -27,6 +27,7 @@ from src.context.layer import ContextLayer
|
|||||||
from src.context.scheduler import ContextScheduler
|
from src.context.scheduler import ContextScheduler
|
||||||
from src.context.store import ContextStore
|
from src.context.store import ContextStore
|
||||||
from src.core.criticality import CriticalityAssessor
|
from src.core.criticality import CriticalityAssessor
|
||||||
|
from src.core.kill_switch import KillSwitchOrchestrator
|
||||||
from src.core.priority_queue import PriorityTaskQueue
|
from src.core.priority_queue import PriorityTaskQueue
|
||||||
from src.core.risk_manager import CircuitBreakerTripped, FatFingerRejected, RiskManager
|
from src.core.risk_manager import CircuitBreakerTripped, FatFingerRejected, RiskManager
|
||||||
from src.db import (
|
from src.db import (
|
||||||
@@ -43,11 +44,14 @@ from src.logging_config import setup_logging
|
|||||||
from src.markets.schedule import MARKETS, MarketInfo, get_next_market_open, get_open_markets
|
from src.markets.schedule import MARKETS, MarketInfo, get_next_market_open, get_open_markets
|
||||||
from src.notifications.telegram_client import NotificationFilter, TelegramClient, TelegramCommandHandler
|
from src.notifications.telegram_client import NotificationFilter, TelegramClient, TelegramCommandHandler
|
||||||
from src.strategy.models import DayPlaybook, MarketOutlook
|
from src.strategy.models import DayPlaybook, MarketOutlook
|
||||||
|
from src.strategy.exit_rules import ExitRuleConfig, ExitRuleInput, evaluate_exit
|
||||||
from src.strategy.playbook_store import PlaybookStore
|
from src.strategy.playbook_store import PlaybookStore
|
||||||
from src.strategy.pre_market_planner import PreMarketPlanner
|
from src.strategy.pre_market_planner import PreMarketPlanner
|
||||||
|
from src.strategy.position_state_machine import PositionState
|
||||||
from src.strategy.scenario_engine import ScenarioEngine
|
from src.strategy.scenario_engine import ScenarioEngine
|
||||||
|
|
||||||
logger = logging.getLogger(__name__)
|
logger = logging.getLogger(__name__)
|
||||||
|
KILL_SWITCH = KillSwitchOrchestrator()
|
||||||
|
|
||||||
|
|
||||||
def safe_float(value: str | float | None, default: float = 0.0) -> float:
|
def safe_float(value: str | float | None, default: float = 0.0) -> float:
|
||||||
@@ -784,7 +788,24 @@ async def trading_cycle(
|
|||||||
stop_loss_threshold = stock_playbook.scenarios[0].stop_loss_pct
|
stop_loss_threshold = stock_playbook.scenarios[0].stop_loss_pct
|
||||||
take_profit_threshold = stock_playbook.scenarios[0].take_profit_pct
|
take_profit_threshold = stock_playbook.scenarios[0].take_profit_pct
|
||||||
|
|
||||||
if loss_pct <= stop_loss_threshold:
|
exit_eval = evaluate_exit(
|
||||||
|
current_state=PositionState.HOLDING,
|
||||||
|
config=ExitRuleConfig(
|
||||||
|
hard_stop_pct=stop_loss_threshold,
|
||||||
|
be_arm_pct=max(0.5, take_profit_threshold * 0.4),
|
||||||
|
arm_pct=take_profit_threshold,
|
||||||
|
),
|
||||||
|
inp=ExitRuleInput(
|
||||||
|
current_price=current_price,
|
||||||
|
entry_price=entry_price,
|
||||||
|
peak_price=max(entry_price, current_price),
|
||||||
|
atr_value=0.0,
|
||||||
|
pred_down_prob=0.0,
|
||||||
|
liquidity_weak=market_data.get("volume_ratio", 1.0) < 1.0,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
if exit_eval.reason == "hard_stop":
|
||||||
decision = TradeDecision(
|
decision = TradeDecision(
|
||||||
action="SELL",
|
action="SELL",
|
||||||
confidence=95,
|
confidence=95,
|
||||||
@@ -800,7 +821,7 @@ async def trading_cycle(
|
|||||||
loss_pct,
|
loss_pct,
|
||||||
stop_loss_threshold,
|
stop_loss_threshold,
|
||||||
)
|
)
|
||||||
elif loss_pct >= take_profit_threshold:
|
elif exit_eval.reason == "arm_take_profit":
|
||||||
decision = TradeDecision(
|
decision = TradeDecision(
|
||||||
action="SELL",
|
action="SELL",
|
||||||
confidence=90,
|
confidence=90,
|
||||||
@@ -876,6 +897,15 @@ async def trading_cycle(
|
|||||||
trade_price = current_price
|
trade_price = current_price
|
||||||
trade_pnl = 0.0
|
trade_pnl = 0.0
|
||||||
if decision.action in ("BUY", "SELL"):
|
if decision.action in ("BUY", "SELL"):
|
||||||
|
if KILL_SWITCH.new_orders_blocked:
|
||||||
|
logger.critical(
|
||||||
|
"KillSwitch block active: skip %s order for %s (%s)",
|
||||||
|
decision.action,
|
||||||
|
stock_code,
|
||||||
|
market.name,
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
broker_held_qty = (
|
broker_held_qty = (
|
||||||
_extract_held_qty_from_balance(
|
_extract_held_qty_from_balance(
|
||||||
balance_data, stock_code, is_domestic=market.is_domestic
|
balance_data, stock_code, is_domestic=market.is_domestic
|
||||||
@@ -944,6 +974,25 @@ async def trading_cycle(
|
|||||||
except Exception as notify_exc:
|
except Exception as notify_exc:
|
||||||
logger.warning("Fat finger notification failed: %s", notify_exc)
|
logger.warning("Fat finger notification failed: %s", notify_exc)
|
||||||
raise # Re-raise to prevent trade
|
raise # Re-raise to prevent trade
|
||||||
|
except CircuitBreakerTripped as exc:
|
||||||
|
ks_report = await KILL_SWITCH.trigger(
|
||||||
|
reason=f"circuit_breaker:{market.code}:{stock_code}:{exc.pnl_pct:.2f}",
|
||||||
|
snapshot_state=lambda: logger.critical(
|
||||||
|
"KillSwitch snapshot %s/%s pnl=%.2f threshold=%.2f",
|
||||||
|
market.code,
|
||||||
|
stock_code,
|
||||||
|
exc.pnl_pct,
|
||||||
|
exc.threshold,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
if ks_report.errors:
|
||||||
|
logger.critical(
|
||||||
|
"KillSwitch step errors for %s/%s: %s",
|
||||||
|
market.code,
|
||||||
|
stock_code,
|
||||||
|
"; ".join(ks_report.errors),
|
||||||
|
)
|
||||||
|
raise
|
||||||
|
|
||||||
# 5. Send order
|
# 5. Send order
|
||||||
order_succeeded = True
|
order_succeeded = True
|
||||||
@@ -1845,6 +1894,15 @@ async def run_daily_session(
|
|||||||
trade_pnl = 0.0
|
trade_pnl = 0.0
|
||||||
order_succeeded = True
|
order_succeeded = True
|
||||||
if decision.action in ("BUY", "SELL"):
|
if decision.action in ("BUY", "SELL"):
|
||||||
|
if KILL_SWITCH.new_orders_blocked:
|
||||||
|
logger.critical(
|
||||||
|
"KillSwitch block active: skip %s order for %s (%s)",
|
||||||
|
decision.action,
|
||||||
|
stock_code,
|
||||||
|
market.name,
|
||||||
|
)
|
||||||
|
continue
|
||||||
|
|
||||||
daily_broker_held_qty = (
|
daily_broker_held_qty = (
|
||||||
_extract_held_qty_from_balance(
|
_extract_held_qty_from_balance(
|
||||||
balance_data, stock_code, is_domestic=market.is_domestic
|
balance_data, stock_code, is_domestic=market.is_domestic
|
||||||
@@ -1911,6 +1969,16 @@ async def run_daily_session(
|
|||||||
logger.warning("Fat finger notification failed: %s", notify_exc)
|
logger.warning("Fat finger notification failed: %s", notify_exc)
|
||||||
continue # Skip this order
|
continue # Skip this order
|
||||||
except CircuitBreakerTripped as exc:
|
except CircuitBreakerTripped as exc:
|
||||||
|
ks_report = await KILL_SWITCH.trigger(
|
||||||
|
reason=f"daily_circuit_breaker:{market.code}:{stock_code}:{exc.pnl_pct:.2f}",
|
||||||
|
snapshot_state=lambda: logger.critical(
|
||||||
|
"Daily KillSwitch snapshot %s/%s pnl=%.2f threshold=%.2f",
|
||||||
|
market.code,
|
||||||
|
stock_code,
|
||||||
|
exc.pnl_pct,
|
||||||
|
exc.threshold,
|
||||||
|
),
|
||||||
|
)
|
||||||
logger.critical("Circuit breaker tripped — stopping session")
|
logger.critical("Circuit breaker tripped — stopping session")
|
||||||
try:
|
try:
|
||||||
await telegram.notify_circuit_breaker(
|
await telegram.notify_circuit_breaker(
|
||||||
@@ -1921,6 +1989,13 @@ async def run_daily_session(
|
|||||||
logger.warning(
|
logger.warning(
|
||||||
"Circuit breaker notification failed: %s", notify_exc
|
"Circuit breaker notification failed: %s", notify_exc
|
||||||
)
|
)
|
||||||
|
if ks_report.errors:
|
||||||
|
logger.critical(
|
||||||
|
"Daily KillSwitch step errors for %s/%s: %s",
|
||||||
|
market.code,
|
||||||
|
stock_code,
|
||||||
|
"; ".join(ks_report.errors),
|
||||||
|
)
|
||||||
raise
|
raise
|
||||||
|
|
||||||
# Send order
|
# Send order
|
||||||
|
|||||||
104
src/strategy/exit_rules.py
Normal file
104
src/strategy/exit_rules.py
Normal file
@@ -0,0 +1,104 @@
|
|||||||
|
"""Composite exit rules: hard stop, break-even lock, ATR trailing, model assist."""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from dataclasses import dataclass
|
||||||
|
|
||||||
|
from src.strategy.position_state_machine import PositionState, StateTransitionInput, promote_state
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class ExitRuleConfig:
|
||||||
|
hard_stop_pct: float = -2.0
|
||||||
|
be_arm_pct: float = 1.2
|
||||||
|
arm_pct: float = 3.0
|
||||||
|
atr_multiplier_k: float = 2.2
|
||||||
|
model_prob_threshold: float = 0.62
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class ExitRuleInput:
|
||||||
|
current_price: float
|
||||||
|
entry_price: float
|
||||||
|
peak_price: float
|
||||||
|
atr_value: float = 0.0
|
||||||
|
pred_down_prob: float = 0.0
|
||||||
|
liquidity_weak: bool = False
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class ExitEvaluation:
|
||||||
|
state: PositionState
|
||||||
|
should_exit: bool
|
||||||
|
reason: str
|
||||||
|
unrealized_pnl_pct: float
|
||||||
|
trailing_stop_price: float | None
|
||||||
|
|
||||||
|
|
||||||
|
def evaluate_exit(
|
||||||
|
*,
|
||||||
|
current_state: PositionState,
|
||||||
|
config: ExitRuleConfig,
|
||||||
|
inp: ExitRuleInput,
|
||||||
|
) -> ExitEvaluation:
|
||||||
|
"""Evaluate composite exit logic and return updated state."""
|
||||||
|
if inp.entry_price <= 0 or inp.current_price <= 0:
|
||||||
|
return ExitEvaluation(
|
||||||
|
state=current_state,
|
||||||
|
should_exit=False,
|
||||||
|
reason="invalid_price",
|
||||||
|
unrealized_pnl_pct=0.0,
|
||||||
|
trailing_stop_price=None,
|
||||||
|
)
|
||||||
|
|
||||||
|
unrealized = (inp.current_price - inp.entry_price) / inp.entry_price * 100.0
|
||||||
|
hard_stop_hit = unrealized <= config.hard_stop_pct
|
||||||
|
take_profit_hit = unrealized >= config.arm_pct
|
||||||
|
|
||||||
|
trailing_stop_price: float | None = None
|
||||||
|
trailing_stop_hit = False
|
||||||
|
if inp.atr_value > 0 and inp.peak_price > 0:
|
||||||
|
trailing_stop_price = inp.peak_price - (config.atr_multiplier_k * inp.atr_value)
|
||||||
|
trailing_stop_hit = inp.current_price <= trailing_stop_price
|
||||||
|
|
||||||
|
be_lock_threat = current_state in (PositionState.BE_LOCK, PositionState.ARMED) and (
|
||||||
|
inp.current_price <= inp.entry_price
|
||||||
|
)
|
||||||
|
model_exit_signal = inp.pred_down_prob >= config.model_prob_threshold and inp.liquidity_weak
|
||||||
|
|
||||||
|
next_state = promote_state(
|
||||||
|
current=current_state,
|
||||||
|
inp=StateTransitionInput(
|
||||||
|
unrealized_pnl_pct=unrealized,
|
||||||
|
be_arm_pct=config.be_arm_pct,
|
||||||
|
arm_pct=config.arm_pct,
|
||||||
|
hard_stop_hit=hard_stop_hit,
|
||||||
|
trailing_stop_hit=trailing_stop_hit,
|
||||||
|
model_exit_signal=model_exit_signal,
|
||||||
|
be_lock_threat=be_lock_threat,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
|
||||||
|
if hard_stop_hit:
|
||||||
|
reason = "hard_stop"
|
||||||
|
elif trailing_stop_hit:
|
||||||
|
reason = "atr_trailing_stop"
|
||||||
|
elif be_lock_threat:
|
||||||
|
reason = "be_lock_threat"
|
||||||
|
elif model_exit_signal:
|
||||||
|
reason = "model_liquidity_exit"
|
||||||
|
elif take_profit_hit:
|
||||||
|
# Backward-compatible immediate profit-taking path.
|
||||||
|
reason = "arm_take_profit"
|
||||||
|
else:
|
||||||
|
reason = "hold"
|
||||||
|
|
||||||
|
should_exit = next_state == PositionState.EXITED or take_profit_hit
|
||||||
|
|
||||||
|
return ExitEvaluation(
|
||||||
|
state=next_state,
|
||||||
|
should_exit=should_exit,
|
||||||
|
reason=reason,
|
||||||
|
unrealized_pnl_pct=unrealized,
|
||||||
|
trailing_stop_price=trailing_stop_price,
|
||||||
|
)
|
||||||
70
src/strategy/position_state_machine.py
Normal file
70
src/strategy/position_state_machine.py
Normal file
@@ -0,0 +1,70 @@
|
|||||||
|
"""Position state machine for staged exit control.
|
||||||
|
|
||||||
|
State progression is monotonic (promotion-only) except terminal EXITED.
|
||||||
|
"""
|
||||||
|
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
from dataclasses import dataclass
|
||||||
|
from enum import Enum
|
||||||
|
|
||||||
|
|
||||||
|
class PositionState(str, Enum):
|
||||||
|
HOLDING = "HOLDING"
|
||||||
|
BE_LOCK = "BE_LOCK"
|
||||||
|
ARMED = "ARMED"
|
||||||
|
EXITED = "EXITED"
|
||||||
|
|
||||||
|
|
||||||
|
_STATE_RANK: dict[PositionState, int] = {
|
||||||
|
PositionState.HOLDING: 0,
|
||||||
|
PositionState.BE_LOCK: 1,
|
||||||
|
PositionState.ARMED: 2,
|
||||||
|
PositionState.EXITED: 3,
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@dataclass(frozen=True)
|
||||||
|
class StateTransitionInput:
|
||||||
|
unrealized_pnl_pct: float
|
||||||
|
be_arm_pct: float
|
||||||
|
arm_pct: float
|
||||||
|
hard_stop_hit: bool = False
|
||||||
|
trailing_stop_hit: bool = False
|
||||||
|
model_exit_signal: bool = False
|
||||||
|
be_lock_threat: bool = False
|
||||||
|
|
||||||
|
|
||||||
|
def evaluate_exit_first(inp: StateTransitionInput) -> bool:
|
||||||
|
"""Return True when terminal exit conditions are met.
|
||||||
|
|
||||||
|
EXITED must be evaluated before any promotion.
|
||||||
|
"""
|
||||||
|
return (
|
||||||
|
inp.hard_stop_hit
|
||||||
|
or inp.trailing_stop_hit
|
||||||
|
or inp.model_exit_signal
|
||||||
|
or inp.be_lock_threat
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def promote_state(current: PositionState, inp: StateTransitionInput) -> PositionState:
|
||||||
|
"""Promote to highest admissible state for current tick/bar.
|
||||||
|
|
||||||
|
Rules:
|
||||||
|
- EXITED has highest precedence and is terminal.
|
||||||
|
- Promotions are monotonic (no downgrade).
|
||||||
|
"""
|
||||||
|
if current == PositionState.EXITED:
|
||||||
|
return PositionState.EXITED
|
||||||
|
|
||||||
|
if evaluate_exit_first(inp):
|
||||||
|
return PositionState.EXITED
|
||||||
|
|
||||||
|
target = PositionState.HOLDING
|
||||||
|
if inp.unrealized_pnl_pct >= inp.arm_pct:
|
||||||
|
target = PositionState.ARMED
|
||||||
|
elif inp.unrealized_pnl_pct >= inp.be_arm_pct:
|
||||||
|
target = PositionState.BE_LOCK
|
||||||
|
|
||||||
|
return target if _STATE_RANK[target] > _STATE_RANK[current] else current
|
||||||
55
tests/test_kill_switch.py
Normal file
55
tests/test_kill_switch.py
Normal file
@@ -0,0 +1,55 @@
|
|||||||
|
import pytest
|
||||||
|
|
||||||
|
from src.core.kill_switch import KillSwitchOrchestrator
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_kill_switch_executes_steps_in_order() -> None:
|
||||||
|
ks = KillSwitchOrchestrator()
|
||||||
|
calls: list[str] = []
|
||||||
|
|
||||||
|
async def _cancel() -> None:
|
||||||
|
calls.append("cancel")
|
||||||
|
|
||||||
|
def _refresh() -> None:
|
||||||
|
calls.append("refresh")
|
||||||
|
|
||||||
|
def _reduce() -> None:
|
||||||
|
calls.append("reduce")
|
||||||
|
|
||||||
|
def _snapshot() -> None:
|
||||||
|
calls.append("snapshot")
|
||||||
|
|
||||||
|
def _notify() -> None:
|
||||||
|
calls.append("notify")
|
||||||
|
|
||||||
|
report = await ks.trigger(
|
||||||
|
reason="test",
|
||||||
|
cancel_pending_orders=_cancel,
|
||||||
|
refresh_order_state=_refresh,
|
||||||
|
reduce_risk=_reduce,
|
||||||
|
snapshot_state=_snapshot,
|
||||||
|
notify=_notify,
|
||||||
|
)
|
||||||
|
|
||||||
|
assert report.steps == [
|
||||||
|
"block_new_orders",
|
||||||
|
"cancel_pending_orders",
|
||||||
|
"refresh_order_state",
|
||||||
|
"reduce_risk",
|
||||||
|
"snapshot_state",
|
||||||
|
"notify",
|
||||||
|
]
|
||||||
|
assert calls == ["cancel", "refresh", "reduce", "snapshot", "notify"]
|
||||||
|
assert report.errors == []
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_kill_switch_collects_step_errors() -> None:
|
||||||
|
ks = KillSwitchOrchestrator()
|
||||||
|
|
||||||
|
def _boom() -> None:
|
||||||
|
raise RuntimeError("boom")
|
||||||
|
|
||||||
|
report = await ks.trigger(reason="test", cancel_pending_orders=_boom)
|
||||||
|
assert any(err.startswith("cancel_pending_orders:") for err in report.errors)
|
||||||
@@ -13,6 +13,7 @@ from src.db import init_db, log_trade
|
|||||||
from src.evolution.scorecard import DailyScorecard
|
from src.evolution.scorecard import DailyScorecard
|
||||||
from src.logging.decision_logger import DecisionLogger
|
from src.logging.decision_logger import DecisionLogger
|
||||||
from src.main import (
|
from src.main import (
|
||||||
|
KILL_SWITCH,
|
||||||
_apply_dashboard_flag,
|
_apply_dashboard_flag,
|
||||||
_determine_order_quantity,
|
_determine_order_quantity,
|
||||||
_extract_avg_price_from_balance,
|
_extract_avg_price_from_balance,
|
||||||
@@ -77,6 +78,14 @@ def _make_sell_match(stock_code: str = "005930") -> ScenarioMatch:
|
|||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.fixture(autouse=True)
|
||||||
|
def _reset_kill_switch_state() -> None:
|
||||||
|
"""Prevent cross-test leakage from global kill-switch state."""
|
||||||
|
KILL_SWITCH.clear_block()
|
||||||
|
yield
|
||||||
|
KILL_SWITCH.clear_block()
|
||||||
|
|
||||||
|
|
||||||
class TestExtractAvgPriceFromBalance:
|
class TestExtractAvgPriceFromBalance:
|
||||||
"""Tests for _extract_avg_price_from_balance() (issue #249)."""
|
"""Tests for _extract_avg_price_from_balance() (issue #249)."""
|
||||||
|
|
||||||
@@ -5039,3 +5048,71 @@ class TestOverseasGhostPositionClose:
|
|||||||
and "[ghost-close]" in (c.kwargs.get("rationale") or "")
|
and "[ghost-close]" in (c.kwargs.get("rationale") or "")
|
||||||
]
|
]
|
||||||
assert not ghost_close_calls, "Ghost-close must NOT be triggered for non-잔고없음 errors"
|
assert not ghost_close_calls, "Ghost-close must NOT be triggered for non-잔고없음 errors"
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.asyncio
|
||||||
|
async def test_kill_switch_block_skips_actionable_order_execution() -> None:
|
||||||
|
"""Active kill-switch must prevent actionable order execution."""
|
||||||
|
db_conn = init_db(":memory:")
|
||||||
|
decision_logger = DecisionLogger(db_conn)
|
||||||
|
|
||||||
|
broker = MagicMock()
|
||||||
|
broker.get_current_price = AsyncMock(return_value=(100.0, 0.5, 0.0))
|
||||||
|
broker.get_balance = AsyncMock(
|
||||||
|
return_value={
|
||||||
|
"output1": [],
|
||||||
|
"output2": [
|
||||||
|
{
|
||||||
|
"tot_evlu_amt": "100000",
|
||||||
|
"dnca_tot_amt": "50000",
|
||||||
|
"pchs_amt_smtl_amt": "50000",
|
||||||
|
}
|
||||||
|
],
|
||||||
|
}
|
||||||
|
)
|
||||||
|
broker.send_order = AsyncMock(return_value={"msg1": "OK"})
|
||||||
|
|
||||||
|
market = MagicMock()
|
||||||
|
market.name = "Korea"
|
||||||
|
market.code = "KR"
|
||||||
|
market.exchange_code = "KRX"
|
||||||
|
market.is_domestic = True
|
||||||
|
|
||||||
|
telegram = MagicMock()
|
||||||
|
telegram.notify_trade_execution = AsyncMock()
|
||||||
|
telegram.notify_fat_finger = AsyncMock()
|
||||||
|
telegram.notify_circuit_breaker = AsyncMock()
|
||||||
|
telegram.notify_scenario_matched = AsyncMock()
|
||||||
|
|
||||||
|
settings = MagicMock()
|
||||||
|
settings.POSITION_SIZING_ENABLED = False
|
||||||
|
settings.CONFIDENCE_THRESHOLD = 80
|
||||||
|
|
||||||
|
try:
|
||||||
|
KILL_SWITCH.new_orders_blocked = True
|
||||||
|
await trading_cycle(
|
||||||
|
broker=broker,
|
||||||
|
overseas_broker=MagicMock(),
|
||||||
|
scenario_engine=MagicMock(evaluate=MagicMock(return_value=_make_buy_match())),
|
||||||
|
playbook=_make_playbook(),
|
||||||
|
risk=MagicMock(),
|
||||||
|
db_conn=db_conn,
|
||||||
|
decision_logger=decision_logger,
|
||||||
|
context_store=MagicMock(
|
||||||
|
get_latest_timeframe=MagicMock(return_value=None),
|
||||||
|
set_context=MagicMock(),
|
||||||
|
),
|
||||||
|
criticality_assessor=MagicMock(
|
||||||
|
assess_market_conditions=MagicMock(return_value=MagicMock(value="NORMAL")),
|
||||||
|
get_timeout=MagicMock(return_value=5.0),
|
||||||
|
),
|
||||||
|
telegram=telegram,
|
||||||
|
market=market,
|
||||||
|
stock_code="005930",
|
||||||
|
scan_candidates={},
|
||||||
|
settings=settings,
|
||||||
|
)
|
||||||
|
finally:
|
||||||
|
KILL_SWITCH.clear_block()
|
||||||
|
|
||||||
|
broker.send_order.assert_not_called()
|
||||||
|
|||||||
38
tests/test_strategy_exit_rules.py
Normal file
38
tests/test_strategy_exit_rules.py
Normal file
@@ -0,0 +1,38 @@
|
|||||||
|
from src.strategy.exit_rules import ExitRuleConfig, ExitRuleInput, evaluate_exit
|
||||||
|
from src.strategy.position_state_machine import PositionState
|
||||||
|
|
||||||
|
|
||||||
|
def test_hard_stop_exit() -> None:
|
||||||
|
out = evaluate_exit(
|
||||||
|
current_state=PositionState.HOLDING,
|
||||||
|
config=ExitRuleConfig(hard_stop_pct=-2.0, arm_pct=3.0),
|
||||||
|
inp=ExitRuleInput(current_price=97.0, entry_price=100.0, peak_price=100.0),
|
||||||
|
)
|
||||||
|
assert out.should_exit is True
|
||||||
|
assert out.reason == "hard_stop"
|
||||||
|
|
||||||
|
|
||||||
|
def test_take_profit_exit_for_backward_compatibility() -> None:
|
||||||
|
out = evaluate_exit(
|
||||||
|
current_state=PositionState.HOLDING,
|
||||||
|
config=ExitRuleConfig(hard_stop_pct=-2.0, arm_pct=3.0),
|
||||||
|
inp=ExitRuleInput(current_price=104.0, entry_price=100.0, peak_price=104.0),
|
||||||
|
)
|
||||||
|
assert out.should_exit is True
|
||||||
|
assert out.reason == "arm_take_profit"
|
||||||
|
|
||||||
|
|
||||||
|
def test_model_assist_exit_signal() -> None:
|
||||||
|
out = evaluate_exit(
|
||||||
|
current_state=PositionState.ARMED,
|
||||||
|
config=ExitRuleConfig(model_prob_threshold=0.62, arm_pct=10.0),
|
||||||
|
inp=ExitRuleInput(
|
||||||
|
current_price=101.0,
|
||||||
|
entry_price=100.0,
|
||||||
|
peak_price=105.0,
|
||||||
|
pred_down_prob=0.8,
|
||||||
|
liquidity_weak=True,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
assert out.should_exit is True
|
||||||
|
assert out.reason == "model_liquidity_exit"
|
||||||
30
tests/test_strategy_state_machine.py
Normal file
30
tests/test_strategy_state_machine.py
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
from src.strategy.position_state_machine import (
|
||||||
|
PositionState,
|
||||||
|
StateTransitionInput,
|
||||||
|
promote_state,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def test_gap_jump_promotes_to_armed_directly() -> None:
|
||||||
|
state = promote_state(
|
||||||
|
PositionState.HOLDING,
|
||||||
|
StateTransitionInput(
|
||||||
|
unrealized_pnl_pct=4.0,
|
||||||
|
be_arm_pct=1.2,
|
||||||
|
arm_pct=2.8,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
assert state == PositionState.ARMED
|
||||||
|
|
||||||
|
|
||||||
|
def test_exited_has_priority_over_promotion() -> None:
|
||||||
|
state = promote_state(
|
||||||
|
PositionState.HOLDING,
|
||||||
|
StateTransitionInput(
|
||||||
|
unrealized_pnl_pct=5.0,
|
||||||
|
be_arm_pct=1.2,
|
||||||
|
arm_pct=2.8,
|
||||||
|
hard_stop_hit=True,
|
||||||
|
),
|
||||||
|
)
|
||||||
|
assert state == PositionState.EXITED
|
||||||
Reference in New Issue
Block a user