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()