"Recommended company groups model."

from pydantic import BaseModel, Field


class CompanyGroup(BaseModel):
    group_name: str = Field(description="Name of the group")
    companies: list[str] = Field(
        description="List of company urls only. Don't add anything else", default=[]
    )
    rationale: str = Field(description="Rationale for the group", default="")


class CompanyGroups(BaseModel):
    groups: list[CompanyGroup] = []