From 5ec5946a90ca2ad8c27021c2ebcff78223034621 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Sat, 30 May 2026 13:41:01 +0800 Subject: [PATCH] fix: derive sage_root from script location instead of hardcoded path --- scripts/migrate_llmcatelog_ids.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/scripts/migrate_llmcatelog_ids.py b/scripts/migrate_llmcatelog_ids.py index 0579843..8b0ff52 100644 --- a/scripts/migrate_llmcatelog_ids.py +++ b/scripts/migrate_llmcatelog_ids.py @@ -24,8 +24,9 @@ import argparse import sys import os -# 确保 Sage 虚拟环境的包可用 -sage_root = os.path.expanduser('~/repos/sage') +# 从脚本位置推断 sage 根目录(脚本在 pkgs/llmage/scripts/ 下) +_script_dir = os.path.dirname(os.path.abspath(__file__)) +sage_root = os.path.abspath(os.path.join(_script_dir, '..', '..', '..')) sys.path.insert(0, sage_root) sys.path.insert(0, os.path.join(sage_root, 'py3/lib/python3.10/site-packages'))