2024-07-28 15:21:51 -07:00
|
|
|
from typing import Optional
|
2024-02-24 17:03:55 -08:00
|
|
|
|
2024-07-28 15:21:51 -07:00
|
|
|
from pydantic import BaseModel, ConfigDict
|
2024-02-24 17:03:55 -08:00
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class KclCodeCompletionParams(BaseModel):
|
|
|
|
"""Extra params for the completions."""
|
|
|
|
|
2024-08-20 13:10:56 -04:00
|
|
|
language: str = ""
|
2024-02-24 17:03:55 -08:00
|
|
|
|
|
|
|
next_indent: Optional[int] = None
|
|
|
|
|
|
|
|
prompt_tokens: Optional[int] = None
|
|
|
|
|
|
|
|
suffix_tokens: Optional[int] = None
|
|
|
|
|
2024-08-20 13:10:56 -04:00
|
|
|
trim_by_indentation: bool = False
|
2024-02-24 17:03:55 -08:00
|
|
|
|
|
|
|
model_config = ConfigDict(protected_namespaces=())
|