Update qlib/mcp_output/mcp_plugin/mcp_service.py

qlib/mcp_output/mcp_plugin/mcp_service.py

@@ -9,8 +9,12 @@ print(f"Using Qlib data path: {data_dir}")
 
 qlib.init(provider_uri=data_dir, region="cn")
 
-# 2️⃣ 输出 instruments 目录信息
-instruments_path = os.path.join(data_dir, "instruments")
+# 根据你的数据包实际目录结构调整
+instruments_path = os.path.join(data_dir, "instruments")  # 或 "cn_data/instruments"
+calendars_path = os.path.join(data_dir, "calendars")
+features_path = os.path.join(data_dir, "features")
+
+# 打印 instruments
 if os.path.exists(instruments_path):
     print("\n=== Instruments directory ===")
     for freq in os.listdir(instruments_path):
@@ -21,31 +25,13 @@ if os.path.exists(instruments_path):
 else:
     print("Instruments directory not found!")
 
-# 3️⃣ 输出 calendars 目录信息
-calendars_path = os.path.join(data_dir, "calendars")
-if os.path.exists(calendars_path):
-    print("\n=== Calendars directory ===")
-    files = os.listdir(calendars_path)
-    print(f"{len(files)} files -> {files[:10]}{'...' if len(files) > 10 else ''}")
-else:
-    print("Calendars directory not found!")
-
-# 4️⃣ 输出 features 目录信息
-features_path = os.path.join(data_dir, "features")
-if os.path.exists(features_path):
-    print("\n=== Features directory ===")
-    for f in os.listdir(features_path):
-        f_path = os.path.join(features_path, f)
-        if os.path.isdir(f_path):
-            n_files = len(os.listdir(f_path))
-            print(f"{f}/ : {n_files} feature files")
-else:
-    print("Features directory not found!")
-
-# 5️⃣ 可以进一步尝试获取标的列表
-instruments = D.list_instruments(freq="day", market="cn")
-print(f"\nNumber of day-frequency instruments: {len(instruments)}")
-print(f"Sample instruments: {instruments[:10]}")
+# 尝试列出 day 级标的
+try:
+    instruments = D.list_instruments(freq="day", market="cn")
+    print(f"\nNumber of day-frequency instruments: {len(instruments)}")
+    print(f"Sample instruments: {instruments[:10]}")
+except Exception as e:
+    print("Error listing day-frequency instruments:", e)
 
 
 # Path settings