"""
config.py

This module contains configuration settings for the Lung Cancer Treatment Recommendation System.
It centralizes parameters for embedding, chunking, database connections, search weights, and LLM configuration.
"""
import os
from dotenv import load_dotenv

load_dotenv(override=True)

# Embedding Configuration
EMBEDDING_MODEL = os.getenv('OLLAMA_EMBEDDING_MODEL', 'ncbi/MedCPT-Query-Encoder')

# Chunking Configuration
MAX_TOKENS = 400
MIN_TOKENS = 20

# Database Configuration
DB_HOST = os.getenv('DB_HOST', 'localhost')
DB_USER = os.getenv('DB_USER', 'my_user')
DB_PASSWORD = os.getenv('DB_PASSWORD', 'MyStrongPass@1234')
DB_NAME = os.getenv('DB_NAME', 'Ai_knowledge_spine_DB')
DB_PORT = 5432

# Qdrant Configuration
QDRANT_URL = os.getenv('QDRANT_URL')
QDRANT_API_KEY = os.getenv('QDRANT_API_KEY')
COLLECTION_NAME = os.getenv('COLLECTION_NAME', 'medical_knowledge_base_v2')

# Neo4j Configuration
NEO4J_URI = os.getenv('NEO4J_URI')
NEO4J_USER = os.getenv('NEO4J_USER')
NEO4J_PASSWORD = os.getenv('NEO4J_PASSWORD')

# Search Configuration
VECTOR_WEIGHT = 0.6
KEYWORD_WEIGHT = 0.4
TOP_K = 5

# LLM Configuration
LLM_MODEL = 'qwen3.5:9b'
LLM_HOST = 'http://127.0.0.1:11434'

# Data Configuration
DISEASE = 'nsclc'