"""tau_rag.intelligence — Legal argument extraction + strategy synthesis.

This package provides the upper layer of legal AI: given user-provided facts +
the existing tau-rag retrieval system, it produces structured legal arguments
(IRAC), case factor analysis, and an actionable legal/mediation strategy.

It uses ONLY local models (no external LLM API):
  • Retrieval comes from the existing tau-rag MultiRetriever.
  • Argument structuring uses templated IRAC + scoring (argument_generator).
  • Factor extraction uses LegalFactor weights (case_analyzer).
  • Optional polish via the local TAU LLM checkpoint.

Public API:
    from tau_rag.intelligence import StrategySynthesizer
    syn = StrategySynthesizer()
    strategy = syn.synthesize(user_facts="...", side="plaintiff", domain="contracts")
"""
from .argument_generator import (
    ArgumentGenerator,
    LegalArgument,
    ArgumentType,
    ArgumentStrength,
    ArgumentSet,
    IRACElement,
    RhetoricalStrategy,
    create_argument_generator,
)
from .case_analyzer import (
    CaseAnalyzer,
    LegalFactor,
    FactorType,
    CaseType,
    CaseOutcome,
    CausalChain,
)
from .variation_generator import VariationGenerator
from .strategy_synthesizer import StrategySynthesizer, StrategyResult
from .auto_labeler import (
    AutoLabeler, LabelSignals,
    extract_statute_citations, extract_case_citations,
    STATUTE_PATTERN, CASE_PATTERN,
)
from .case_based_arguments import (
    CaseBasedArgumentExtractor,
    ArgumentTemplate,
    DraftedArgument,
)
from .outcome_signals import (
    OutcomeSignals,
    compute_outcome_signals,
)
from .retrieval_signals import (
    RetrievalSignals,
    compute_retrieval_signals,
)

__all__ = [
    "ArgumentGenerator", "LegalArgument", "ArgumentType", "ArgumentStrength",
    "ArgumentSet", "IRACElement", "RhetoricalStrategy",
    "create_argument_generator",
    "CaseAnalyzer", "LegalFactor", "FactorType", "CaseType", "CaseOutcome",
    "CausalChain",
    "VariationGenerator",
    "StrategySynthesizer", "StrategyResult",
    "AutoLabeler", "LabelSignals",
    "extract_statute_citations", "extract_case_citations",
    "CaseBasedArgumentExtractor", "ArgumentTemplate", "DraftedArgument",
    "OutcomeSignals", "compute_outcome_signals",
    "RetrievalSignals", "compute_retrieval_signals",
]