schemas.py

from typing import List, Optional, Tuple

from pydantic import BaseModel, Field


# pydantic schemas
class ElemListSchema(BaseModel):
    elem_list: List[str]


class RelationshipsSchema(BaseModel):
    relationships: List[Tuple[str, ...]]


class Sensor(BaseModel):
    name: str = Field("name of the sensor")
    uuid: Optional[str] = Field("Identifier of the sensor")
    unit: Optional[str] = Field("Unit of measure of the sensor")


class SensorSchema(BaseModel):
    sensors: List[Sensor] = Field("List of the sensors")


class TTLSchema(BaseModel):
    ttl_output: str = Field(
        ..., description="The generated BrickSchema turtle/rdf script."
    )


class TTLToBuildingPromptSchema(BaseModel):
    building_description: str = Field(
        ..., description="The generated building description."
    )
    key_elements: List[str] = Field(
        ..., description="The generated list of key elements."
    )