schemas.py

"""schemas.py — Shared data contracts between all agents."""
from dataclasses import dataclass, field
from typing import Optional


@dataclass
class SearchResult:
    url: str
    title: str
    content: str


@dataclass
class Finding:
    claim: str
    source_url: str
    source_title: str
    source_credibility: float
    claim_credibility: float
    topic_relevance: float
    quote: Optional[str] = None


@dataclass
class KeyClaim:
    claim: str
    supporting_sources: list[str]
    contested: bool = False
    contest_note: Optional[str] = None


@dataclass
class SynthesizedSection:
    title: str
    summary: str
    key_claims: list[KeyClaim]
    confidence: float

    def __post_init__(self):
        # Allow dicts from JSON parsing to be auto-converted
        self.key_claims = [
            KeyClaim(**c) if isinstance(c, dict) else c
            for c in self.key_claims
        ]


@dataclass
class Synthesis:
    topic: str
    sections: list[SynthesizedSection]
    gaps: list[str]
    overall_confidence: float


@dataclass
class Report:
    topic: str
    markdown: str
    sources: list[str]
    overall_confidence: float
    gaps: list[str]