File size: 594 Bytes
e84d389 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | from pydantic import BaseModel, Field
from typing import List, Optional, Dict, Any
from datetime import datetime
class TranscriptionRequest(BaseModel):
youtube_url: str
class QueryRequest(BaseModel):
query: str
session_id: str
class QueryResponse(BaseModel):
answer: str
session_id: str
source_documents: Optional[List[str]]
class VideoData(BaseModel):
video_id: str
user_id: str
title: str
source_type: str
source_url: Optional[str]
created_at: datetime = Field(default_factory=datetime.utcnow)
transcription: str
size: Optional[int] |