from typing import Optional
from uuid import UUID
from datetime import datetime

from fastapi_pagination import Params
from pydantic import BaseModel


class CreateEvaluationTask(BaseModel):
    name: str
    evaluation_dataset_id: int
    chat_engine: str = "default"
    run_size: Optional[int] = None


class EvaluationTaskOverview(BaseModel):
    not_start: int
    succeed: int
    errored: int
    progressing: int
    cancel: int
    avg_factual_correctness: Optional[float]
    avg_semantic_similarity: Optional[float]
    min_factual_correctness: Optional[float]
    min_semantic_similarity: Optional[float]
    max_factual_correctness: Optional[float]
    max_semantic_similarity: Optional[float]
    std_factual_correctness: Optional[float]
    std_semantic_similarity: Optional[float]


class EvaluationTaskSummary(BaseModel):
    id: Optional[int]
    name: str
    user_id: UUID
    dataset_id: int
    created_at: Optional[datetime]
    updated_at: Optional[datetime]

    summary: EvaluationTaskOverview


class UpdateEvaluationDataset(BaseModel):
    name: str


class CreateEvaluationDataset(BaseModel):
    name: str
    upload_id: Optional[int] = None


class ModifyEvaluationDatasetItem(BaseModel):
    query: str
    reference: str
    retrieved_contexts: list[str]
    extra: dict
    evaluation_dataset_id: int


class ParamsWithKeyword(Params):
    keyword: Optional[str] = None