schemas/log.py

from typing import Optional

from pydantic import BaseModel


class LogCreate(BaseModel):
    llmPrompt: Optional[str] = None
    llmResponse: Optional[str] = None
    userRequest: str
    llm_classifier: Optional[str] = None
    query_type: Optional[str] = None
    userName: str