diff --git a/![新版麦麦开始学习.bat b/![新版麦麦开始学习.bat index ca38689cf..41fc98368 100644 --- a/![新版麦麦开始学习.bat +++ b/![新版麦麦开始学习.bat @@ -19,7 +19,7 @@ if %ERRORLEVEL% neq 0 ( ) REM 运行预处理脚本 -python "%~dp0raw_data_preprocessor.py" +python "%~dp0scripts\raw_data_preprocessor.py" if %ERRORLEVEL% neq 0 ( echo 错误: raw_data_preprocessor.py 执行失败 pause @@ -27,7 +27,7 @@ if %ERRORLEVEL% neq 0 ( ) REM 运行信息提取脚本 -python "%~dp0info_extraction.py" +python "%~dp0scripts\info_extraction.py" if %ERRORLEVEL% neq 0 ( echo 错误: info_extraction.py 执行失败 pause @@ -35,7 +35,7 @@ if %ERRORLEVEL% neq 0 ( ) REM 运行OpenIE导入脚本 -python "%~dp0import_openie.py" +python "%~dp0scripts\import_openie.py" if %ERRORLEVEL% neq 0 ( echo 错误: import_openie.py 执行失败 pause diff --git a/import_openie.py b/scripts/import_openie.py similarity index 98% rename from import_openie.py rename to scripts/import_openie.py index 43fcd21fb..b669bc6b7 100644 --- a/import_openie.py +++ b/scripts/import_openie.py @@ -4,7 +4,9 @@ # print("未找到quick_algo库,无法使用quick_algo算法") # print("请安装quick_algo库 - 在lib.quick_algo中,执行命令:python setup.py build_ext --inplace") - +import sys +import os +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) from typing import Dict, List from src.plugins.knowledge.src.lpmmconfig import PG_NAMESPACE, global_config @@ -15,8 +17,9 @@ from src.plugins.knowledge.src.kg_manager import KGManager from src.common.logger import get_module_logger from src.plugins.knowledge.src.utils.hash import get_sha256 -# 添加在现有导入之后 -import sys + +# 添加项目根目录到 sys.path + logger = get_module_logger("LPMM知识库-OpenIE导入") diff --git a/info_extraction.py b/scripts/info_extraction.py similarity index 98% rename from info_extraction.py rename to scripts/info_extraction.py index b6ad8a9c2..e1274ec6b 100644 --- a/info_extraction.py +++ b/scripts/info_extraction.py @@ -4,6 +4,8 @@ import signal from concurrent.futures import ThreadPoolExecutor, as_completed from threading import Lock, Event import sys +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) +# 添加项目根目录到 sys.path import tqdm diff --git a/raw_data_preprocessor.py b/scripts/raw_data_preprocessor.py similarity index 95% rename from raw_data_preprocessor.py rename to scripts/raw_data_preprocessor.py index 7b8d400cf..89ee3ee14 100644 --- a/raw_data_preprocessor.py +++ b/scripts/raw_data_preprocessor.py @@ -2,10 +2,13 @@ import json import os from pathlib import Path import sys # 新增系统模块导入 +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) from src.common.logger import get_module_logger logger = get_module_logger("LPMM数据库-原始数据处理") +# 添加项目根目录到 sys.path + def check_and_create_dirs(): """检查并创建必要的目录""" diff --git a/scripts/tempCodeRunnerFile.py b/scripts/tempCodeRunnerFile.py new file mode 100644 index 000000000..026931f7b --- /dev/null +++ b/scripts/tempCodeRunnerFile.py @@ -0,0 +1 @@ +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) \ No newline at end of file diff --git a/scripts/tool_call_benchmark.py b/scripts/tool_call_benchmark.py index a3e282734..1ab8bdd14 100644 --- a/scripts/tool_call_benchmark.py +++ b/scripts/tool_call_benchmark.py @@ -1,3 +1,9 @@ +import sys +import os + +# 添加项目根目录到 sys.path +sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))) + import asyncio import time from src.plugins.models.utils_model import LLMRequest @@ -64,12 +70,6 @@ async def test_with_tool_calls(): prompt = "请分析当前天气情况,并查询今日历史上的重要事件。并且3.9和3.11谁比较大?请使用适当的工具来获取这些信息。" prompt = """ 你的名字是麦麦,你包容开放,情绪敏感,有时候有些搞怪幽默, 是一个学习心理学和脑科学的女大学生,现在在读大二,你会刷贴吧,有时候会喜欢说一些奇怪的话,喜欢刷小红书 -刚刚你的内心想法是:漂移菌提到罐罐被吃完了,可以顺着这个梗继续玩一下,比如假装委屈"那今晚的加班费是不是也要被吃掉了"或者"猫娘罢工警告"。不过薯薯和薯宝之前已经接了不少梗,漂移菌刚刚也参与了,可能话题热度还在,可以再互动一下。如果没人接话,或许可以问问大家有没有遇到过类似"代码写完但奖励被吃掉"的搞笑职场经历,换个轻松的话题方向。 - -暂时不需要使用工具。 ------------------------------------ -现在是2025-04-25 17:38:37,你正在上网,和qq群里的网友们聊天,以下是正在进行的聊天内容: -2025-04-25 17:34:08麦麦(你) 说:[表达了:顽皮、嬉戏。]; 2025-04-25 17:34:39漂移菌 说:@麦麦。(id:3936257206) 你是一只猫娘; 2025-04-25 17:34:42薯宝 说:🤣; 2025-04-25 17:34:43麦麦(你) 说:行啊 工资分我一半;