models/Query.py

from pydantic import BaseModel
from typing import List



class Query(BaseModel):
    
    id: str
    prompt: str

class Query_Multiple(BaseModel):
    prompt: List[Query]


class SimilarPrompt(BaseModel):
    prompt: str
    distance: float  

class SearchResponse(BaseModel):
    results: List[SimilarPrompt]