Test / src /config.py
Архипов Дмитрий
test
565e754
import os
from dotenv import load_dotenv
from qdrant_client import QdrantClient
from sqlalchemy import create_engine
# from src.data.parser import PyroSource
load_dotenv()
# Парсер
API_ID = os.environ.get("TELEGRAM_API_ID")
API_HASH = os.environ.get("TELEGRAM_API_HASH")
CHANNEL_ID = os.environ.get("CHANNEL_ID")
# pyro_source = PyroSource(api_id=API_ID, api_hash=API_HASH)
# Sql
DB_USER = os.getenv("DB_USER")
DB_PASS = os.getenv("DB_PASS")
DB_HOST = os.getenv("DB_HOST")
DB_PORT = os.getenv("DB_PORT")
DB_NAME = os.getenv("DB_NAME")
PATH_TO_CERT = os.getenv("PATH_TO_CERT")
connection_str = f"postgresql+psycopg2://{DB_USER}:{DB_PASS}@{DB_HOST}:{DB_PORT}/{DB_NAME}"
sql_client = create_engine(
connection_str,
connect_args={
"sslmode": "verify-full",
"sslrootcert": PATH_TO_CERT,
"target_session_attrs": "read-write"
}
)
# Qdrant
QDRANT_URL = os.getenv("QDRANT_URL")
qdrant_client = QdrantClient(url=QDRANT_URL)
# LLM
LLM_API_KEY = os.environ.get("OPENROUTER_API_KEY")
LLM = "qwen/qwen3-next-80b-a3b-instruct"
# Chat settings
CHAT_HISTORY_LENGTH = 5
ENABLE_QUESTION_ENRICHMENT = True