state.py

from __future__ import annotations
from typing import TypedDict


class PipelineState(TypedDict, total=False):
    # ── Filters (populated from CONFIG by run.py) ──────────────────────────
    run_mode:    str                        # "live" | "backtest"
    exchanges:   list[str]
    languages:   list[str]
    date_range:  tuple[str, str]            # effective range, resolved from run_mode
    subjects:    list[str]
    industries:  list[str] | None
    tags:        list[str] | None
    continents:  list[str] | None
    countries:   list[str] | None
    states:      list[str] | None
    page_size:   int

    # ── Computed during run ────────────────────────────────────────────────
    base_url:        str
    article_stubs:   list[dict]             # {title, published_date, source, link}
    articles:        list[dict]             # {title, published_date, source, link, body}
    export_path:     str

    # ── Signal results (link → record) ────────────────────────────────────
    signal_records:  dict                   # link → {extracted, enriched, signal, evaluation}

    # ── Final output ───────────────────────────────────────────────────────
    merged_articles: list[dict]
    run_summary:     dict


class ArticleSignalState(TypedDict, total=False):
    article:       dict          # full article dict with body
    extracted:     dict | None
    enriched:      dict | None
    signal_record: dict | None   # {link, extracted, enriched, signal, evaluation}
    skipped:       bool
    skip_reason:   str | None    # "no_ticker" | "error:<msg>"