"Recommended company openai structure"

from pydantic import BaseModel, Field


class Company(BaseModel):
    "Hypotehtical company"
    company_name: str = Field(description="Name of the fictional company")
    product_descriptions: list = Field(
        description="List of product descriptions at minimum 5"
    )
    SIC_code: str = Field(
        description="Assumed SIC code for the company. Not NAIC or other codes"
    )
    industry: str = Field(description="Industry based on the SIC_code.")
    sector: str = Field(description="Sector based on the SIC_code.")


class Companies(BaseModel):
    "Companies"
    companies: list[Company] = []