utils/schemas.py

from langchain.pydantic_v1 import BaseModel, Field

class ChatBotInput(BaseModel):
    prompt: str = Field(description="should be a user query")

class SlotScheduleInput(BaseModel):
    address: str = Field(description="Address to pickup the tyres")
    tyre_counts: str = Field(description="Number of the tyres to pickup")

class SlotDeleteInput(BaseModel):
    slot_id: str = Field(description="Slot id to delete a slot. Slot id must be in formate 'SLOT-int', where int is any positive integer greater or equal to 0")
    

class SlotUpdateInput(BaseModel):
    slot_id: str = Field(description="Slot id to delete a slot. Slot id must be in formate 'SLOT-int', where int is any positive integer greater or equal to 0")
    address: str = Field(description="Address to pickup the tyres")
    tyre_counts: str = Field(description="Number of the tyres to pickup")

class InvoiceGenInput(BaseModel):
    invoice_num: str = Field(description="Invoice Number to generate the invoice. invoice number must be number between 10000 to 99999.")
    

class PriceEstimationInput(BaseModel):
    category: str = Field(description="Category of the tyre. It must be from ['bike', 'car', 'truck']")