https://twitter.com/jxnlco/status/1753433211147825533

snippet.py

import datetime
from pydantic import BaseModel
import sqlite3

db = sqlite3.connect("mydb")

class Document(BaseModel):
    name: str
    chunk: str
    date: datetime.date

class Search(BaseModel):
    query: str
    max_date: datetime.date
    min_date: datetime.date
    
    def search(self):
        SQL_QUERY = f"""
        SELECT 
            name,
            chunk,
            date 
        FROM documents 
        WHERE 
            chunk MATCH '{self.query}' 
            AND create_date BETWEEN '{self.min_date}' AND '{self.max_date}'
        """
        
        results = db.execute(SQL_QUERY)
        
        return [
            Document(name=result[0], chunk=result[1], date=result[2])
            for result in results.fetchall()
        ]