36 lines
854 B
Python
36 lines
854 B
Python
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=())
|