Spaces:
Build error
Build error
File size: 623 Bytes
7b2511b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
import os
from dotenv import load_dotenv
load_dotenv()
from dataclasses import dataclass
from typing import Optional
@dataclass
class DatabaseSettings:
url: str
pool_size: int = 5
max_overflow: int = 10
pool_timeout: int = 30
@dataclass
class MemorySettings:
token_limit: int = 4096
token_counter: str = "simple" # "simple" or "tiktoken"
model_name: Optional[str] = None #
class Settings:
POSTGRES_DB_URL: str = os.getenv("POSTGRES_DB_URL")
print(POSTGRES_DB_URL)
SQLITE_DB_URL: str = os.getenv("SQLITE_DB_URL")
# Add other settings as needed
settings = Settings()
|