* better default types Signed-off-by: Jess Frazelle <github@jessfraz.com> * more mypy fixes Signed-off-by: Jess Frazelle <github@jessfraz.com> * more fixes for mypy Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix mypy; Signed-off-by: Jess Frazelle <github@jessfraz.com> * updates Signed-off-by: Jess Frazelle <github@jessfraz.com> * fix mypy; Signed-off-by: Jess Frazelle <github@jessfraz.com> * I have generated the latest API! --------- Signed-off-by: Jess Frazelle <github@jessfraz.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
32 lines
689 B
Python
32 lines
689 B
Python
from typing import List, Optional
|
|
|
|
from pydantic import BaseModel, ConfigDict
|
|
|
|
from ..models.kcl_code_completion_params import KclCodeCompletionParams
|
|
|
|
|
|
class KclCodeCompletionRequest(BaseModel):
|
|
"""A request to generate KCL code completions."""
|
|
|
|
extra: KclCodeCompletionParams = {"language": "", "trim_by_indentation": False} # type: ignore
|
|
|
|
max_tokens: Optional[int] = None
|
|
|
|
n: Optional[int] = None
|
|
|
|
nwo: Optional[str] = None
|
|
|
|
prompt: str = ""
|
|
|
|
stop: Optional[List[str]] = None
|
|
|
|
stream: bool = False
|
|
|
|
suffix: str = ""
|
|
|
|
temperature: Optional[float] = None
|
|
|
|
top_p: Optional[float] = None
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|