import os

API_KEY = "sk-263038d4bf4e46a0bed16532587cff40"

# {project_root}/medsynthai
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))

# LLM model configuration based on agno
LLM_CONFIG = {
    "deepseek": {
        "class": "DeepSeek",
        "params": {
            "id": "deepseek-chat",
            "api_key": API_KEY,
            "base_url": "https://api.deepseek.com"
        }
    },
    "ollama": {
        "class": "Ollama",
        "params": {
            "id": "qwen2.5:latest",
            "host": "127.0.0.1"
        }
    },
    "deepseek-v3": {
        "class": "OpenAILike",
        "params": {
            "id": "deepseek-v3",
            "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
            "api_key": API_KEY
        }
    },
    "deepseek-r1": {
        "class": "OpenAILike",
        "params": {
            "id": "deepseek-r1",
            "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
            "api_key": API_KEY
        }
    },
    "qwen-max": {
        "class": "OpenAILike",
        "params": {
            "id": "qwen-max",
            "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
            "api_key": API_KEY
        }
    },
    "qwen-vl-max": {
        "class": "OpenAILike",  # 使用OpenAI兼容类
        "params": {
            "id": "qwen-vl-max",
            "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",  # OpenAI兼容端点      # 降低随机性，提高结果一致性
            "api_key": API_KEY
        }
    },
    "aliyun": {
        "class": "OpenAILike",
        "params": {
            "id": "qwen-max",  # 默认使用qwen-max模型
            "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
            "api_key": API_KEY
        }
    }
    
}

AGENT_CONFIG = {
    "medical_image_analysis_agent": "qwen-vl-max"
}

RAG_CONFIG = {
    "lightrag": {
        "working_dir": "./Vector_DB_Med",
        "tokenizer_name": "trueto/medbert-base-chinese",
        "model_name": "trueto/medbert-base-chinese",
        "embedding_dim": 768,
        "max_token_size": 512
    },
    "chroma_db": {
        "api_key": API_KEY,
        "base_url": "https://dashscope.aliyuncs.com/compatible-mode/v1",
        "collection_name": "doctor",
        "batch_size": 100,
        "chroma_db_path": os.path.join(BASE_DIR, "static/rag/chroma_db"),
        "csv_path": os.path.join(BASE_DIR, "static/files/zhongkang_doctor_list.csv")
    }
}