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:
kingbri
2024-09-14 21:56:16 -04:00
parent 6f28cfe905
commit d013729b7d

View File

@@ -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):