src/service/models.py

from pydantic import BaseModel , field_validator 


class FileMeta(BaseModel): 
    filename : str 
    user_query : str
    @field_validator("filename") 
    @classmethod 

    def validate_extension(cls , v): 
        allowed_ext = (".csv" , ".db" , ".sql") 
        if not v.lower().endswith(allowed_ext) : 
            raise ValueError("Only .csv, .db, .sql files are allowed") 
        return v