mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-11 22:30:41 +03:00
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
Python
from functools import cached_property
|
|
from os import environ
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from dotenv import load_dotenv
|
|
from pydantic import HttpUrl, ValidationInfo, field_validator
|
|
from pydantic_settings import BaseSettings
|
|
|
|
from constants import API_PREFIX
|
|
|
|
BASE_DIR = Path(__file__).parent.parent
|
|
SHARED_DIR = BASE_DIR.resolve().joinpath("shared")
|
|
SHARED_DIR.mkdir(exist_ok=True)
|
|
|
|
SHARED_DIR.joinpath("logs").mkdir(exist_ok=True)
|
|
DIR_LOGS = SHARED_DIR.joinpath("logs")
|
|
|
|
env_path = f"{BASE_DIR}/settings/.env"
|
|
|
|
if environ.get("STAGE") == "runtests":
|
|
if "LOCALTEST" in environ:
|
|
env_path = f"{BASE_DIR}/settings/.env.local.runtests"
|
|
else:
|
|
env_path = f"{BASE_DIR}/settings/.env.ci.runtests"
|
|
|
|
load_dotenv(env_path, override=True)
|
|
|
|
|
|
class SentrySettings(BaseSettings):
|
|
SENTRY_DSN: HttpUrl | None = None
|
|
DEPLOY_ENVIRONMENT: str | None = None
|
|
SENTRY_TRACES_SAMPLE_RATE: float = 0.95
|
|
|
|
|
|
class AppSettings(SentrySettings, BaseSettings):
|
|
"""Application settings."""
|
|
|
|
PROJECT_NAME: str = "chat gpt bot"
|
|
APP_HOST: str = "0.0.0.0"
|
|
APP_PORT: int = 8000
|
|
STAGE: str = "dev"
|
|
DEBUG: bool = False
|
|
|
|
TELEGRAM_API_TOKEN: str = "123456789:AABBCCDDEEFFaabbccddeeff-1234567890"
|
|
# webhook settings
|
|
START_WITH_WEBHOOK: bool = False
|
|
DOMAIN: str = "https://localhost"
|
|
URL_PREFIX: str = ""
|
|
|
|
GPT_MODEL: str = "gpt-3.5-turbo-stream-AItianhuSpace"
|
|
# quantity of workers for uvicorn
|
|
WORKERS_COUNT: int = 1
|
|
# Enable uvicorn reloading
|
|
RELOAD: bool = False
|
|
|
|
@field_validator("START_WITH_WEBHOOK")
|
|
def star_with_webhook_validator(cls, field_value: Any, info: ValidationInfo) -> Any:
|
|
if field_value == "false":
|
|
return False
|
|
return field_value
|
|
|
|
@cached_property
|
|
def api_prefix(self) -> str:
|
|
if self.URL_PREFIX:
|
|
return "/" + "/".join([self.URL_PREFIX.strip("/"), API_PREFIX.strip("/")])
|
|
return API_PREFIX
|
|
|
|
@cached_property
|
|
def bot_webhook_url(self) -> str:
|
|
return "/".join([self.api_prefix, self.TELEGRAM_API_TOKEN])
|
|
|
|
class Config:
|
|
case_sensitive = True
|
|
|
|
|
|
def get_settings() -> AppSettings:
|
|
return AppSettings()
|
|
|
|
|
|
settings = get_settings()
|