mirror of
https://github.com/theroyallab/tabbyAPI.git
synced 2026-04-27 01:38:56 +00:00
Config: Add aliases for logging config
Config.yml and args take in two different values. Signed-off-by: kingbri <bdashore3@proton.me>
This commit is contained in:
@@ -1,4 +1,4 @@
|
|||||||
from pydantic import BaseModel, ConfigDict, Field
|
from pydantic import AliasChoices, BaseModel, ConfigDict, Field
|
||||||
from typing import List, Literal, Optional, Union
|
from typing import List, Literal, Optional, Union
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
@@ -47,21 +47,31 @@ class NetworkConfig(BaseModel):
|
|||||||
description=("Decide whether to send error tracebacks over the API"),
|
description=("Decide whether to send error tracebacks over the API"),
|
||||||
)
|
)
|
||||||
api_servers: Optional[List[Literal["OAI", "Kobold"]]] = Field(
|
api_servers: Optional[List[Literal["OAI", "Kobold"]]] = Field(
|
||||||
[
|
default_factory=list,
|
||||||
"OAI",
|
|
||||||
],
|
|
||||||
description=("API servers to enable. Options: (OAI, Kobold)"),
|
description=("API servers to enable. Options: (OAI, Kobold)"),
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
# TODO: Migrate config.yml to have the log_ prefix
|
||||||
|
# This is a breaking change.
|
||||||
class LoggingConfig(BaseModel):
|
class LoggingConfig(BaseModel):
|
||||||
"""Model for logging configuration."""
|
"""Model for logging configuration."""
|
||||||
|
|
||||||
log_prompt: Optional[bool] = Field(False, description=("Enable prompt logging"))
|
log_prompt: Optional[bool] = Field(
|
||||||
log_generation_params: Optional[bool] = Field(
|
False,
|
||||||
False, description=("Enable generation parameter logging")
|
description=("Enable prompt logging"),
|
||||||
|
validation_alias=AliasChoices("log_prompt", "prompt"),
|
||||||
|
)
|
||||||
|
log_generation_params: Optional[bool] = Field(
|
||||||
|
False,
|
||||||
|
description=("Enable generation parameter logging"),
|
||||||
|
validation_alias=AliasChoices("log_generation_params", "generation_params"),
|
||||||
|
)
|
||||||
|
log_requests: Optional[bool] = Field(
|
||||||
|
False,
|
||||||
|
description=("Enable request logging"),
|
||||||
|
validation_alias=AliasChoices("log_requests", "requests"),
|
||||||
)
|
)
|
||||||
log_requests: Optional[bool] = Field(False, description=("Enable request logging"))
|
|
||||||
|
|
||||||
|
|
||||||
class ModelConfig(BaseModel):
|
class ModelConfig(BaseModel):
|
||||||
|
|||||||
Reference in New Issue
Block a user