mirror of
https://github.com/Balshgit/gpt_chat_bot.git
synced 2025-09-11 22:30:41 +03:00
23031b0777
* update chat_microservice * reformat logger_conf * add database * add service and repository logic * fix constants gpt base url * add models endpoints
120 lines
3.2 KiB
Python
120 lines
3.2 KiB
Python
from functools import cached_property, lru_cache
|
|
from os import environ
|
|
from pathlib import Path
|
|
from typing import Any
|
|
|
|
from dotenv import load_dotenv
|
|
from pydantic import model_validator
|
|
from pydantic_settings import BaseSettings
|
|
from yarl import URL
|
|
|
|
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: str | 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
|
|
# quantity of workers for uvicorn
|
|
WORKERS_COUNT: int = 1
|
|
# Enable uvicorn reloading
|
|
RELOAD: bool = False
|
|
|
|
TELEGRAM_API_TOKEN: str = "123456789:AABBCCDDEEFFaabbccddeeff-1234567890"
|
|
# webhook settings
|
|
START_WITH_WEBHOOK: bool = False
|
|
DOMAIN: str = "https://localhost"
|
|
URL_PREFIX: str = ""
|
|
|
|
DB_FILE: Path = SHARED_DIR / "chat_gpt.db"
|
|
DB_ECHO: bool = False
|
|
|
|
# ==== gpt settings ====
|
|
GPT_MODEL: str = "gpt-3.5-turbo-stream-DeepAi"
|
|
GPT_BASE_HOST: str = "http://chat_service:8858"
|
|
|
|
ENABLE_JSON_LOGS: bool = True
|
|
ENABLE_SENTRY_LOGS: bool = False
|
|
GRAYLOG_HOST: str | None = None
|
|
GRAYLOG_PORT: int | None = None
|
|
LOG_TO_FILE: str | None = None
|
|
|
|
@model_validator(mode="before") # type: ignore[arg-type]
|
|
def validate_boolean_fields(self) -> Any:
|
|
values_dict: dict[str, Any] = self # type: ignore[assignment]
|
|
for value in (
|
|
"ENABLE_JSON_LOGS",
|
|
"ENABLE_SENTRY_LOGS",
|
|
"START_WITH_WEBHOOK",
|
|
"RELOAD",
|
|
"DEBUG",
|
|
):
|
|
setting_value: str | None = values_dict.get(value)
|
|
if setting_value and setting_value.lower() == "false":
|
|
values_dict[value] = False
|
|
return values_dict
|
|
|
|
@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 token_part(self) -> str:
|
|
return self.TELEGRAM_API_TOKEN[15:30]
|
|
|
|
@cached_property
|
|
def bot_webhook_url(self) -> str:
|
|
return "/".join([self.api_prefix, self.token_part])
|
|
|
|
@cached_property
|
|
def db_url(self) -> URL:
|
|
"""
|
|
Assemble database URL from settings.
|
|
|
|
:return: database URL.
|
|
"""
|
|
return URL.build(
|
|
scheme="sqlite+aiosqlite",
|
|
path=f"///{self.DB_FILE}",
|
|
)
|
|
|
|
class Config:
|
|
case_sensitive = True
|
|
|
|
|
|
@lru_cache(maxsize=None)
|
|
def get_settings() -> AppSettings:
|
|
return AppSettings()
|
|
|
|
|
|
settings = get_settings()
|