mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-06-27 16:10:30 +08:00
Yannic Kilcher
41 lines
1.1 KiB
Python
41 lines
1.1 KiB
Python
from typing import Any
|
|
|
|
import pydantic
|
|
|
|
|
|
class Settings(pydantic.BaseSettings):
|
|
redis_host: str = "localhost"
|
|
redis_port: int = 6379
|
|
redis_db: int = 0
|
|
|
|
allowed_worker_compat_hashes: list[str] = ["distilgpt2"]
|
|
|
|
sse_retry_timeout: int = 15000
|
|
update_alembic: bool = True
|
|
alembic_retries: int = 5
|
|
alembic_retry_timeout: int = 1
|
|
|
|
postgres_host: str = "localhost"
|
|
postgres_port: str = "5432"
|
|
postgres_user: str = "postgres"
|
|
postgres_password: str = "postgres"
|
|
postgres_db: str = "postgres"
|
|
|
|
database_uri: str | None = None
|
|
|
|
@pydantic.validator("database_uri", pre=True)
|
|
def assemble_db_connection(cls, v: str | None, values: dict[str, Any]) -> Any:
|
|
if isinstance(v, str):
|
|
return v
|
|
return pydantic.PostgresDsn.build(
|
|
scheme="postgresql",
|
|
user=values.get("postgres_user"),
|
|
password=values.get("postgres_password"),
|
|
host=values.get("postgres_host"),
|
|
port=values.get("postgres_port"),
|
|
path=f"/{values.get('postgres_db') or ''}",
|
|
)
|
|
|
|
|
|
settings = Settings()
|