from dataclasses import dataclass, field
from datetime import datetime


@dataclass
class Patent:
    patent_id: str
    pdf_link: str
    pdf_path: str | None = None
    summary: dict | None = None


@dataclass
class Patents:
    patents: list[Patent]


@dataclass
class CompanyAnalysisResult:
    """Result of analyzing a single company's patent portfolio."""

    company_name: str
    analysis: str
    patent_count: int
    success: bool
    error: str | None = None
    model: str | None = None
    timestamp: datetime = field(default_factory=datetime.now)


@dataclass
class BatchAnalysisResult:
    """Result of batch analyzing multiple companies."""

    results: list[CompanyAnalysisResult]
    total_companies: int
    successful: int
    failed: int
    timestamp: datetime = field(default_factory=datetime.now)