2024-07-28 15:20:05 -07:00
|
|
|
import datetime
|
|
|
|
from typing import Any, Dict, List, Literal, Optional, Union
|
|
|
|
from uuid import UUID
|
2024-02-24 17:03:55 -08:00
|
|
|
|
2024-07-28 15:20:05 -07:00
|
|
|
from pydantic import AnyUrl, Base64Bytes, BaseModel, ConfigDict
|
|
|
|
from pydantic_extra_types.phone_numbers import PhoneNumber
|
2024-02-24 17:03:55 -08:00
|
|
|
|
|
|
|
from ..models.kcl_code_completion_params import KclCodeCompletionParams
|
2024-07-28 15:20:05 -07:00
|
|
|
from .base64data import Base64Data
|
2024-02-24 17:03:55 -08:00
|
|
|
|
|
|
|
|
|
|
|
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=())
|