config.py 3.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130
  1. # 测试环境
  2. # 本地milvus
  3. milvus_uri = "http://10.1.14.18:19530"
  4. # 测试环境 mysql 数据库配置
  5. mysql_config = {
  6. # "host": "10.168.100.144",
  7. # "port": 3306,
  8. # "host": "xia0miduo.gicp.net",
  9. "host": "10.1.14.18",
  10. # "port": 3336,
  11. "port": 3306,
  12. # "user": "root",
  13. "user": "test",
  14. # "password": "T@kai2025",
  15. "password": "J7#mQ!vP9xL@nR2sK&ea",
  16. "database": "jk_rag",
  17. # "database": "deepseek_local",
  18. # "database": "rag_master",
  19. "autocommit": True
  20. }
  21. # 测试minio配置
  22. minio_config = {
  23. "minio_endpoint" : '10.1.14.18:9000',
  24. "minio_access_key" : 'jkragminio',
  25. "minio_secret_key" : 'Hfe2RpFaU92objoHxt*AxEjh',
  26. "minio_bucket" : 'jk-rag',
  27. "minio_url": "http://10.1.14.18:9000",
  28. "flag": False
  29. }
  30. # 测试环境vllm 链接
  31. # vllm_url = "http://xia0miduo.gicp.net:8102/v1"
  32. model_name_vllm_url_dict = {
  33. "deepseek-chat":"https://api.deepseek.com",
  34. "DeepSeek-R1-Distill-Qwen-14B": "http://xia0miduo.gicp.net:8102/v1",
  35. "deepcoder-14b-sft": "http://xia0miduo.gicp.net:8102/v1",
  36. "Qwen3-Coder-30B-loft": "http://127.0.0.1:8102/v1",
  37. "/opt/vllm/models/Qwen/Qwen3-30B-A3B-Instruct-2507": "http://127.0.0.1:8102/v1"
  38. }
  39. # 测试环境redis 配置
  40. redis_config_dict = {
  41. "host": "10.1.14.17",
  42. "port": 6379,
  43. "db": 1
  44. }
  45. # 进度反馈Java中转服务器配置
  46. progress_callback_config = {
  47. # "base_url": "http://10.10.10.5:9080",
  48. "base_url": "http://10.1.14.17:9080",
  49. # "default_url": "http://10.10.10.2:9080/resource/sse/send",
  50. # "default_url": "http://10.168.100.4:9080/resource/sse/send",
  51. "timeout": 5,
  52. "estimate_seconds": 120
  53. }
  54. # Dots OCR 解析器配置
  55. dots_ocr_config = {
  56. "protocol": "http",
  57. "ip": "localhost",
  58. "port": 8090,
  59. "model_name": "model",
  60. "temperature": 0.1,
  61. "top_p": 1.0,
  62. "max_completion_tokens": 16384,
  63. "num_thread": 64,
  64. "dpi": 200,
  65. "output_dir": "./tmp_file/dots_parsed"
  66. }
  67. # PaddleOCR-VL 解析器配置
  68. paddleocr_config = {
  69. "service_url": "http://127.0.0.1:8119",
  70. # "vl_server_url": "http://127.0.0.1:8118/v1",
  71. "vl_server_url": "http://127.0.0.1:8118/v1",
  72. "output_dir": "./tmp_file/paddleocr_parsed",
  73. "timeout": 600000,
  74. "title_optimization": {
  75. "enable": True,
  76. "api_key": "empty",
  77. "base_url": "http://xia0miduo.gicp.net:8102/v1",
  78. "model": "Qwen3-Coder-30B-loft",
  79. "max_retries": 3
  80. }
  81. }
  82. # 线上环境
  83. # # 线上环境 mysql 数据库配置
  84. # mysql_config = {
  85. # "host": "127.0.0.1",
  86. # "port": 3306,
  87. # "user": "root",
  88. # "password": "Lx304307910",
  89. # "database": "chat_deepseek",
  90. # }
  91. # # 线上milvus配置
  92. # milvus_uri = "http://127.0.0.1:19530"
  93. # # 线上minio配置
  94. # minio_config = {
  95. # "minio_endpoint" : '10.1.28.14:9000',
  96. # "minio_access_key" : 'uavadmin',
  97. # "minio_secret_key" : 'YEDG9DZ8oby97UGMyxW&',
  98. # "minio_bucket" : 'deepseek-doc',
  99. # "minio_url": "https://10.1.28.14:9000",
  100. # "flag": True
  101. # }
  102. # # 线上vllm 链接
  103. # # vllm_url = "http://10.1.27.6:11817/v1"
  104. # model_name_vllm_url_dict = {
  105. # "Qwen3-30B": "http://10.1.27.6:11817/v1",
  106. # "Qwen2-72B": "http://10.1.27.6:11818/v1",
  107. # }
  108. # # 线上环境redis 配置
  109. # redis_config_dict = {
  110. # "host": "localhost",
  111. # "port": 6379,
  112. # "db": 1
  113. # }