refactor: 移除硬编码配置,使用环境变量管理API密钥和路径
重构XinferenceEmbeddings和XinferenceReRankerModel类,移除硬编码的API密钥和路径 统一使用环境变量管理配置,提高安全性和灵活性 修改数据库路径为相对路径,增强可移植性
This commit is contained in:
@@ -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(
|
||||
|
||||
Reference in New Issue
Block a user