diff --git a/.gitignore b/.gitignore index 51a11d8c2..deec90be9 100644 --- a/.gitignore +++ b/.gitignore @@ -188,3 +188,7 @@ cython_debug/ # jieba jieba.cache + + +# vscode +/.vscode \ No newline at end of file diff --git a/bot.py b/bot.py index 8ef087476..d0922e4d6 100644 --- a/bot.py +++ b/bot.py @@ -17,11 +17,11 @@ print(rainbow_text) '''彩蛋''' # 初次启动检测 -if not os.path.exists("config/bot_config.toml") or not os.path.exists(".env"): +if not os.path.exists("config/bot_config.toml"): logger.info("检测到bot_config.toml不存在,正在从模板复制") import shutil - shutil.copy("config/bot_config_template.toml", "config/bot_config.toml") + shutil.copy("templete/bot_config_template.toml", "config/bot_config.toml") logger.info("复制完成,请修改config/bot_config.toml和.env.prod中的配置后重新启动") # 初始化.env 默认ENVIRONMENT=prod diff --git a/src/plugins/models/utils_model.py b/src/plugins/models/utils_model.py index 5d1f90ebb..abc6f027b 100644 --- a/src/plugins/models/utils_model.py +++ b/src/plugins/models/utils_model.py @@ -213,12 +213,11 @@ class LLM_request: ) return content, reasoning_content - async def get_embedding(self, text: str, model: str = "BAAI/bge-m3") -> Union[list, None]: + async def get_embedding(self, text: str) -> Union[list, None]: """异步方法:获取文本的embedding向量 Args: text: 需要获取embedding的文本 - model: 使用的模型名称,默认为"BAAI/bge-m3" Returns: list: embedding向量,如果失败则返回None @@ -233,7 +232,7 @@ class LLM_request: endpoint="/embeddings", prompt=text, payload={ - "model": model, + "model": self.model_name, "input": text, "encoding_format": "float" }, diff --git a/config/auto_format.py b/templete/auto_format.py similarity index 100% rename from config/auto_format.py rename to templete/auto_format.py diff --git a/config/bot_config_template.toml b/templete/bot_config_template.toml similarity index 100% rename from config/bot_config_template.toml rename to templete/bot_config_template.toml