import datetime
from typing import Any, Dict, List, Literal, Optional, Union
from uuid import UUID

from pydantic import AnyUrl, Base64Bytes, BaseModel, ConfigDict
from pydantic_extra_types.phone_numbers import PhoneNumber

from ..models.kcl_code_completion_params import KclCodeCompletionParams
from .base64data import Base64Data


class KclCodeCompletionRequest(BaseModel):
    """A request to generate KCL code completions."""

    extra: Optional[KclCodeCompletionParams] = None

    max_tokens: Optional[int] = None

    n: Optional[int] = None

    nwo: Optional[str] = None

    prompt: Optional[str] = None

    stop: Optional[List[str]] = None

    stream: Optional[bool] = None

    suffix: Optional[str] = None

    temperature: Optional[float] = None

    top_p: Optional[float] = None

    model_config = ConfigDict(protected_namespaces=())