App/Embedding/Schemas.py

from typing import Optional
from pydantic import BaseModel


class BaseRequest(BaseModel):
    userId: int
    taskId: str
    fileName: str
    youtubeLink: Optional[str]
    telegramId: Optional[str]

class SearchRequest(BaseModel):
    # userId: int
    taskId: str
    query: str
    # youtubeLink: Optional[str]
    # telegramId: Optional[str]



class GetTranscriptions(BaseModel):
    userId: int