refactor: 移除硬编码配置,使用环境变量管理API密钥和路径

重构XinferenceEmbeddings和XinferenceReRankerModel类,移除硬编码的API密钥和路径
统一使用环境变量管理配置,提高安全性和灵活性
修改数据库路径为相对路径,增强可移植性
This commit is contained in:
2025-09-24 10:21:09 +08:00
parent 0541db1e34
commit d8d32ced9a
4 changed files with 30 additions and 23 deletions
+3 -3
View File
@@ -53,12 +53,12 @@ class BatchQueryResponse(BaseModel):
# 封装查询数据的相关代码
class QingDanDingEQueryService:
def __init__(self, db_path="/data/QueryRewrite/data/db/qingdan_ding_e_ku.db"):
self.db_path = db_path
def __init__(self):
self.db_path = f"{os.getcwd()}/data/db/qingdan_ding_e_ku.db"
self.top_k = TOP_K
# 初始化向量检索相关组件
self.embedding_function = XinferenceEmbeddings(api_key="")
self.embedding_function = XinferenceEmbeddings()
# 初始化向量数据库连接
self.ding_e_vector_db = SQLiteVSS(