# 测试环境

# 本地milvus
milvus_uri = "http://10.1.14.18:19530"

# 测试环境 mysql 数据库配置
mysql_config = {
        # "host": "10.168.100.144",
        # "port": 3306,
        # "host": "xia0miduo.gicp.net",
        "host": "10.1.14.18",
        # "port": 3336,
        "port": 3306,
        # "user": "root",
        "user": "test",
        # "password": "T@kai2025",
        "password": "J7#mQ!vP9xL@nR2sK&ea",
        "database": "jk_rag",
        # "database": "deepseek_local",
        # "database": "rag_master",
        "autocommit": True
}

# 测试minio配置
minio_config = {
    "minio_endpoint" : '10.1.14.18:9000',
    "minio_access_key" : 'jkragminio',
    "minio_secret_key" : 'Hfe2RpFaU92objoHxt*AxEjh',
    "minio_bucket" : 'jk-rag',
    "minio_url": "http://10.1.14.18:9000",
    "flag": False
}

# 测试环境vllm 链接
# vllm_url = "http://xia0miduo.gicp.net:8102/v1"
model_name_vllm_url_dict = {
    "deepseek-chat":"https://api.deepseek.com",
    "DeepSeek-R1-Distill-Qwen-14B": "http://xia0miduo.gicp.net:8102/v1",
    "deepcoder-14b-sft": "http://xia0miduo.gicp.net:8102/v1",
    "Qwen3-Coder-30B-loft": "http://127.0.0.1:8102/v1",
    "/opt/vllm/models/Qwen/Qwen3-30B-A3B-Instruct-2507": "http://127.0.0.1:8102/v1"
}


# 测试环境redis 配置
redis_config_dict = {
    "host": "10.1.14.17",
    "port": 6379,
    "db": 1
}

# 进度反馈Java中转服务器配置
progress_callback_config = {
    # "base_url": "http://10.10.10.5:9080",
    "base_url": "http://10.1.14.17:9080",
    # "default_url": "http://10.10.10.2:9080/resource/sse/send",
    # "default_url": "http://10.168.100.4:9080/resource/sse/send",
    "timeout": 5,
    "estimate_seconds": 120
}

# Dots OCR 解析器配置
dots_ocr_config = {
    "protocol": "http",
    "ip": "localhost",
    "port": 8090,
    "model_name": "model",
    "temperature": 0.1,
    "top_p": 1.0,
    "max_completion_tokens": 16384,
    "num_thread": 64,
    "dpi": 200,
    "output_dir": "./tmp_file/dots_parsed"
}

# PaddleOCR-VL 解析器配置
paddleocr_config = {
    "service_url": "http://127.0.0.1:8119",
    # "vl_server_url": "http://127.0.0.1:8118/v1",
    "vl_server_url": "http://127.0.0.1:8118/v1",
    "output_dir": "./tmp_file/paddleocr_parsed",
    "timeout": 600000,
    "title_optimization": {
        "enable": True,
        "api_key": "empty",
        "base_url": "http://xia0miduo.gicp.net:8102/v1",
        "model": "Qwen3-Coder-30B-loft",
        "max_retries": 3
    }
} 



# 线上环境

# # 线上环境 mysql 数据库配置
# mysql_config = {
#         "host": "127.0.0.1",
#         "port": 3306,
#         "user": "root",
#         "password": "Lx304307910",
#         "database": "chat_deepseek",
# }

# # 线上milvus配置
# milvus_uri = "http://127.0.0.1:19530"

# # 线上minio配置
# minio_config = {
#         "minio_endpoint" : '10.1.28.14:9000',
#         "minio_access_key" : 'uavadmin',
#         "minio_secret_key" : 'YEDG9DZ8oby97UGMyxW&',
#         "minio_bucket" : 'deepseek-doc',
#         "minio_url": "https://10.1.28.14:9000",
#         "flag": True
# }

# # 线上vllm 链接
# # vllm_url = "http://10.1.27.6:11817/v1"
# model_name_vllm_url_dict = {
#     "Qwen3-30B": "http://10.1.27.6:11817/v1",
#     "Qwen2-72B": "http://10.1.27.6:11818/v1",
# }

# # 线上环境redis 配置
# redis_config_dict = {
#     "host": "localhost",
#     "port": 6379,
#     "db": 1
# }