guohanghui commited on
Commit
6201aa8
·
verified ·
1 Parent(s): 015c8c4

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +5 -18
Dockerfile CHANGED
@@ -19,29 +19,16 @@ RUN python /tmp/fix_qlib_version.py
19
 
20
  # 4. 环境变量
21
  ENV PYTHONPATH=/app/qlib/source:$PYTHONPATH
22
- ENV QLIB_DATA_DIR=/home/user/.qlib/qlib_data
23
  EXPOSE 7860
24
  ENV MCP_TRANSPORT=http
25
  ENV MCP_PORT=7860
26
 
27
  # ✅ 关键修改:在启动前自动下载数据
28
- ENTRYPOINT ["bash", "-c", "\
29
- echo '🚀 Preparing Qlib data...'; \
30
- if [ ! -d \"$HOME/qlib/qlib_data/cn_data\" ]; then \
31
- echo '📦 Downloading Qlib daily data...'; \
32
- python -m qlib.cli.data qlib_data --target_dir $HOME/qlib/qlib_data/cn_data --region cn; \
33
- else \
34
- echo '✅ Qlib daily data already exists.'; \
35
- fi; \
36
- if [ ! -d \"$HOME/qlib/qlib_data/cn_data_1min\" ]; then \
37
- echo '📦 Downloading Qlib 1min data...'; \
38
- python -m qlib.cli.data qlib_data --target_dir $HOME/qlib/qlib_data/cn_data_1min --region cn --interval 1min; \
39
- else \
40
- echo '✅ Qlib 1min data already exists.'; \
41
- fi; \
42
- echo '🚀 Starting MCP service...'; \
43
- exec python qlib/mcp_output/start_mcp.py \
44
- "]
45
 
46
  # 🧩 保留原 CMD(不会被覆盖)
47
  CMD ["python", "qlib/mcp_output/start_mcp.py"]
 
19
 
20
  # 4. 环境变量
21
  ENV PYTHONPATH=/app/qlib/source:$PYTHONPATH
22
+ ENV QLIB_DATA_DIR=/home/user/qlib/qlib_data
23
  EXPOSE 7860
24
  ENV MCP_TRANSPORT=http
25
  ENV MCP_PORT=7860
26
 
27
  # ✅ 关键修改:在启动前自动下载数据
28
+ RUN mkdir -p /home/user/qlib/qlib_data/cn_data && \
29
+ wget -q https://github.com/chenditc/investment_data/releases/latest/download/qlib_bin.tar.gz -O /tmp/qlib_bin.tar.gz && \
30
+ tar -zxvf /tmp/qlib_bin.tar.gz -C /home/user/qlib/qlib_data/cn_data --strip-components=1 && \
31
+ rm -f /tmp/qlib_bin.tar.gz
 
 
 
 
 
 
 
 
 
 
 
 
 
32
 
33
  # 🧩 保留原 CMD(不会被覆盖)
34
  CMD ["python", "qlib/mcp_output/start_mcp.py"]